自定义 Spring 定时任务使用的线程池

1 背景

在 Spring 项目中，如果不手动配置定时任务线程池，那么所有的 @Scheduled 定时任务只能依靠同一个线程执行，因为默认的线程池的线程数为1。这意味着如果其中一个任务卡住了，其他所有定时任务都会被阻塞。为了提高并发能力，我们需要自定义Spring 定时任务使用的线程池。

2 方案

自定义 Spring 定时任务使用的线程池的方案案例如下所示：

java 复制代码

@Configuration
@EnableScheduling
public class SchedulingConfig implements SchedulingConfigurer {

    @Override
    public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
        ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
        scheduler.setPoolSize(Runtime.getRuntime().availableProcessors() * 2); // 设置最大并行任务数
        scheduler.setThreadNamePrefix("custom-scheduled-task-");
        scheduler.setWaitForTasksToCompleteOnShutdown(true); // 关闭 Spring 容器时等待任务完成
        scheduler.setAwaitTerminationSeconds(60); // 等待任务结束的最长时间
        scheduler.initialize(); // 必须初始化

        taskRegistrar.setTaskScheduler(scheduler);
    }
}

3 彩蛋

（1）Spring 定时任务使用 Spring 内置线程池的危害如下所示：

java 复制代码

1. 线程竞争和阻塞
串行执行风险：默认线程池线程数量有限，多个定时任务将排队执行
阻塞影响：耗时任务会阻塞其他任务的执行时机，影响整体调度准确性

2. 性能瓶颈
吞吐量限制：有限的线程数量无法充分利用系统资源
执行延迟：任务可能因线程不足而延迟执行，错过预期执行时间点

3. 资源争用
CPU密集型任务：多个CPU密集型任务在同一线程池中运行会导致资源争用
I/O密集型任务：阻塞性I/O操作会占用线程，影响其他任务执行

4. 异常传播
单点故障：一个任务的异常可能影响整个线程池的稳定性
任务丢失：严重异常可能导致后续任务无法正常执行

（2）Spring 定时任务依赖的线程池为 ThreadPoolTaskScheduler，其默认的线程数为1，源码如下所示。

定时任务配置入口（spring-boot-autoconfigure 包下）

java 复制代码

@ConditionalOnClass(ThreadPoolTaskScheduler.class)
@Configuration(proxyBeanMethods = false)
@EnableConfigurationProperties(TaskSchedulingProperties.class)
public class TaskSchedulingAutoConfiguration {

    @Bean
    @ConditionalOnMissingBean(SchedulingConfigurer.class)
    @ConditionalOnMissingBean(TaskScheduler.class)
    public ThreadPoolTaskScheduler taskScheduler(TaskSchedulerBuilder builder) {
        // 使用 builder 创建 ThreadPoolTaskScheduler
        return builder.build();
    }
}

定时任务类

java 复制代码

public class ThreadPoolTaskScheduler extends ExecutorConfigurationSupport
		implements AsyncListenableTaskExecutor, SchedulingTaskExecutor, TaskScheduler {

	private volatile int poolSize = 1;

	private volatile boolean removeOnCancelPolicy = false;

	@Nullable
	private volatile ErrorHandler errorHandler;

	@Nullable
	private ScheduledExecutorService scheduledExecutor;

	......

	// 创建线程池
	@Override
	protected ExecutorService initializeExecutor(
			ThreadFactory threadFactory, RejectedExecutionHandler rejectedExecutionHandler) {

		this.scheduledExecutor = createExecutor(this.poolSize, threadFactory, rejectedExecutionHandler);

		if (this.removeOnCancelPolicy) {
			if (this.scheduledExecutor instanceof ScheduledThreadPoolExecutor) {
				((ScheduledThreadPoolExecutor) this.scheduledExecutor).setRemoveOnCancelPolicy(true);
			}
			else {
				logger.debug("Could not apply remove-on-cancel policy - not a ScheduledThreadPoolExecutor");
			}
		}

		return this.scheduledExecutor;
	}
	
	......
}