From 902da6b5410325411a0473f923632fa92d39838e Mon Sep 17 00:00:00 2001 From: Ivan Topolnak Date: Mon, 26 Aug 2013 19:24:56 -0300 Subject: Some sort of basic logging with UOW --- kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'kamon-uow/src/main/scala/kamon/logging/UowDirectives.scala') 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)) -- cgit v1.2.3