diff options
Diffstat (limited to 'kamon-trace/src/main/scala')
-rw-r--r-- | kamon-trace/src/main/scala/kamon/trace/Trace.scala | 4 |
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) |