blob: 7d26016ed8cc21b2dbb45fdfcafc998188aba053 (
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
package akka.instrumentation
import org.aspectj.lang.annotation._
import kamon.trace.{ Trace, ContextAware }
import akka.dispatch.sysmsg.EarliestFirstSystemMessageList
import org.aspectj.lang.ProceedingJoinPoint
@Aspect
class SystemMessageTraceContextMixin {
@DeclareMixin("akka.dispatch.sysmsg.SystemMessage+")
def mixin: ContextAware = ContextAware.default
@Pointcut("execution(akka.dispatch.sysmsg.SystemMessage+.new(..)) && this(ctx)")
def envelopeCreation(ctx: ContextAware): Unit = {}
@After("envelopeCreation(ctx)")
def afterEnvelopeCreation(ctx: ContextAware): Unit = {
// Necessary to force the initialization of ContextAware at the moment of creation.
ctx.traceContext
}
}
@Aspect
class RepointableActorRefTraceContextMixin {
@DeclareMixin("akka.actor.RepointableActorRef")
def mixin: ContextAware = ContextAware.default
@Pointcut("execution(akka.actor.RepointableActorRef.new(..)) && this(ctx)")
def envelopeCreation(ctx: ContextAware): Unit = {}
@After("envelopeCreation(ctx)")
def afterEnvelopeCreation(ctx: ContextAware): Unit = {
// Necessary to force the initialization of ContextAware at the moment of creation.
ctx.traceContext
}
@Pointcut("execution(* akka.actor.RepointableActorRef.point(..)) && this(repointableActorRef)")
def repointableActorRefCreation(repointableActorRef: ContextAware): Unit = {}
@Around("repointableActorRefCreation(repointableActorRef)")
def afterRepointableActorRefCreation(pjp: ProceedingJoinPoint, repointableActorRef: ContextAware): Any = {
Trace.withContext(repointableActorRef.traceContext) {
pjp.proceed()
}
}
}
@Aspect
class ActorSystemMessagePassingTracing {
@Pointcut("execution(* akka.actor.ActorCell.invokeAll$1(..)) && args(messages, *)")
def systemMessageProcessing(messages: EarliestFirstSystemMessageList): Unit = {}
@Around("systemMessageProcessing(messages)")
def aroundSystemMessageInvoke(pjp: ProceedingJoinPoint, messages: EarliestFirstSystemMessageList): Any = {
if (messages.nonEmpty) {
val ctx = messages.head.asInstanceOf[ContextAware].traceContext
Trace.withContext(ctx)(pjp.proceed())
} else pjp.proceed()
}
}
|