From 61029e115272b9af3f4460b311d3a2e650c806e3 Mon Sep 17 00:00:00 2001
From: Diego <diegolparra@gmail.com>
Date: Tue, 15 Aug 2017 13:16:45 -0300
Subject: expose SpanCodec through Kamon::contextCodec

---
 kamon-core/src/main/scala/kamon/Kamon.scala | 7 ++++++-
 1 file changed, 6 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 884f2d93..2c0561e2 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.{Context, Storage}
+import kamon.context.{Codec, Context, Storage}
 import org.slf4j.LoggerFactory
 
 import scala.util.Try
@@ -42,6 +42,7 @@ object Kamon extends MetricLookup with ReporterRegistry with Tracer {
   private val _reporters = new ReporterRegistryImpl(_metrics, _config)
   private val _tracer = Tracer.Default(Kamon, _reporters, _config)
   private val _contextStorage = Storage.ThreadLocal()
+  private val _contextCodec = new Codec(_config)
   private var _onReconfigureHooks = Seq.empty[OnReconfigureHook]
 
   def environment: Environment =
@@ -57,6 +58,7 @@ object Kamon extends MetricLookup with ReporterRegistry with Tracer {
     _metrics.reconfigure(config)
     _reporters.reconfigure(config)
     _tracer.reconfigure(config)
+    _contextCodec.reconfigure(config)
 
     _onReconfigureHooks.foreach(hook => {
       Try(hook.onReconfigure(config)).failed.foreach(error =>
@@ -98,6 +100,9 @@ object Kamon extends MetricLookup with ReporterRegistry with Tracer {
   override def identityProvider: IdentityProvider =
     _tracer.identityProvider
 
+  def contextCodec(): Codec =
+      _contextCodec
+
   def currentContext(): Context =
     _contextStorage.current()
 
-- 
cgit v1.2.3