diff options
author | Ivan Topolnak <ivantopo@gmail.com> | 2013-11-07 18:41:33 -0300 |
---|---|---|
committer | Ivan Topolnak <ivantopo@gmail.com> | 2013-11-07 18:41:33 -0300 |
commit | bf86900669d649308f4914c54e6fe076510506a6 (patch) | |
tree | d8bf9af9f5c8a946d757137a303f6956c05edb03 /kamon-trace/src/main/scala/kamon/trace/Trace.scala | |
parent | 2b63540e5fffab545d0846cfb3dab5c0e1d0c9e1 (diff) | |
download | Kamon-bf86900669d649308f4914c54e6fe076510506a6.tar.gz Kamon-bf86900669d649308f4914c54e6fe076510506a6.tar.bz2 Kamon-bf86900669d649308f4914c54e6fe076510506a6.zip |
halfway to our own NewRelic Agent
Diffstat (limited to 'kamon-trace/src/main/scala/kamon/trace/Trace.scala')
-rw-r--r-- | kamon-trace/src/main/scala/kamon/trace/Trace.scala | 27 |
1 files changed, 19 insertions, 8 deletions
diff --git a/kamon-trace/src/main/scala/kamon/trace/Trace.scala b/kamon-trace/src/main/scala/kamon/trace/Trace.scala index 232b7420..6e01ad26 100644 --- a/kamon-trace/src/main/scala/kamon/trace/Trace.scala +++ b/kamon-trace/src/main/scala/kamon/trace/Trace.scala @@ -26,7 +26,12 @@ object Trace extends ExtensionId[TraceExtension] with ExtensionIdProvider { def context() = traceContext.value def set(ctx: TraceContext) = traceContext.value = Some(ctx) - def start(name: String)(implicit system: ActorSystem) = set(newTraceContext) + def clear: Unit = traceContext.value = None + def start(name: String)(implicit system: ActorSystem) = { + val ctx = newTraceContext() + ctx.start(name) + set(ctx) + } def finish(): Option[TraceContext] = { val ctx = context() @@ -39,21 +44,27 @@ object Trace extends ExtensionId[TraceExtension] with ExtensionIdProvider { } class TraceExtension(system: ExtendedActorSystem) extends Kamon.Extension { - def manager: ActorRef = system.actorOf(Props[TraceManager]) + val manager: ActorRef = system.actorOf(Props[TraceManager], "kamon-trace") } -class TraceManager extends Actor { +class TraceManager extends Actor with ActorLogging { var listeners: Seq[ActorRef] = Seq.empty def receive = { - case Register => listeners = sender +: listeners + case Register => + listeners = sender +: listeners + log.info("Registered [{}] as listener for Kamon traces", sender) + case segment: UowSegment => - context.child(segment.id.toString) match { - case Some(agreggator) => agreggator ! segment - case None => context.actorOf(UowTraceAggregator.props(self, 30 seconds)) - } + val tracerName = segment.id.toString + context.child(tracerName).getOrElse(newTracer(tracerName)) ! segment case trace: UowTrace => + println("Delivering a trace to: " + listeners) listeners foreach(_ ! trace) } + + def newTracer(name: String): ActorRef = { + context.actorOf(UowTraceAggregator.props(self, 30 seconds), name) + } } |