From 5c8eb362ea6a957449419cec6c48b54159d4c68e Mon Sep 17 00:00:00 2001
From: Ivan Topolnak <ivantopo@gmail.com>
Date: Mon, 11 Nov 2013 18:11:33 -0300
Subject: Fix UowDirectives

---
 kamon-trace/src/main/scala/kamon/trace/Trace.scala | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'kamon-trace/src/main')

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)
-- 
cgit v1.2.3