From 23a731bbba037e798a81450f7f59099c860511ca Mon Sep 17 00:00:00 2001
From: Woonduk Kang <wd.kang@navercorp.com>
Date: Tue, 28 Jul 2015 11:20:28 +0900
Subject: [PATCH] #714 trace api refactoring

---
 .../pinpoint/profiler/context/DefaultTrace.java      | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultTrace.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultTrace.java
index be72cf2bd..f7656b8de 100644
--- a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultTrace.java
+++ b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultTrace.java
@@ -111,6 +111,12 @@ public final class DefaultTrace implements Trace {
         return span;
     }
 
+    private SpanEventRecorder wrappedSpanEventRecorder(SpanEvent spanEvent) {
+        final WrappedSpanEventRecorder spanEventRecorder = this.spanEventRecorder;
+        spanEventRecorder.setWrapped(spanEvent);
+        return spanEventRecorder;
+    }
+
     public void setStorage(Storage storage) {
         this.storage = storage;
     }
@@ -136,8 +142,7 @@ public final class DefaultTrace implements Trace {
             callStack.push(spanEvent);
         }
 
-        spanEventRecorder.setWrapped(spanEvent);
-        return spanEventRecorder;
+        return wrappedSpanEventRecorder(spanEvent);
     }
 
     @Override
@@ -301,8 +306,7 @@ public final class DefaultTrace implements Trace {
             spanEvent = new SpanEvent(spanRecorder.getSpan());
         }
 
-        spanEventRecorder.setWrapped(spanEvent);
-        return spanEventRecorder;
+        return wrappedSpanEventRecorder(spanEvent);
     }
 
     @Override