Merge pull request #3 from Nepxion/zifeihan

Avoid calling in the same thread

Author: HaojunRen

discovery-agent-starter/src/main/java/com/nepxion/discovery/agent/async/AsyncContext.java

@@ -12,11 +12,21 @@
 public class AsyncContext {
     private Object[] objects;
 
+    /**
+     * Record the original thread and compare it with the thread that called runnable/callable.
+     */
+    private Thread originThread;
+
     public AsyncContext(Object[] objects) {
         this.objects = objects;
+        this.originThread = Thread.currentThread();
     }
 
     public Object[] getObjects() {
         return objects;
     }
+
+    public Thread getOriginThread() {
+        return originThread;
+    }
 }

discovery-agent-starter/src/main/java/com/nepxion/discovery/agent/plugin/spring/async/WrapCallable.java

@@ -27,6 +27,13 @@ public WrapCallable(Callable<T> callable) {
 
     @Override
     public T call() throws Exception {
+
+        //Avoid calling in the same thread.
+        if (asyncContext.getOriginThread().equals(Thread.currentThread())) {
+            return callable.call();
+        }
+
+        //Call the original method and copy some objects held by ThreadLocal.
         Object[] objects = asyncContext.getObjects();
         ThreadLocalCopier.before(objects);
         try {

discovery-agent-starter/src/main/java/com/nepxion/discovery/agent/plugin/thread/interceptor/ThreadCallInterceptor.java

@@ -21,13 +21,26 @@ public static void before(Object object) {
             if (null == asyncContext) {
                 return;
             }
+            //Avoid calling in the same thread.
+            if (asyncContext.getOriginThread().equals(Thread.currentThread())) {
+                return;
+            }
             Object[] objects = asyncContext.getObjects();
             ThreadLocalCopier.before(objects);
         }
     }
 
     public static void after(Object object) {
         if (object instanceof AsyncContextAccessor) {
+            AsyncContextAccessor asyncContextAccessor = (AsyncContextAccessor) object;
+            AsyncContext asyncContext = asyncContextAccessor.getAsyncContext();
+            if (null == asyncContext) {
+                return;
+            }
+            //Avoid calling in the same thread.
+            if (asyncContext.getOriginThread().equals(Thread.currentThread())) {
+                return;
+            }
             ThreadLocalCopier.after();
         }
     }