如何确保AsyncTaskExecutor在应用停止前优雅清理虚拟线程？

你自定义的VirtualThreadTaskExecutor使用虚拟线程执行任务，由于虚拟线程默认是守护线程，在应用优雅关闭阶段（Spring容器销毁但JVM未退出），这些线程可能仍会后台运行。以下是几种可靠的清理方案：

方案一：追踪虚拟线程并在容器销毁时主动中断/等待 #

通过在自定义Executor中维护活跃虚拟线程的集合，结合Spring的生命周期回调（如@PreDestroy），在应用停止时主动发送中断信号并等待线程退出。

修改后的VirtualThreadTaskExecutor实现 #

import org.springframework.core.task.AsyncTaskExecutor;
import javax.annotation.PreDestroy;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

public class VirtualThreadTaskExecutor implements AsyncTaskExecutor {
    private final ThreadFactory threadFactory;
    // 用线程安全集合追踪活跃虚拟线程
    private final Set<Thread> activeVirtualThreads = ConcurrentHashMap.newKeySet();

    public VirtualThreadTaskExecutor() {
        this.threadFactory = Thread.ofVirtual().name("my-app-virtual-thread-", 0).factory();
    }

    @Override
    public void execute(Runnable task) {
        var wrapped = MyTaskWrapper.wrap(task);
        Thread virtualThread = threadFactory.newThread(wrapped);
        activeVirtualThreads.add(virtualThread);
        
        // 线程结束后自动从集合移除
        virtualThread.setUncaughtExceptionHandler((t, e) -> activeVirtualThreads.remove(t));
        // 线程正常结束也移除
        virtualThread.onTermination(() -> activeVirtualThreads.remove(virtualThread));
        
        virtualThread.start();
    }

    @PreDestroy
    public void cleanupVirtualThreads() throws InterruptedException {
        long timeoutMillis = 5000; // 自定义超时时间，比如5秒
        long endWaitTime = System.currentTimeMillis() + timeoutMillis;

        // 先等待线程自行完成
        while (!activeVirtualThreads.isEmpty() && System.currentTimeMillis() < endWaitTime) {
            Thread.sleep(100);
        }

        // 对剩余活跃线程发送中断信号
        for (Thread vt : activeVirtualThreads) {
            if (vt.isAlive()) {
                vt.interrupt();
            }
        }

        // 额外等待一段时间让线程处理中断并清理
        Thread.sleep(500);
        activeVirtualThreads.clear();
    }

    // 实现AsyncTaskExecutor的其他方法（如submit等）...
}

任务包装器优化（确保响应中断） #

虚拟线程的中断是协作式的，需要任务正确处理InterruptedException才能优雅退出。可以在MyTaskWrapper中统一处理：

public class MyTaskWrapper {
    public static Runnable wrap(Runnable task) {
        return () -> {
            try {
                task.run();
            } catch (InterruptedException e) {
                // 记录中断日志，清理资源
                Thread.currentThread().interrupt(); // 保留中断状态
            } catch (Exception e) {
                // 处理业务异常
            } finally {
                // 执行通用清理逻辑
            }
        };
    }
}

方案二：复用Spring的SimpleAsyncTaskExecutor（适配虚拟线程） #

你提到的SimpleAsyncTaskExecutor的setTaskTerminationTimeout方法可以在close()时等待任务超时终止，只需将其线程工厂替换为虚拟线程工厂即可：

@Bean
public AsyncTaskExecutor virtualThreadExecutor() {
    SimpleAsyncTaskExecutor executor = new SimpleAsyncTaskExecutor();
    executor.setThreadFactory(Thread.ofVirtual().name("my-app-virtual-thread-", 0).factory());
    executor.setTaskTerminationTimeout(5000); // 设置5秒超时
    executor.setWaitForTasksToCompleteOnShutdown(true); // 关闭时等待任务完成
    return executor;
}

这种方式无需自定义Executor，直接利用Spring原生组件的生命周期管理能力，适合快速实现需求。

关键注意事项 #

• 虚拟线程作为守护线程，JVM退出时会被强制终止，但**优雅关闭场景下（如Spring容器销毁）**需要主动触发清理，避免任务在后台继续执行。
• 中断信号仅会触发阻塞方法抛出InterruptedException，如果任务中没有阻塞逻辑，需要手动检查线程中断状态（Thread.currentThread().isInterrupted()）来退出循环或终止任务。
• 追踪虚拟线程时需使用线程安全集合，避免并发修改问题。

内容的提问来源于stack exchange，提问作者Govinda Sakhare