root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Provide first-class virtual thread option for common executors 

Closes gh-33807
This commit is contained in:
Juergen Hoeller 2024-10-28 22:14:41 +01:00
parent 3732c71168
commit 015beb0058
1 changed files with 27 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
spring-context/src/main/java/org/springframework/scheduling/concurrent/ExecutorConfigurationSupport.java
View File

@ -36,6 +36,8 @@ import org.springframework.context.ApplicationListener;
import org.springframework.context.Lifecycle;
import org.springframework.context.SmartLifecycle;
import org.springframework.context.event.ContextClosedEvent;
import org.springframework.core.task.SimpleAsyncTaskExecutor;
import org.springframework.core.task.VirtualThreadTaskExecutor;
import org.springframework.lang.Nullable;
/**
@ -75,6 +77,8 @@ public abstract class ExecutorConfigurationSupport extends CustomizableThreadFac
protected final Log logger = LogFactory.getLog(getClass());
private boolean virtualThreads = false;
private ThreadFactory threadFactory = this;
private boolean threadNamePrefixSet = false;
@ -104,6 +108,25 @@ public abstract class ExecutorConfigurationSupport extends CustomizableThreadFac
private volatile boolean lateShutdown;
/**
* Specify whether to use virtual threads instead of platform threads.
* This is off by default, setting up a traditional platform thread pool.
* <p>Set this flag to {@code true} on Java 21 or higher for a tightly
* managed thread pool setup with virtual threads. In contrast to
* {@link SimpleAsyncTaskExecutor}, this is integrated with Spring's
* lifecycle management for stopping and restarting execution threads,
* including an early stop signal for a graceful shutdown arrangement.
* <p>Specify either this or {@link #setThreadFactory}, not both.
* @since 6.2
* @see #setThreadFactory
* @see VirtualThreadTaskExecutor#getVirtualThreadFactory()
* @see SimpleAsyncTaskExecutor#setVirtualThreads
*/
public void setVirtualThreads(boolean virtualThreads) {
this.virtualThreads = virtualThreads;
this.threadFactory = this;
}
/**
* Set the ThreadFactory to use for the ExecutorService's thread pool.
* The default is the underlying ExecutorService's default thread factory.
@ -120,6 +143,7 @@ public abstract class ExecutorConfigurationSupport extends CustomizableThreadFac
*/
public void setThreadFactory(@Nullable ThreadFactory threadFactory) {
this.threadFactory = (threadFactory != null ? threadFactory : this);
this.virtualThreads = false;
}
@Override
@ -282,7 +306,9 @@ public abstract class ExecutorConfigurationSupport extends CustomizableThreadFac
if (!this.threadNamePrefixSet && this.beanName != null) {
setThreadNamePrefix(this.beanName + "-");
}
this.executor = initializeExecutor(this.threadFactory, this.rejectedExecutionHandler);
ThreadFactory factory = (this.virtualThreads ?
new VirtualThreadTaskExecutor(getThreadNamePrefix()).getVirtualThreadFactory() : this.threadFactory);
this.executor = initializeExecutor(factory, this.rejectedExecutionHandler);
this.lifecycleDelegate = new ExecutorLifecycleDelegate(this.executor);
}