From 96e89d0908c479360d4a915c916e341e148a91c7 Mon Sep 17 00:00:00 2001 From: Ivan Topolnak Date: Wed, 25 Sep 2013 19:45:54 -0300 Subject: wip --- kamon-core/src/main/scala/kamon/trace/UowTracing.scala | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'kamon-core/src/main/scala/kamon/trace') diff --git a/kamon-core/src/main/scala/kamon/trace/UowTracing.scala b/kamon-core/src/main/scala/kamon/trace/UowTracing.scala index 8efc46b2..48def942 100644 --- a/kamon-core/src/main/scala/kamon/trace/UowTracing.scala +++ b/kamon-core/src/main/scala/kamon/trace/UowTracing.scala @@ -1,8 +1,13 @@ package kamon.trace -import akka.actor.{Props, ActorRef, Actor} +import akka.actor._ import kamon.trace.UowTracing.{Start, Finish, Rename} import scala.concurrent.duration.Duration +import kamon.trace.UowTracing.Finish +import kamon.trace.UowTracing.Rename +import kamon.trace.UowTrace +import kamon.trace.UowTracing.Start +import scala.Some sealed trait UowSegment { def timestamp: Long @@ -22,7 +27,7 @@ object UowTracing { case class UowTrace(name: String, segments: Seq[UowSegment]) -class UowTraceAggregator(reporting: ActorRef, aggregationTimeout: Duration) extends Actor { +class UowTraceAggregator(reporting: ActorRef, aggregationTimeout: Duration) extends Actor with ActorLogging { context.setReceiveTimeout(aggregationTimeout) self ! Start() @@ -33,6 +38,9 @@ class UowTraceAggregator(reporting: ActorRef, aggregationTimeout: Duration) exte case finish: Finish => segments = segments :+ finish; finishTracing() case Rename(newName) => name = Some(newName) case segment: UowSegment => segments = segments :+ segment + case ReceiveTimeout => + log.warning("Transaction {} did not complete properly, the recorded segments are: {}", name, segments) + context.stop(self) } def finishTracing(): Unit = { -- cgit v1.2.3