diff options
author | Diego Parra <diegolparra@gmail.com> | 2013-05-29 23:36:28 -0300 |
---|---|---|
committer | Diego Parra <diegolparra@gmail.com> | 2013-05-29 23:36:28 -0300 |
commit | b4ae12153f7c971ef13c9048584fd0b611de5e2d (patch) | |
tree | cfc9315f3c6d100c3f95677b4f47682f6b41b4e4 /src/main | |
parent | 03487dbfcaa9c5295c9735018e1a9a2bd728c780 (diff) | |
download | Kamon-b4ae12153f7c971ef13c9048584fd0b611de5e2d.tar.gz Kamon-b4ae12153f7c971ef13c9048584fd0b611de5e2d.tar.bz2 Kamon-b4ae12153f7c971ef13c9048584fd0b611de5e2d.zip |
WIP: aspects refactor
Diffstat (limited to 'src/main')
3 files changed, 16 insertions, 28 deletions
diff --git a/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj b/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj index bd2a4d47..867ec372 100644 --- a/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj +++ b/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj @@ -1,29 +1,26 @@ package kamon.instrumentation; import kamon.TraceContext; -import scala.concurrent.impl.Future; -import scala.Option; privileged public aspect PromiseCompletingRunnableInstrumentation { declare parents : Future.PromiseCompletingRunnable extends TraceContextHolder; - - pointcut run(Future.PromiseCompletingRunnable runnable) + pointcut run(scala.concurrent.impl.Future.PromiseCompletingRunnable runnable) : execution(* scala.concurrent.impl.Future.PromiseCompletingRunnable.run()) && this(runnable); - void around(TraceContextHolder runnable) + void around(Object runnable) : run(runnable) { - if(runnable.getContext().isDefined()) { - System.out.println("########################################################3 There is some context"); - TraceContext.set(runnable.getContext().get()); - proceed(runnable); + TraceContextHolder contextHolder = (TraceContextHolder) runnable; + + if(contextHolder.context().isDefined()) { + TraceContext.set(contextHolder.context().get()); + proceed(contextHolder); TraceContext.clear(); } else { - System.out.println("########################################################3 There is NOOOOO context"); - proceed(runnable); + proceed(contextHolder); } } } diff --git a/src/main/scala/kamon/instrumentation/TraceContextHolder.java b/src/main/scala/kamon/instrumentation/TraceContextHolder.java deleted file mode 100644 index 529ce86a..00000000 --- a/src/main/scala/kamon/instrumentation/TraceContextHolder.java +++ /dev/null @@ -1,17 +0,0 @@ -package kamon.instrumentation; -import scala.Option; -import kamon.TraceContext; - -abstract public class TraceContextHolder implements Runnable { - public final Option<TraceContext> context; - - public TraceContextHolder() { - context = TraceContext.current(); - } - - public Option<TraceContext> getContext() { - return context; - } - - -}
\ No newline at end of file diff --git a/src/main/scala/kamon/instrumentation/TraceContextHolder.scala b/src/main/scala/kamon/instrumentation/TraceContextHolder.scala new file mode 100644 index 00000000..b44547dd --- /dev/null +++ b/src/main/scala/kamon/instrumentation/TraceContextHolder.scala @@ -0,0 +1,8 @@ +package kamon.instrumentation; + +import kamon.TraceContext; +import scala.Option; + +class TraceContextHolder { + final val context:Option[TraceContext] = TraceContext.current +}
\ No newline at end of file |