aboutsummaryrefslogtreecommitdiff
path: root/kamon-spray
diff options
context:
space:
mode:
authorIvan Topolnak <ivantopo@gmail.com>2013-11-11 18:11:33 -0300
committerIvan Topolnak <ivantopo@gmail.com>2013-11-11 18:11:33 -0300
commit5c8eb362ea6a957449419cec6c48b54159d4c68e (patch)
treee6ea0b2a7b79d9d5440f767336a824030b5b60e3 /kamon-spray
parent777ad06ea31f8e1508ded4332b7152c0cd182146 (diff)
downloadKamon-5c8eb362ea6a957449419cec6c48b54159d4c68e.tar.gz
Kamon-5c8eb362ea6a957449419cec6c48b54159d4c68e.tar.bz2
Kamon-5c8eb362ea6a957449419cec6c48b54159d4c68e.zip
Fix UowDirectives
Diffstat (limited to 'kamon-spray')
-rw-r--r--kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala3
1 files changed, 1 insertions, 2 deletions
diff --git a/kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala b/kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala
index 685388df..0c0ea214 100644
--- a/kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala
+++ b/kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala
@@ -12,8 +12,7 @@ trait UowDirectives extends BasicDirectives {
val uowHeader = request.headers.find(_.name == "X-UOW")
val generatedUow = uowHeader.map(_.value).getOrElse(UowDirectives.newUow)
- // TODO: Tracer will always have a context at this point, just rename the uow.
- Trace.set(Trace.context().getOrElse(Trace.newTraceContext()).copy(uow = generatedUow))
+ Trace.transformContext(_.copy(uow = generatedUow))
request
}
}