fix: CallDepth thread safety (#559)

Author: YongwuHe

arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/ctx/ArexThreadLocal.java

@@ -1,5 +1,7 @@
 package io.arex.agent.bootstrap.ctx;
 
+import io.arex.agent.bootstrap.internal.CallDepth;
+
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.Map;
@@ -45,7 +47,11 @@ public void superRemove() {
     }
 
     public T copyValue() {
-        return get();
+        T value = get();
+        if (value instanceof CallDepth) {
+            return (T) ((CallDepth) value).copy();
+        }
+        return value;
     }
 
     private static final ThreadLocal<WeakHashMap<ArexThreadLocal<Object>, ?>> holder =
@@ -76,7 +82,7 @@ public static class Transmitter {
 
         public static Object capture() {
             HashMap<ArexThreadLocal<Object>, Object> values = captureValues();
-            return values == null ? null : new Snapshot(captureValues());
+            return values == null ? null : new Snapshot(values);
         }
 
         private static HashMap<ArexThreadLocal<Object>, Object> captureValues() {

arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/internal/CallDepth.java

@@ -39,4 +39,10 @@ public int getAndIncrement() {
     public int decrementAndGet() {
         return --this.depth;
     }
+
+    public CallDepth copy() {
+        CallDepth copy = new CallDepth();
+        copy.depth = this.depth;
+        return copy;
+    }
 }

arex-instrumentation/internal/arex-executors/src/main/java/io/arex/inst/executors/ForkJoinTaskConstructorInstrumentation.java

@@ -1,5 +1,6 @@
 package io.arex.inst.executors;
 
+import io.arex.agent.bootstrap.TraceContextManager;
 import io.arex.agent.bootstrap.ctx.ArexThreadLocal;
 import io.arex.agent.bootstrap.internal.Cache;
 import io.arex.inst.extension.MethodInstrumentation;
@@ -31,6 +32,9 @@ public List<MethodInstrumentation> methodAdvices() {
     public static class ConstructorAdvice {
         @Advice.OnMethodExit(suppress = Throwable.class)
         public static void onExit(@Advice.This Object task) {
+            if (TraceContextManager.get() == null) {
+                return;
+            }
             final Object captured = ArexThreadLocal.Transmitter.capture();
             if (captured != null) {
                 Cache.CAPTURED_CACHE.put(task, captured);