! all: improve the metric recorders infrastructure

1 files changed, 30 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/trace/TracerExtensionSettings.scala b/kamon-core/src/main/scala/kamon/trace/TracerExtensionSettings.scala
new file mode 100644
index 00000000..e6c2d3ef
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/trace/TracerExtensionSettings.scala
@@ -0,0 +1,30 @@
+package kamon.trace
+
+import java.util.concurrent.TimeUnit
+
+import akka.actor.ActorSystem
+
+case class TraceSettings(levelOfDetail: LevelOfDetail, sampler: Sampler)
+
+object TraceSettings {
+  def apply(system: ActorSystem): TraceSettings = {
+    val tracerConfig = system.settings.config.getConfig("kamon.trace")
+
+    val detailLevel: LevelOfDetail = tracerConfig.getString("level-of-detail") match {
+      case "metrics-only" ⇒ LevelOfDetail.MetricsOnly
+      case "simple-trace" ⇒ LevelOfDetail.SimpleTrace
+      case other          ⇒ sys.error(s"Unknown tracer level of detail [$other] present in the configuration file.")
+    }
+
+    val sampler: Sampler =
+      if (detailLevel == LevelOfDetail.MetricsOnly) NoSampling
+      else tracerConfig.getString("sampling") match {
+        case "all"       ⇒ SampleAll
+        case "random"    ⇒ new RandomSampler(tracerConfig.getInt("random-sampler.chance"))
+        case "ordered"   ⇒ new OrderedSampler(tracerConfig.getInt("ordered-sampler.interval"))
+        case "threshold" ⇒ new ThresholdSampler(tracerConfig.getDuration("threshold-sampler.minimum-elapsed-time", TimeUnit.NANOSECONDS))
+      }
+
+    TraceSettings(detailLevel, sampler)
+  }
+}
\ No newline at end of file