From 5c8eb362ea6a957449419cec6c48b54159d4c68e Mon Sep 17 00:00:00 2001 From: Ivan Topolnak Date: Mon, 11 Nov 2013 18:11:33 -0300 Subject: Fix UowDirectives --- kamon-spray/src/main/scala/kamon/spray/UowDirectives.scala | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'kamon-spray/src/main/scala/kamon/spray') 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 } } -- cgit v1.2.3