diff options
Diffstat (limited to 'kamon-spray')
| -rw-r--r-- | kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala | 3 |
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 } } |