From 109ba0e17bf6d5ffbcf784e808fbb8e8f3d32a8c Mon Sep 17 00:00:00 2001
From: Ivan Topolnjak <ivantopo@gmail.com>
Date: Sun, 12 Oct 2014 02:26:09 +0200
Subject: = core: fix NPE when finishing segments, fixes #88

---
 .../src/main/scala/test/SimpleRequestProcessor.scala        | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'kamon-playground/src/main')

diff --git a/kamon-playground/src/main/scala/test/SimpleRequestProcessor.scala b/kamon-playground/src/main/scala/test/SimpleRequestProcessor.scala
index 1db2db67..b6bcc677 100644
--- a/kamon-playground/src/main/scala/test/SimpleRequestProcessor.scala
+++ b/kamon-playground/src/main/scala/test/SimpleRequestProcessor.scala
@@ -21,6 +21,7 @@ import akka.routing.RoundRobinPool
 import akka.util.Timeout
 import kamon.Kamon
 import kamon.metric.Subscriptions.TickMetricSnapshot
+import kamon.metric.TraceMetrics.HttpClientRequest
 import kamon.metric._
 import kamon.spray.KamonTraceDirectives
 import kamon.trace.TraceRecorder
@@ -124,6 +125,16 @@ object SimpleRequestProcessor extends App with SimpleRoutingApp with RequestBuil
             throw new NullPointerException
             "okk"
           }
+        } ~
+        path("segment") {
+          complete {
+            val segment = TraceRecorder.startSegment(HttpClientRequest("hello-world"))
+            (replier ? "hello").mapTo[String].onComplete { t =>
+              segment.get.finish()
+            }
+
+            "segment"
+          }
         }
     }
   }
@@ -166,7 +177,7 @@ class Replier extends Actor with ActorLogging {
       if (TraceRecorder.currentContext.isEmpty)
         log.warning("PROCESSING A MESSAGE WITHOUT CONTEXT")
 
-      log.info("Processing at the Replier, and self is: {}", self)
+      //log.info("Processing at the Replier, and self is: {}", self)
       sender ! anything
   }
 }
-- 
cgit v1.2.3