Included test for cases in which TraceContext are not available at creation time

1 files changed, 2 insertions, 15 deletions

diff --git a/src/main/scala/kamon/TraceContext.scala b/src/main/scala/kamon/TraceContext.scala
index 5787167b..e3582c60 100644
--- a/src/main/scala/kamon/TraceContext.scala
+++ b/src/main/scala/kamon/TraceContext.scala
@@ -14,9 +14,9 @@ object TraceContext {
     override def initialValue(): Option[TraceContext] = None
   }
 
-  def current = context.get
+  def current = context.get()
 
-  def clear = context.remove
+  def clear = context.remove()
 
   def set(ctx: TraceContext) = context.set(Some(ctx))
 
@@ -27,16 +27,3 @@ trait TraceEntry
 case class MessageExecutionTime(actorPath: ActorPath, initiated: Long, ended: Long)
 
 case class CodeBlockExecutionTime(blockName: String, begin: Long, end: Long) extends TraceEntry
-
-
-
-
-trait TraceSupport {
-  def withContext[Out](func: => Any => Out, ctx: TraceContext) = {
-    TraceContext.set(ctx)
-    val result = func
-    TraceContext.clear
-
-    result
-  }
-}