blob: 83e46be39400c6644cfef1ad5dc0b9ca4912b43c (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
package kamon.instrumentation;
import kamon.TraceContext;
import scala.concurrent.impl.Future;
import scala.Option;
public aspect PromiseCompletingRunnableInstrumentation {
public final Option<TraceContext> Future.PromiseCompletingRunnable.context = TraceContext.current();
pointcut run(Future.PromiseCompletingRunnable runnable)
: execution(* scala.concurrent.impl.Future.PromiseCompletingRunnable.run()) && this(runnable);
void around(Future.PromiseCompletingRunnable runnable)
: run(runnable) {
if(runnable.context.isDefined()) {
TraceContext.set((TraceContext) runnable.context.get());
proceed(runnable);
TraceContext.clear();
} else {
proceed(runnable);
}
}
}
|