blob: ad87950570adfc75f8e2ed47c68422d8ba2f2e8f (
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
|
package kamon.actor
import akka.actor.{ActorRef, Actor}
trait EnhancedActor extends Actor {
protected[this] var transactionContext: TransactionContext = _
final def receive = {
case a: Any => {
a match {
case ContextAwareMessage(ctx, message) => {
transactionContext = ctx
println(s"Actor ${self.path.toString}. Current context: ${transactionContext}")
wrappedReceive(message)
}
case message: Any => wrappedReceive(message)
}
}
}
def wrappedReceive: Receive
def superTell(target: ActorRef, message: Any) = {
target.tell(ContextAwareMessage(transactionContext, message), self)
}
}
case class ContextAwareMessage(context: TransactionContext, message: Any)
case class TransactionContext(id: Long, entries: List[ContextEntry]) {
def append(entry: ContextEntry) = this.copy(entries = entry :: this.entries)
}
sealed trait ContextEntry
case class DeveloperComment(comment: String) extends ContextEntry
case class MessageExecutionTime(actorPath: String, begin: Long, end: Long) extends ContextEntry
|