From f965144ed4ec7d46d66e785586bb1a2a08f98161 Mon Sep 17 00:00:00 2001
From: michaeloffner <michaeloffner@users.noreply.github.com>
Date: Mon, 16 Dec 2024 16:54:26 +0100
Subject: [PATCH] LDEV-5203 - fix creating virtual thread

---
 .../java/lucee/runtime/thread/ThreadUtil.java | 56 +++++++++++++------
 loader/build.xml                              |  2 +-
 loader/pom.xml                                |  2 +-
 3 files changed, 41 insertions(+), 19 deletions(-)

diff --git a/core/src/main/java/lucee/runtime/thread/ThreadUtil.java b/core/src/main/java/lucee/runtime/thread/ThreadUtil.java
index 7797bde979..33440f6f85 100755
--- a/core/src/main/java/lucee/runtime/thread/ThreadUtil.java
+++ b/core/src/main/java/lucee/runtime/thread/ThreadUtil.java
@@ -33,8 +33,10 @@
 import javax.servlet.http.HttpSession;
 
 import lucee.aprint;
+import lucee.print;
 import lucee.commons.io.DevNullOutputStream;
 import lucee.commons.io.SystemUtil;
+import lucee.commons.io.log.LogUtil;
 import lucee.commons.io.res.Resource;
 import lucee.commons.lang.ExceptionUtil;
 import lucee.commons.lang.Pair;
@@ -54,8 +56,17 @@
 public class ThreadUtil {
 
 	private static final boolean ALLOW_FUTURE_THREADS = false;
-	private static final Class<?> THREAD_CLASS = Thread.class;
+	// private static final Class<?> THREAD_CLASS = Thread.class;
 	private static final Class<?> RUNNABLE_CLASS = Runnable.class;
+	private static Class<?> threadBuilderClass;
+	private static boolean virtualDisabled = false;
+
+	public static Class<?> getThreadBuilderClass() throws ClassNotFoundException {
+		if (threadBuilderClass == null) {
+			threadBuilderClass = Class.forName("java.lang.Thread$Builder$OfVirtual");
+		}
+		return threadBuilderClass;
+	}
 
 	// do not change, used in Redis extension
 	public static PageContextImpl clonePageContext(PageContext pc, OutputStream os, boolean stateless, boolean register2Thread, boolean register2RunningThreads) {
@@ -205,20 +216,20 @@ public static Thread getThread(Runnable task) {
 
 	public static Thread getThread(Runnable task, boolean allowVirtual) {
 		if (allowVirtual && SystemUtil.JAVA_VERSION >= SystemUtil.JAVA_VERSION_19) {
+
 			try {
-				// return Thread.ofVirtual().unstarted(task);
+				// Get Thread.ofVirtual()
 				MethodHandles.Lookup lookup = MethodHandles.lookup();
-				MethodHandle ofVirtualHandle = lookup.findStatic(THREAD_CLASS, "ofVirtual", MethodType.methodType(THREAD_CLASS.getDeclaredClasses()[0]));
-				Object builder = ofVirtualHandle.invoke();
-				MethodHandle unstartedHandle = lookup.findVirtual(builder.getClass(), "unstarted", MethodType.methodType(THREAD_CLASS, RUNNABLE_CLASS));
-				return (Thread) unstartedHandle.invoke(builder, task);
+				MethodHandle ofVirtualHandle = lookup.findStatic(Thread.class, "ofVirtual", MethodType.methodType(getThreadBuilderClass()));
+				MethodHandle unstartedHandle = lookup.findVirtual(Class.forName("java.lang.Thread$Builder"), "unstarted", MethodType.methodType(Thread.class, Runnable.class));
+				return (Thread) unstartedHandle.bindTo(ofVirtualHandle.invoke()).invoke(task);
 			}
 			catch (Throwable e) {
 				ExceptionUtil.rethrowIfNecessary(e);
+				LogUtil.log("threading", e);
 			}
 		}
 		return new Thread(task);
-
 	}
 
 	public static ExecutorService createExecutorService(int maxThreads) {
@@ -226,21 +237,30 @@ public static ExecutorService createExecutorService(int maxThreads) {
 	}
 
 	public static ExecutorService createExecutorService(int maxThreads, boolean allowVirtual) {
-		if (allowVirtual && SystemUtil.JAVA_VERSION >= SystemUtil.JAVA_VERSION_19) {
-			// FUTURE use newVirtualThreadPerTaskExecutor natively
-			try {
-				MethodHandles.Lookup lookup = MethodHandles.lookup();
-				MethodType methodType = MethodType.methodType(ExecutorService.class);
-				MethodHandle methodHandle = lookup.findStatic(Executors.class, "newVirtualThreadPerTaskExecutor", methodType);
-				return (ExecutorService) methodHandle.invoke();
-			}
-			catch (Throwable e) {
-				ExceptionUtil.rethrowIfNecessary(e);
+		if (!virtualDisabled) {
+			if (allowVirtual && SystemUtil.JAVA_VERSION >= SystemUtil.JAVA_VERSION_19) {
+				// FUTURE use newVirtualThreadPerTaskExecutor natively
+				try {
+					MethodHandles.Lookup lookup = MethodHandles.lookup();
+					MethodType methodType = MethodType.methodType(ExecutorService.class);
+					MethodHandle methodHandle = lookup.findStatic(Executors.class, "newVirtualThreadPerTaskExecutor", methodType);
+					return (ExecutorService) methodHandle.invoke();
+				}
+				catch (Throwable e) {
+					virtualDisabled = true;
+					ExceptionUtil.rethrowIfNecessary(e);
+					LogUtil.log("threading", e);
+				}
 			}
 		}
 		return Executors.newFixedThreadPool(maxThreads);
 	}
 
+	public static void main(String[] args) {
+		ExecutorService t = createExecutorService();
+		print.e(t);
+	}
+
 	public static ExecutorService createExecutorService() {
 		if (SystemUtil.JAVA_VERSION >= SystemUtil.JAVA_VERSION_19) {
 			// FUTURE use newVirtualThreadPerTaskExecutor natively
@@ -251,7 +271,9 @@ public static ExecutorService createExecutorService() {
 				return (ExecutorService) methodHandle.invoke();
 			}
 			catch (Throwable e) {
+				print.e(e);
 				ExceptionUtil.rethrowIfNecessary(e);
+				LogUtil.log("threading", e);
 			}
 		}
 		return Executors.newSingleThreadExecutor();
diff --git a/loader/build.xml b/loader/build.xml
index 35a3ff2da1..f43a7b276e 100644
--- a/loader/build.xml
+++ b/loader/build.xml
@@ -2,7 +2,7 @@
 <project default="core" basedir="." name="Lucee" 
   xmlns:resolver="antlib:org.apache.maven.resolver.ant">
 
-  <property name="version" value="6.2.0.217-SNAPSHOT"/>
+  <property name="version" value="6.2.0.218-SNAPSHOT"/>
 
   <taskdef uri="antlib:org.apache.maven.resolver.ant" resource="org/apache/maven/resolver/ant/antlib.xml">
     <classpath>
diff --git a/loader/pom.xml b/loader/pom.xml
index 35af946d7c..ec9daa3721 100644
--- a/loader/pom.xml
+++ b/loader/pom.xml
@@ -3,7 +3,7 @@
 
   <groupId>org.lucee</groupId>
   <artifactId>lucee</artifactId>
-  <version>6.2.0.217-SNAPSHOT</version>
+  <version>6.2.0.218-SNAPSHOT</version>
   <packaging>jar</packaging>
 
   <name>Lucee Loader Build</name>