Support Virtual Threads in the WorkManager

- Fixes #147
quarkiverse · Jan 30, 2025 · daa5e7b · daa5e7b
1 parent 316a236
commit daa5e7b
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/deployment/pom.xml b/deployment/pom.xml
@@ -25,6 +25,10 @@
       <groupId>io.quarkus</groupId>
       <artifactId>quarkus-smallrye-context-propagation-deployment</artifactId>
     </dependency>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-virtual-threads-deployment</artifactId>
+    </dependency>
     <dependency>
       <groupId>io.quarkiverse.ironjacamar</groupId>
       <artifactId>quarkus-ironjacamar</artifactId>

diff --git a/runtime/pom.xml b/runtime/pom.xml
@@ -26,6 +26,10 @@
       <groupId>io.quarkus</groupId>
       <artifactId>quarkus-smallrye-context-propagation</artifactId>
     </dependency>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-virtual-threads</artifactId>
+    </dependency>
     <dependency>
       <groupId>jakarta.resource</groupId>
       <artifactId>jakarta.resource-api</artifactId>

diff --git a/runtime/src/main/java/io/quarkiverse/ironjacamar/runtime/IronJacamarRecorder.java b/runtime/src/main/java/io/quarkiverse/ironjacamar/runtime/IronJacamarRecorder.java
@@ -6,14 +6,14 @@
 import java.time.Duration;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.ExecutorService;
 import java.util.concurrent.TimeUnit;
 import java.util.function.Function;
 import java.util.function.Supplier;
 
 import jakarta.enterprise.inject.spi.DeploymentException;
 import jakarta.resource.ResourceException;
 
-import org.eclipse.microprofile.context.ManagedExecutor;
 import org.jboss.jca.core.api.bootstrap.CloneableBootstrapContext;
 import org.jboss.jca.core.api.connectionmanager.ccm.CachedConnectionManager;
 import org.jboss.jca.core.bootstrapcontext.BaseCloneableBootstrapContext;
@@ -33,6 +33,7 @@
 import io.quarkus.narayana.jta.runtime.TransactionManagerConfiguration;
 import io.quarkus.runtime.RuntimeValue;
 import io.quarkus.runtime.annotations.Recorder;
+import io.quarkus.virtual.threads.VirtualThreadsRecorder;
 import io.smallrye.common.annotation.Identifier;
 import io.vertx.core.DeploymentOptions;
 import io.vertx.core.Future;
@@ -131,7 +132,7 @@ public void initDefaultBootstrapContext(BeanContainer beanContainer) {
         TransactionIntegration transactionIntegration = beanContainer.beanInstance(TransactionIntegration.class);
         SecurityIntegration securityIntegration = beanContainer.beanInstance(QuarkusSecurityIntegration.class);
         BaseCloneableBootstrapContext bootstrapContext = new BaseCloneableBootstrapContext();
-        ManagedExecutor executorService = beanContainer.beanInstance(ManagedExecutor.class);
+        ExecutorService executorService = VirtualThreadsRecorder.getCurrent();
 
         // Create WorkManagerImpl
         WorkManagerImpl workManager = new WorkManagerImpl();