From 786f89f28eb36b446357fd7fdcc8651a65e3da86 Mon Sep 17 00:00:00 2001
From: wusheng <wu.sheng@foxmail.com>
Date: Thu, 2 Mar 2017 11:51:43 +0800
Subject: [PATCH] Create a new method about getTraceSegmentId

---
 .../a/eye/skywalking/api/context/ContextManager.java | 12 ++++++++++++
 .../a/eye/skywalking/api/context/TracerContext.java  |  7 +++++++
 2 files changed, 19 insertions(+)

diff --git a/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/ContextManager.java b/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/ContextManager.java
index 9fe501c2a..c9cbd8df8 100644
--- a/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/ContextManager.java
+++ b/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/ContextManager.java
@@ -53,6 +53,18 @@ public enum ContextManager implements TracerContextListener {
         get().extract(carrier);
     }
 
+    /**
+     * @return the {@link TraceSegment#traceSegmentId} if exist. Otherwise, "N/A".
+     */
+    public String getTraceSegmentId(){
+        TracerContext segment = CONTEXT.get();
+        if(segment == null){
+            return "N/A";
+        }else{
+            return segment.getTraceSegmentId();
+        }
+    }
+
     public Span createSpan(String operationName) {
         return get().createSpan(operationName);
     }
diff --git a/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/TracerContext.java b/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/TracerContext.java
index 31e6fe6de..59e220cf2 100644
--- a/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/TracerContext.java
+++ b/skywalking-sniffer/skywalking-api/src/main/java/com/a/eye/skywalking/api/context/TracerContext.java
@@ -88,6 +88,13 @@ public final class TracerContext {
         stopSpan(span, System.currentTimeMillis());
     }
 
+    /**
+     * @return the current trace segment id.
+     */
+    String getTraceSegmentId(){
+        return segment.getTraceSegmentId();
+    }
+
     public void stopSpan(Span span, Long endTime){
         Span lastSpan = peek();
         if (lastSpan == span) {