+ core: java variant of withMdc

1 files changed, 6 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/trace/logging/MdcKeysSupport.scala b/kamon-core/src/main/scala/kamon/trace/logging/MdcKeysSupport.scala
index 5097fe6b..756138d2 100644
--- a/kamon-core/src/main/scala/kamon/trace/logging/MdcKeysSupport.scala
+++ b/kamon-core/src/main/scala/kamon/trace/logging/MdcKeysSupport.scala
@@ -18,6 +18,7 @@ package kamon.trace.logging
 
 import kamon.trace.TraceLocal.AvailableToMdc
 import kamon.trace.{ Tracer, EmptyTraceContext, MetricsOnlyContext, TraceContext }
+import kamon.util.Supplier
 
 import org.slf4j.MDC
 
@@ -28,6 +29,9 @@ trait MdcKeysSupport {
     try thunk finally keys.foreach(key ⇒ MDC.remove(key))
   }
 
+  // Java variant.
+  def withMdc[A](thunk:Supplier[A]): A = withMdc(thunk.get)
+
   private[this] def copyToMdc(traceContext: TraceContext): Iterable[String] = traceContext match {
     case ctx: MetricsOnlyContext ⇒
       ctx.traceLocalStorage.underlyingStorage.collect {
@@ -37,3 +41,5 @@ trait MdcKeysSupport {
     case EmptyTraceContext ⇒ Iterable.empty[String]
   }
 }
+
+object MdcKeysSupport extends MdcKeysSupport
\ No newline at end of file