Fix the profiling segment count not correct (#491)

Author: mrproliu

apm-sniffer/apm-agent-core/src/main/java/org/apache/skywalking/apm/agent/core/profile/ProfileStatusContext.java

@@ -31,11 +31,13 @@
 public class ProfileStatusContext {
 
     private volatile ProfileStatus status;
+    private volatile boolean fromFirstSegment;
     private volatile long firstSegmentCreateTime;
     private volatile AtomicInteger subThreadProfilingCount;
 
     private ProfileStatusContext(ProfileStatus status, long firstSegmentCreateTime, AtomicInteger subThreadProfilingCount) {
         this.status = status;
+        this.fromFirstSegment = true;
         this.firstSegmentCreateTime = firstSegmentCreateTime;
         this.subThreadProfilingCount = subThreadProfilingCount;
     }
@@ -62,6 +64,10 @@ public long firstSegmentCreateTime() {
         return this.firstSegmentCreateTime;
     }
 
+    public boolean isFromFirstSegment() {
+        return fromFirstSegment;
+    }
+
     /**
      * The profile monitoring is watching, wait for some profile conditions.
      */
@@ -83,6 +89,7 @@ public ProfileStatusContext clone() {
      */
     public boolean continued(ContextSnapshot snapshot) {
         this.status = snapshot.getProfileStatusContext().get();
+        this.fromFirstSegment = false;
         this.firstSegmentCreateTime = snapshot.getProfileStatusContext().firstSegmentCreateTime();
         this.subThreadProfilingCount = snapshot.getProfileStatusContext().subThreadProfilingCount;
         return this.isBeingWatched() &&

apm-sniffer/apm-agent-core/src/main/java/org/apache/skywalking/apm/agent/core/profile/ProfileTaskExecutionContext.java

@@ -131,7 +131,7 @@ public void profilingRecheck(TracingContext tracingContext, String traceSegmentI
 
         // update profiling status
         tracingContext.profileStatus()
-                      .updateStatus(attemptProfiling(tracingContext, traceSegmentId, firstSpanOPName));
+            .updateStatus(attemptProfiling(tracingContext, traceSegmentId, firstSpanOPName));
     }
 
     /**
@@ -147,7 +147,9 @@ public void stopTracingProfile(TracingContext tracingContext) {
 
                 // setting stop running
                 currentProfiler.stopProfiling();
-                currentEndpointProfilingCount.addAndGet(-1);
+                if (currentProfiler.profilingStatus().isFromFirstSegment()) {
+                    currentEndpointProfilingCount.addAndGet(-1);
+                }
                 break;
             }
         }