From f2cad965fd81fa563c9935d83c6a05772882b6bd Mon Sep 17 00:00:00 2001
From: dataprince <dataprince@163.com>
Date: Sat, 6 Jan 2024 10:06:05 +0800
Subject: [PATCH] =?UTF-8?q?=E5=90=AF=E7=94=A8JAVA21=E8=99=9A=E6=8B=9F?=
 =?UTF-8?q?=E7=BA=BF=E7=A8=8B=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ruoyi-admin/src/main/resources/application.yml            | 4 ++++
 .../java/com/ruoyi/common/web/config/UndertowConfig.java  | 8 ++++++++
 2 files changed, 12 insertions(+)

diff --git a/ruoyi-admin/src/main/resources/application.yml b/ruoyi-admin/src/main/resources/application.yml
index 1b0454b..1cbccf3 100644
--- a/ruoyi-admin/src/main/resources/application.yml
+++ b/ruoyi-admin/src/main/resources/application.yml
@@ -68,6 +68,10 @@ user:
 spring:
   application:
     name: ${ruoyi.name}
+  threads:
+    # 启用JAVA21虚拟线程
+    virtual:
+      enabled: true
   # 资源信息
   messages:
     # 国际化资源文件路径
diff --git a/ruoyi-common/ruoyi-common-web/src/main/java/com/ruoyi/common/web/config/UndertowConfig.java b/ruoyi-common/ruoyi-common-web/src/main/java/com/ruoyi/common/web/config/UndertowConfig.java
index 6659c3c..eb79e08 100644
--- a/ruoyi-common/ruoyi-common-web/src/main/java/com/ruoyi/common/web/config/UndertowConfig.java
+++ b/ruoyi-common/ruoyi-common-web/src/main/java/com/ruoyi/common/web/config/UndertowConfig.java
@@ -1,11 +1,14 @@
 package com.ruoyi.common.web.config;
 
+import com.ruoyi.common.core.utils.SpringUtils;
 import io.undertow.server.DefaultByteBufferPool;
 import io.undertow.websockets.jsr.WebSocketDeploymentInfo;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
 import org.springframework.boot.web.embedded.undertow.UndertowServletWebServerFactory;
 import org.springframework.boot.web.server.WebServerFactoryCustomizer;
 
+import java.util.concurrent.Executors;
+
 /**
  * Undertow 自定义配置
  *
@@ -24,6 +27,11 @@ public class UndertowConfig implements WebServerFactoryCustomizer<UndertowServle
             WebSocketDeploymentInfo webSocketDeploymentInfo = new WebSocketDeploymentInfo();
             webSocketDeploymentInfo.setBuffers(new DefaultByteBufferPool(false, 512));
             deploymentInfo.addServletContextAttribute("io.undertow.websockets.jsr.WebSocketDeploymentInfo", webSocketDeploymentInfo);
+            // 启用JAVA21虚拟线程
+            if (SpringUtils.getProperty("spring.threads.virtual.enabled", Boolean.class, false)) {
+                deploymentInfo.setExecutor(Executors.newVirtualThreadPerTaskExecutor());
+            }
+
         });
     }