fix abortProcess concurrency problem. #2957

Author: hengyunabc

core/src/main/java/com/taobao/arthas/core/command/monitor200/AbstractTraceAdviceListener.java

@@ -9,6 +9,8 @@
 import com.taobao.arthas.core.util.LogUtil;
 import com.taobao.arthas.core.util.ThreadLocalWatch;
 
+import java.util.concurrent.atomic.AtomicBoolean;
+
 /**
  * @author ralf0131 2017-01-06 16:02.
  */
@@ -17,6 +19,7 @@ public class AbstractTraceAdviceListener extends AdviceListenerAdapter {
     protected final ThreadLocalWatch threadLocalWatch = new ThreadLocalWatch();
     protected TraceCommand command;
     protected CommandProcess process;
+    private final AtomicBoolean processAborted = new AtomicBoolean(false);
 
     protected final ThreadLocal<TraceEntity> threadBoundEntity = new ThreadLocal<TraceEntity>();
 
@@ -99,7 +102,6 @@ private void finishing(ClassLoader loader, Advice advice) {
 
                     // 是否到达数量限制
                     if (isLimitExceeded(command.getNumberOfLimit(), process.times().get())) {
-                        // TODO: concurrency issue to abort process
                         abortProcess(process, command.getNumberOfLimit());
                     }
                 }
@@ -112,4 +114,12 @@ private void finishing(ClassLoader loader, Advice advice) {
             }
         }
     }
+
+    @Override
+    protected void abortProcess(CommandProcess process, int limit) {
+        // Only proceed if this thread is the first one to set the flag to true
+        if (processAborted.compareAndSet(false, true)) {
+            super.abortProcess(process, limit);
+        }
+    }
 }