blob: 44c71eafcbe7f7f1a64021d672622f9287b1f942 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
package kamon.spray
import akka.actor.ExtendedActorSystem
case class SprayExtensionSettings(
includeTraceTokenHeader: Boolean,
traceTokenHeaderName: String,
nameGenerator: SprayNameGenerator,
clientInstrumentationLevel: ClientInstrumentationLevel.Level)
object SprayExtensionSettings {
def apply(system: ExtendedActorSystem): SprayExtensionSettings = {
val config = system.settings.config.getConfig("kamon.spray")
val includeTraceTokenHeader: Boolean = config.getBoolean("automatic-trace-token-propagation")
val traceTokenHeaderName: String = config.getString("trace-token-header-name")
val nameGeneratorFQN = config.getString("name-generator")
val nameGenerator: SprayNameGenerator = system.dynamicAccess.createInstanceFor[SprayNameGenerator](nameGeneratorFQN, Nil).get // let's bubble up any problems.
val clientInstrumentationLevel: ClientInstrumentationLevel.Level = config.getString("client.instrumentation-level") match {
case "request-level" ⇒ ClientInstrumentationLevel.RequestLevelAPI
case "host-level" ⇒ ClientInstrumentationLevel.HostLevelAPI
case other ⇒ sys.error(s"Invalid client instrumentation level [$other] found in configuration.")
}
SprayExtensionSettings(includeTraceTokenHeader, traceTokenHeaderName, nameGenerator, clientInstrumentationLevel)
}
}
object ClientInstrumentationLevel {
sealed trait Level
case object RequestLevelAPI extends Level
case object HostLevelAPI extends Level
}
|