SpringBoot 自定义Executor

coding-screen

SpringBoot内置的默认的Executor来执行异步任务,只需要增加@EnableAsync注解,并在需要异步执行的方法上增加@Async注解,非常方便。同时,Spring还提供了内置的调度器,通过增加@EnableScheduling注解启用,在调度任务的执行方法上使用@Scheduled注解定义调度参数。

如果需要同时使用这两个特性,还需要自定义Executor的话,仅使用AsyncConfigurer接口定义Executor类型的Bean会导致Bean声明重复,进而无法实例化Bean。

可以使用如下方式实现定制,并且让异步任务和调度任务共享同一个线程池。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
// AsyncExecutorConfiguration.java

@Configuration
@EnableAsync
@Slf4j
public class AsyncExecutorConfiguration implements AsyncConfigurer {
@Bean
@Override
public Executor getAsyncExecutor() {
ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
scheduler.setPoolSize(8);
scheduler.setThreadNamePrefix("prefix");
return scheduler;
}

@Override
public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
return (ex, method, params) -> {};
}
}

// SchedulingConfiguration.java

@Configuration
@EnableScheduling
public class SchedulingConfiguration implements SchedulingConfigurer {

@Autowired
@Qualifier("getAsyncExecutor")
public Executor mExecutor;

@Override
public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {
taskRegistrar.setScheduler(mExecutor);
}
}

这里创建了两个文件分别提供Async和Scheduled用到的Executor,但是使用的都是同一个线程池。