[ISSUE sofastack#1122] support custom asyncThreadPool

loongs-zhang · Apr 16, 2022 · 5fcb099 · 5fcb099
1 parent 2f23db9
commit 5fcb099
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/api/src/main/java/com/alipay/sofa/rpc/context/AsyncRuntime.java b/core/api/src/main/java/com/alipay/sofa/rpc/context/AsyncRuntime.java
@@ -100,4 +100,13 @@ public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
         }
         return asyncThreadPool;
     }
+
+    /**
+     * 设置callback用的线程池
+     *
+     * @param threadPoolExecutor callback用的线程池
+     */
+    public static void setAsyncThreadPool(ThreadPoolExecutor threadPoolExecutor) {
+        AsyncRuntime.asyncThreadPool = threadPoolExecutor;
+    }
 }