diff options
author | Ivan Topolnak <ivantopo@gmail.com> | 2013-10-10 18:52:09 -0300 |
---|---|---|
committer | Ivan Topolnak <ivantopo@gmail.com> | 2013-10-10 18:52:09 -0300 |
commit | e148933747e8fde17b6ac324df0dee70b8cb9ebc (patch) | |
tree | 6cd084809e8184c74052bcb4201adb3f8e5e4004 /kamon-core/src/main/scala/kamon/trace/UowTracing.scala | |
parent | dbac9011999384a57c0ecc2c9ab5c49870cdce45 (diff) | |
download | Kamon-e148933747e8fde17b6ac324df0dee70b8cb9ebc.tar.gz Kamon-e148933747e8fde17b6ac324df0dee70b8cb9ebc.tar.bz2 Kamon-e148933747e8fde17b6ac324df0dee70b8cb9ebc.zip |
complete spray client instrumentation with experimental branch
Diffstat (limited to 'kamon-core/src/main/scala/kamon/trace/UowTracing.scala')
-rw-r--r-- | kamon-core/src/main/scala/kamon/trace/UowTracing.scala | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/kamon-core/src/main/scala/kamon/trace/UowTracing.scala b/kamon-core/src/main/scala/kamon/trace/UowTracing.scala index c794656d..9ba3813a 100644 --- a/kamon-core/src/main/scala/kamon/trace/UowTracing.scala +++ b/kamon-core/src/main/scala/kamon/trace/UowTracing.scala @@ -2,8 +2,13 @@ package kamon.trace import akka.actor._ import scala.concurrent.duration.Duration +import kamon.trace.UowTracing._ +import scala.Some +import kamon.trace.UowTracing.WebExternalFinish import kamon.trace.UowTracing.Finish import kamon.trace.UowTracing.Rename +import kamon.trace.UowTrace +import kamon.trace.UowTracing.WebExternalStart import scala.Some sealed trait UowSegment { @@ -18,7 +23,9 @@ object UowTracing { case class Start() extends AutoTimestamp case class Finish() extends AutoTimestamp case class Rename(name: String) extends AutoTimestamp - case class WebExternal(start: Long, end: Long, host: String) extends AutoTimestamp + case class WebExternalStart(id: Long, host: String) extends AutoTimestamp + case class WebExternalFinish(id: Long) extends AutoTimestamp + case class WebExternal(start: Long, finish: Long, host: String) extends AutoTimestamp } case class UowTrace(name: String, segments: Seq[UowSegment]) @@ -30,8 +37,14 @@ class UowTraceAggregator(reporting: ActorRef, aggregationTimeout: Duration) exte var name: Option[String] = None var segments: Seq[UowSegment] = Nil + var pendingExternal = List[WebExternalStart]() + def receive = { case finish: Finish => segments = segments :+ finish; finishTracing() + case wes: WebExternalStart => pendingExternal = pendingExternal :+ wes + case finish @ WebExternalFinish(id) => pendingExternal.find(_.id == id).map(start => { + segments = segments :+ WebExternal(start.timestamp, finish.timestamp, start.host) + }) case Rename(newName) => name = Some(newName) case segment: UowSegment => segments = segments :+ segment case ReceiveTimeout => |