优雅关闭线程池

在Spring框架中，优雅地关闭线程池是非常重要的，尤其是在分布式系统和微服务架构中，因为不正确的关闭可能会导致资源泄露、线程挂起等问题。Spring提供了几种机制来帮助你优雅地关闭线程池：

1. 使用ExecutorService和TaskExecutor

如果你在Spring中使用了ExecutorService（如ThreadPoolExecutor），你可以将其声明为一个Bean，并利用Spring的@PreDestroy注解来确保在应用上下文关闭时优雅地关闭线程池。

例如，你可以这样定义一个线程池Bean：

@Configuration
public class ExecutorConfig {

    @Bean(destroyMethod = "shutdown")
    public ExecutorService taskExecutor() {
        ThreadPoolExecutor executor = new ThreadPoolExecutor(
            5, // 核心线程数
            10, // 最大线程数
            60L, // 空闲线程存活时间
            TimeUnit.SECONDS,
            new ArrayBlockingQueue<>(100) // 工作队列
        );
        return executor;
    }
}

这里的关键点是destroyMethod = "shutdown"属性，它告诉Spring在关闭应用上下文时调用shutdown方法。但是，shutdown方法不会等待正在执行的任务完成，而是仅仅阻止新任务的提交。为了确保线程池完全关闭，你可能需要使用shutdownNow或awaitTermination方法。然而，由于awaitTermination方法可能会阻塞调用者直到线程池完全关闭，因此在Spring中直接使用可能不是最佳选择。

2. 使用@PostConstruct和@PreDestroy

另一种方法是创建一个自定义的线程池管理Bean，使用@PostConstruct和@PreDestroy注解：

@Service
public class CustomThreadPool {

    private ExecutorService executor;

    @PostConstruct
    public void init() {
        executor = Executors.newFixedThreadPool(5);
    }

    @PreDestroy
    public void shutdown() {
        executor.shutdown();
        try {
            if (!executor.awaitTermination(60, TimeUnit.SECONDS)) {
                executor.shutdownNow();
                if (!executor.awaitTermination(60, TimeUnit.SECONDS))
                    System.err.println("Pool did not terminate");
            }
        } catch (InterruptedException ie) {
            executor.shutdownNow();
            Thread.currentThread().interrupt();
        }
    }
}

在这个例子中，@PostConstruct注解的方法在Bean初始化之后调用，而@PreDestroy注解的方法在Bean销毁之前调用，确保了线程池的正确关闭。

3. Spring Boot Actuator

如果你使用的是Spring Boot，你可以考虑使用Actuator模块的/shutdown端点。虽然这不是直接与线程池相关的，但它提供了一个统一的方式来关闭整个应用，包括所有运行中的线程池。要使用这个功能，你需要在你的应用中添加Spring Boot Actuator依赖，并启用management.endpoint.shutdown.enabled=true配置。

以上三种方法可以帮助你优雅地关闭Spring中的线程池，避免资源泄露和潜在的性能问题。