blob: a38b10c9322f0ce37eb093641c54b140bd07a083 (
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
|
package kamon.actor
import akka.actor.{ActorRef, Actor}
import kamon.TraceContext
trait TraceableActor extends Actor with TracingImplicitConversions {
final def receive = {
case a: Any => {
a match {
case TraceableMessage(ctx, message) => {
TraceContext.current.set(ctx)
tracedReceive(message)
TraceContext.current.remove()
/** Publish the partial context information to the EventStream */
context.system.eventStream.publish(ctx)
}
case message: Any => tracedReceive(message)
}
}
}
def tracedReceive: Receive
}
class TraceableActorRef(val target: ActorRef) {
def !! (message: Any)(implicit sender: ActorRef) = {
val traceableMessage = TraceableMessage(TraceContext.current.get().fork, message)
target.tell(traceableMessage, sender)
}
}
trait TracingImplicitConversions {
implicit def fromActorRefToTraceableActorRef(actorRef: ActorRef) = new TraceableActorRef(actorRef)
}
case class TraceableMessage(traceContext: TraceContext, message: Any)
|