blob: 392f53b8844ed0fce0ada8050333256d4b073a7d (
plain) (
tree)
|
|
package kamon.trace
import spray.routing.directives.BasicDirectives
import spray.routing._
import kamon.Tracer
import java.util.concurrent.atomic.AtomicLong
import scala.util.Try
import java.net.InetAddress
trait UowDirectives extends BasicDirectives {
def uow: Directive0 = mapRequest { request =>
val uowHeader = request.headers.find(_.name == "X-UOW")
val generatedUow = uowHeader.map(_.value).getOrElse(UowDirectives.newUow)
Tracer.set(Tracer.context().getOrElse(Tracer.newTraceContext()).copy(uow = generatedUow))
request
}
}
object UowDirectives {
val uowCounter = new AtomicLong
val hostnamePrefix = Try(InetAddress.getLocalHost.getHostName).getOrElse("unknown-localhost")
def newUow = "%s-%s".format(hostnamePrefix, uowCounter.incrementAndGet())
}
|