博客

在使用springboot框架时，其注解使用大大提升了我们的编程效率。定时任务可以通过@Scheduled注解实现。

但是该方法存在一个缺点：多个定时任务使用的是同一个调度线程，所以定时任务是阻塞执行的，执行效率不高。比如20:00的任务需要执行2个小时，而恰好你在21:00需要执行另一个任务，那在21:00的任务会被阻塞到22:00进行，可能会导致执行结果不是我们想要的。

通过查看springboot底层代码：

//默认的调度器
if (this.taskScheduler == null) {
    this.localExecutor = Executors.newSingleThreadScheduledExecutor();
    this.taskScheduler = new ConcurrentTaskScheduler(this.localExecutor);
}

//可以看到默认指定的核心线程数为1，也就是单线程
public static ScheduledExecutorService newSingleThreadScheduledExecutor() {
    return new DelegatedScheduledExecutorService
        (new ScheduledThreadPoolExecutor(1));
}

解决方法：通过配置类直接重新配置定时任务的调度器，增加其线程数，实现定时任务的并发执行。

//定时任务配置类
@Configuration
public class ScheduledConfig implements SchedulingConfigurer {

    @Autowired
    private TaskScheduler myThreadPoolTaskScheduler;

    @Override
    public void configureTasks(ScheduledTaskRegistrar scheduledTaskRegistrar) {
        //直接暴力指定线程数
        //scheduledTaskRegistrar.setScheduler(Executors.newScheduledThreadPool(10));
        //自定义线程池
        scheduledTaskRegistrar.setTaskScheduler(myThreadPoolTaskScheduler);
    }

}

//自定义线程池
@Component
public class TaskScheduler {

    @Bean(name = "myThreadPoolTaskScheduler")
    public ThreadPoolTaskScheduler getMyThreadPoolTaskScheduler() {
        ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
        taskScheduler.setPoolSize(10);
        taskScheduler.setThreadNamePrefix("myScheduled-");
        //拒绝策略 callerRunsPolicy-由调用线程处理该任务
        taskScheduler.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        //调度器shutdown被调用时等待当前被调度的任务完成，用来设置线程池关闭的时候等待所有任务都完成再继续销毁其他的Bean。同时，还设置了setAwaitTerminationSeconds(60)，该方法用来设置线程池中任务的等待时间，如果超过这个时候还没有销毁就强制销毁，以确保应用最后能够被关闭，而不是阻塞住。
        taskScheduler.setWaitForTasksToCompleteOnShutdown(true);
        taskScheduler.setAwaitTerminationSeconds(60);
        return taskScheduler;
    }

}