From cd54e4ed73734dbabebbf22e4fa288c9b047992e Mon Sep 17 00:00:00 2001
From: Ivan Topolnjak <ivantopo@gmail.com>
Date: Thu, 24 Aug 2017 11:23:21 +0200
Subject: introduce the SpanCustomizer API

---
 kamon-core/src/main/scala/kamon/Kamon.scala | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'kamon-core/src/main/scala/kamon/Kamon.scala')

diff --git a/kamon-core/src/main/scala/kamon/Kamon.scala b/kamon-core/src/main/scala/kamon/Kamon.scala
index f251b1ec..e83f6a5c 100644
--- a/kamon-core/src/main/scala/kamon/Kamon.scala
+++ b/kamon-core/src/main/scala/kamon/Kamon.scala
@@ -24,7 +24,7 @@ import scala.concurrent.Future
 import java.time.Duration
 import java.util.concurrent.{Executors, ScheduledExecutorService, ScheduledThreadPoolExecutor}
 
-import kamon.context.{Codecs, Context, Storage}
+import kamon.context.{Codecs, Context, Key, Storage}
 import org.slf4j.LoggerFactory
 
 import scala.util.Try
@@ -118,6 +118,9 @@ object Kamon extends MetricLookup with ReporterRegistry with Tracer {
     }
   }
 
+  def withContextKey[T, K](key: Key[K], value: K)(f: => T): T =
+    withContext(currentContext().withKey(key, value))(f)
+
 
   override def loadReportersFromConfig(): Unit =
     _reporterRegistry.loadReportersFromConfig()
-- 
cgit v1.2.3