Преглед изворни кода

add 增加 StringUtils.isVirtual 方法
update 优化 undertow 虚拟线程实现

疯狂的狮子Li пре 2 година
родитељ
комит
739fa9f120

+ 8 - 0
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/utils/SpringUtils.java

@@ -3,13 +3,17 @@ package org.dromara.common.core.utils;
 import cn.hutool.extra.spring.SpringUtil;
 import org.springframework.aop.framework.AopContext;
 import org.springframework.beans.factory.NoSuchBeanDefinitionException;
+import org.springframework.boot.autoconfigure.thread.Threading;
 import org.springframework.context.ApplicationContext;
+import org.springframework.core.env.Environment;
+import org.springframework.stereotype.Component;
 
 /**
  * spring工具类
  *
  * @author Lion Li
  */
+@Component
 public final class SpringUtils extends SpringUtil {
 
     /**
@@ -57,4 +61,8 @@ public final class SpringUtils extends SpringUtil {
         return getApplicationContext();
     }
 
+    public static boolean isVirtual() {
+        return Threading.VIRTUAL.isActive(getBean(Environment.class));
+    }
+
 }

+ 4 - 2
ruoyi-common/ruoyi-common-web/src/main/java/org/dromara/common/web/config/UndertowConfig.java

@@ -25,8 +25,10 @@ public class UndertowConfig implements WebServerFactoryCustomizer<UndertowServle
             webSocketDeploymentInfo.setBuffers(new DefaultByteBufferPool(false, 512));
             deploymentInfo.addServletContextAttribute("io.undertow.websockets.jsr.WebSocketDeploymentInfo", webSocketDeploymentInfo);
             // 使用虚拟线程 解除下方注释 仅限jdk21
-//            if (SpringUtils.getProperty("spring.threads.virtual.enabled", Boolean.class, false)) {
-//                deploymentInfo.setExecutor(Executors.newVirtualThreadPerTaskExecutor());
+//            if (SpringUtils.isVirtual()) {
+//                ExecutorService executor = Executors.newVirtualThreadPerTaskExecutor();
+//                deploymentInfo.setExecutor(executor);
+//                deploymentInfo.setAsyncExecutor(executor);
 //            }
         });
     }