From aa4cd4da8bd5d218fafbe926f1e76dce5db44c94 Mon Sep 17 00:00:00 2001
From: Diego <diegolparra@gmail.com>
Date: Wed, 6 Aug 2014 23:13:33 -0300
Subject: ! core: use the inline variant of TraceRecorder.withTraceContext via
 macros

---
 .../main/scala/spray/can/client/ClientRequestInstrumentation.scala    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'kamon-spray/src/main/scala')

diff --git a/kamon-spray/src/main/scala/spray/can/client/ClientRequestInstrumentation.scala b/kamon-spray/src/main/scala/spray/can/client/ClientRequestInstrumentation.scala
index d787bda4..df1d2b59 100644
--- a/kamon-spray/src/main/scala/spray/can/client/ClientRequestInstrumentation.scala
+++ b/kamon-spray/src/main/scala/spray/can/client/ClientRequestInstrumentation.scala
@@ -63,7 +63,7 @@ class ClientRequestInstrumentation {
 
   @Around("copyingRequestContext(old)")
   def aroundCopyingRequestContext(pjp: ProceedingJoinPoint, old: SegmentCompletionHandleAware): Any = {
-    TraceRecorder.withTraceContext(old.traceContext) {
+    TraceRecorder.withInlineTraceContextReplacement(old.traceContext) {
       pjp.proceed()
     }
   }
@@ -75,7 +75,7 @@ class ClientRequestInstrumentation {
   def aroundDispatchToCommander(pjp: ProceedingJoinPoint, requestContext: SegmentCompletionHandleAware, message: Any) = {
     requestContext.traceContext match {
       case ctx @ Some(_) ⇒
-        TraceRecorder.withTraceContext(ctx) {
+        TraceRecorder.withInlineTraceContextReplacement(ctx) {
           if (message.isInstanceOf[HttpMessageEnd])
             requestContext.segmentCompletionHandle.map(_.finish(Map.empty))
 
-- 
cgit v1.2.3