Fix threadpool monitor when using virtual thread (#1344)

Signed-off-by: JermaineHua <crazyhzm@apache.org>
sofastack · Sep 25, 2024 · d285ec4 · d285ec4
1 parent 5ca7e69
commit d285ec4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/...rpc-sofa-boot/src/main/java/com/alipay/sofa/rpc/boot/container/ServerConfigContainer.java b/...rpc-sofa-boot/src/main/java/com/alipay/sofa/rpc/boot/container/ServerConfigContainer.java
@@ -40,6 +40,7 @@
 import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.Executor;
 import java.util.concurrent.ThreadPoolExecutor;
 
 /**
@@ -275,9 +276,14 @@ private void startCustomThreadPoolMonitor() {
                 } else {
                     customThreadPoolMonitor.setPoolName(pool.getThreadPoolName());
                 }
-                customThreadPoolMonitor.setThreadPoolExecutor(pool.getExecutor());
-                customThreadPoolMonitor.start();
-                poolNames.add(pool.getThreadPoolName());
+
+                Executor tmp = pool.getUserExecutor();
+                if (tmp instanceof ThreadPoolExecutor) {
+                    customThreadPoolMonitor.setThreadPoolExecutor((ThreadPoolExecutor) pool
+                        .getUserExecutor());
+                    customThreadPoolMonitor.start();
+                    poolNames.add(pool.getThreadPoolName());
+                }
             }
         }
     }