diff options
Diffstat (limited to 'kamon-trace/src/main/scala/kamon/trace/Trace.scala')
-rw-r--r-- | kamon-trace/src/main/scala/kamon/trace/Trace.scala | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/kamon-trace/src/main/scala/kamon/trace/Trace.scala b/kamon-trace/src/main/scala/kamon/trace/Trace.scala index d5bb1ddf..0ef438d4 100644 --- a/kamon-trace/src/main/scala/kamon/trace/Trace.scala +++ b/kamon-trace/src/main/scala/kamon/trace/Trace.scala @@ -27,10 +27,12 @@ object Trace extends ExtensionId[TraceExtension] with ExtensionIdProvider { def set(ctx: TraceContext) = traceContext.value = Some(ctx) def clear: Unit = traceContext.value = None - def start(name: String)(implicit system: ActorSystem) = { - val ctx = newTraceContext() + def start(name: String)(implicit system: ActorSystem): TraceContext = { + val ctx = newTraceContext(name) ctx.start(name) set(ctx) + + ctx } def transformContext(f: TraceContext => TraceContext): Unit = { @@ -44,7 +46,7 @@ object Trace extends ExtensionId[TraceExtension] with ExtensionIdProvider { } // TODO: FIX - def newTraceContext()(implicit system: ActorSystem): TraceContext = TraceContext(Kamon(Trace), tranid.getAndIncrement) + def newTraceContext(name: String)(implicit system: ActorSystem): TraceContext = TraceContext(Kamon(Trace), tranid.getAndIncrement, name) } class TraceExtension(system: ExtendedActorSystem) extends Kamon.Extension { |