diff options
author | Ivan Topolnak <itopolnak@despegar.com> | 2014-01-30 18:23:33 -0300 |
---|---|---|
committer | Ivan Topolnak <itopolnak@despegar.com> | 2014-01-30 18:23:33 -0300 |
commit | a15e17d2462105ad8b72054be58dc9e8f9dc64ed (patch) | |
tree | 9e8c3d1fa732918dc600538bab24940bd18b6e79 /kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala | |
parent | e47ed388f6ae147e550c39aa2a3638791ec65b66 (diff) | |
download | Kamon-a15e17d2462105ad8b72054be58dc9e8f9dc64ed.tar.gz Kamon-a15e17d2462105ad8b72054be58dc9e8f9dc64ed.tar.bz2 Kamon-a15e17d2462105ad8b72054be58dc9e8f9dc64ed.zip |
wip
Diffstat (limited to 'kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala')
-rw-r--r-- | kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala b/kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala index 5600d582..b8725dd7 100644 --- a/kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala +++ b/kamon-core/src/main/scala/kamon/instrumentation/FutureTracing.scala @@ -17,29 +17,29 @@ package kamon.instrumentation import org.aspectj.lang.annotation._ import org.aspectj.lang.ProceedingJoinPoint -import kamon.trace.{ ContextAware, TraceContext, Trace } +import kamon.trace.{ TraceContextAware, TraceRecorder } @Aspect class FutureTracing { @DeclareMixin("scala.concurrent.impl.CallbackRunnable || scala.concurrent.impl.Future.PromiseCompletingRunnable") - def mixin: ContextAware = ContextAware.default + def mixin: TraceContextAware = new TraceContextAware {} @Pointcut("execution((scala.concurrent.impl.CallbackRunnable || scala.concurrent.impl.Future.PromiseCompletingRunnable).new(..)) && this(runnable)") - def futureRelatedRunnableCreation(runnable: ContextAware): Unit = {} + def futureRelatedRunnableCreation(runnable: TraceContextAware): Unit = {} @After("futureRelatedRunnableCreation(runnable)") - def afterCreation(runnable: ContextAware): Unit = { + def afterCreation(runnable: TraceContextAware): Unit = { // Force traceContext initialization. runnable.traceContext } @Pointcut("execution(* (scala.concurrent.impl.CallbackRunnable || scala.concurrent.impl.Future.PromiseCompletingRunnable).run()) && this(runnable)") - def futureRelatedRunnableExecution(runnable: ContextAware) = {} + def futureRelatedRunnableExecution(runnable: TraceContextAware) = {} @Around("futureRelatedRunnableExecution(runnable)") - def aroundExecution(pjp: ProceedingJoinPoint, runnable: ContextAware): Any = { - Trace.withContext(runnable.traceContext) { + def aroundExecution(pjp: ProceedingJoinPoint, runnable: TraceContextAware): Any = { + TraceRecorder.withContext(runnable.traceContext) { pjp.proceed() } } |