diff options
Diffstat (limited to 'kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala')
-rw-r--r-- | kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala b/kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala index 58473f13..6d38f3d7 100644 --- a/kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala +++ b/kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala @@ -9,7 +9,9 @@ import kamon.{Tracer, Kamon} trait UowDirectives extends BasicDirectives { def uow: Directive0 = mapRequest { request => - val generatedUow = Some(UowDirectives.newUow) + val uowHeader = request.headers.find(_.name == "X-UOW") + + val generatedUow = uowHeader.map(_.value).orElse(Some(UowDirectives.newUow)) println("Generated UOW: "+generatedUow) Tracer.set(Tracer.newTraceContext().copy(userContext = generatedUow)) |