kamon-core/src/main/scala/kamon/trace/TraceSubscriptions.scala


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.trace

import akka.actor.{ Terminated, ActorRef, Actor }

class TraceSubscriptions extends Actor {
  import TraceSubscriptions._

  var subscribers: List[ActorRef] = Nil

  def receive = {
    case Subscribe(newSubscriber) ⇒
      if (!subscribers.contains(newSubscriber))
        subscribers = context.watch(newSubscriber) :: subscribers

    case Unsubscribe(leavingSubscriber) ⇒
      subscribers = subscribers.filter(_ == leavingSubscriber)

    case Terminated(terminatedSubscriber) ⇒
      subscribers = subscribers.filter(_ == terminatedSubscriber)

    case trace: TraceInfo ⇒
      subscribers.foreach(_ ! trace)
  }
}

object TraceSubscriptions {
  case class Subscribe(subscriber: ActorRef)
  case class Unsubscribe(subscriber: ActorRef)
}