aboutsummaryrefslogtreecommitdiff
path: root/kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala
diff options
context:
space:
mode:
Diffstat (limited to 'kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala')
-rw-r--r--kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala6
1 files changed, 2 insertions, 4 deletions
diff --git a/kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala b/kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala
index 3caba77c..057e339d 100644
--- a/kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala
+++ b/kamon-trace/src/main/scala/kamon/trace/instrumentation/ActorRefTellInstrumentation.scala
@@ -34,11 +34,9 @@ class ActorCellInvokeInstrumentation {
class EnvelopeTracingContext {
@DeclareMixin("akka.dispatch.Envelope")
- def mixin: ContextAware = new ContextAware {
- val traceContext: Option[TraceContext] = Trace.context()
- }
+ def mixin: ContextAware = ContextAware.default
- @Pointcut("execution(akka.dispatch.ContextAware.new(..)) && this(ctx)")
+ @Pointcut("execution(akka.dispatch.Envelope.new(..)) && this(ctx)")
def requestRecordInit(ctx: ContextAware): Unit = {}
@After("requestRecordInit(ctx)")