From 197746563e47783ed4b5f43e94c9aa63734081f6 Mon Sep 17 00:00:00 2001
From: Ivan Topolnak <ivantopo@gmail.com>
Date: Thu, 6 Jun 2013 18:03:20 -0300
Subject: wip

---
 src/main/scala/kamon/Kamon.scala        | 21 ++++++++++++++++++++
 src/main/scala/kamon/TraceContext.scala | 34 +++++++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+)

(limited to 'src')

diff --git a/src/main/scala/kamon/Kamon.scala b/src/main/scala/kamon/Kamon.scala
index ef5f8044..c1b97722 100644
--- a/src/main/scala/kamon/Kamon.scala
+++ b/src/main/scala/kamon/Kamon.scala
@@ -29,3 +29,24 @@ object Kamon {
   def publish(tx: FullTransaction) = publisher ! tx
 
 }
+
+
+
+object Tracer {
+  val ctx = new ThreadLocal[Option[TraceContext]] {
+    override def initialValue() = None
+  }
+
+  def context() = ctx.get()
+  def clear = ctx.remove()
+  def set(traceContext: TraceContext) = ctx.set(Some(traceContext))
+
+  def start = ??? //set(newTraceContext)
+  def stop = ctx.get match {
+    case Some(context) => context.close
+    case None =>
+  }
+
+  //def newTraceContext(): TraceContext = TraceContext()
+
+}
diff --git a/src/main/scala/kamon/TraceContext.scala b/src/main/scala/kamon/TraceContext.scala
index 19ebc578..351446f3 100644
--- a/src/main/scala/kamon/TraceContext.scala
+++ b/src/main/scala/kamon/TraceContext.scala
@@ -28,3 +28,37 @@ object TraceContext {
 trait TraceEntry
 
 case class CodeBlockExecutionTime(name: String, begin: Long, end: Long) extends TraceEntry
+
+
+
+case class TransactionTrace(id: UUID, start: Long, end: Long, entries: Seq[TraceEntry])
+
+
+
+
+
+object Collector {
+
+}
+
+trait TraceEntryStorage {
+  def store(entry: TraceEntry): Boolean
+}
+
+class TransactionContext(val id: UUID, private val storage: TraceEntryStorage) {
+  def store(entry: TraceEntry) = storage.store(entry)
+}
+
+object ThreadLocalTraceEntryStorage extends TraceEntryStorage {
+
+  private val storage = new ThreadLocal[List[TraceEntry]] {
+    override def initialValue(): List[TraceEntry] = Nil
+    def update(f: List[TraceEntry] => List[TraceEntry]) = set(f(get()))
+  }
+
+  def store(entry: TraceEntry): Boolean = {
+    storage.update(entry :: _)
+    true
+  }
+}
+
-- 
cgit v1.2.3