aboutsummaryrefslogtreecommitdiff
path: root/kamon-trace/src/main/scala
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-trace/src/main/scala
parent777ad06ea31f8e1508ded4332b7152c0cd182146 (diff)
downloadKamon-5c8eb362ea6a957449419cec6c48b54159d4c68e.tar.gz
Kamon-5c8eb362ea6a957449419cec6c48b54159d4c68e.tar.bz2
Kamon-5c8eb362ea6a957449419cec6c48b54159d4c68e.zip
Fix UowDirectives
Diffstat (limited to 'kamon-trace/src/main/scala')
-rw-r--r--kamon-trace/src/main/scala/kamon/trace/Trace.scala4
1 files changed, 4 insertions, 0 deletions
diff --git a/kamon-trace/src/main/scala/kamon/trace/Trace.scala b/kamon-trace/src/main/scala/kamon/trace/Trace.scala
index 1956509c..d5bb1ddf 100644
--- a/kamon-trace/src/main/scala/kamon/trace/Trace.scala
+++ b/kamon-trace/src/main/scala/kamon/trace/Trace.scala
@@ -33,6 +33,10 @@ object Trace extends ExtensionId[TraceExtension] with ExtensionIdProvider {
set(ctx)
}
+ def transformContext(f: TraceContext => TraceContext): Unit = {
+ context.map(f).foreach(set(_))
+ }
+
def finish(): Option[TraceContext] = {
val ctx = context()
ctx.map(_.finish)