spring @Async 配置异步线程池
一、认识@Async
在Spring Boot中,可以使用@EnableAsync
注解来开启异步执行功能。
在配置类或启动类上加上@EnableAsync
注解,开启异步执行功能:
@Configuration
@EnableAsync
public class AppConfig {
// ...
}
之后就可以在方法中使用@Async
注解来指定该方法应该在异步线程池中执行。例如:
@Service
public class MyService {
@Async
public void doSomethingAsync() {
// 异步执行的代码
}
}
在上面的示例中,doSomethingAsync()
方法被标记为异步方法,并且将在异步线程池中执行。
而@async
的默认线程池是SimpleAsyncTaskExecutor
。它是一个简单的线程池,每次执行任务时都会创建一个新的线程。这种线程池适用于短期的异步任务,但不适用于长期运行的任务,因为它可能会导致线程资源的浪费和系统负载的增加。
SimpleAsyncTaskExecutor
的主要特点如下:
- 每次执行任务都会创建一个新的线程,任务执行完毕后线程会被销毁。
- 可以设置线程名称前缀,方便调试和监控。
- 可以设置线程优先级,但是在大多数情况下,不建议使用线程优先级。
- 可以设置线程是否为守护线程,如果设置为守护线程,当所有非守护线程执行完毕后,守护线程会自动退出。
- 可以设置线程池大小,但是由于每次执行任务都会创建一个新的线程,因此线程池大小的设置实际上没有意义。
SimpleAsyncTaskExecutor的使用示例:
// 创建SimpleAsyncTaskExecutor
TaskExecutor executor = new SimpleAsyncTaskExecutor();
// 执行异步任务
executor.execute(new Runnable() {
@Override
public void run() {
// 异步任务的代码
}
});
需要注意的是,SimpleAsyncTaskExecutor不适用于高并发的场景,因为每次执行任务都会创建一个新的线程,线程的创建和销毁会带来一定的开销。
二、配置@Async的线程池
为了适应高并发的场景,我们可以自己配置@async
的异步线程池,建议使用ThreadPoolTaskExecutor等线程池实现。
Spring Boot的AsyncConfigurer接口是一个回调接口,用于提供异步执行器的自定义配置。通过实现该接口,可以自定义异步执行器的线程池大小、线程名称前缀、任务拒绝处理等相关参数。
它提供了两个方法来配置异步任务执行的线程池。
- getAsyncExecutor方法:返回一个TaskExecutor类型的bean,它提供了默认的线程池配置,比如线程池大小、队列大小、拒绝策略等等。
- getAsyncUncaughtExceptionHandler方法:返回一个AsyncUncaughtExceptionHandler类型的bean,用来处理异步任务中未捕获的异常。
这些方法分别提供了异步任务执行的线程池和异常处理的配置。我们可以在应用程序中实现AsyncConfigurer接口并重写这些方法以配置异步任务的线程池和异常处理器。以下是一个示例实现:
@Configuration
@EnableAsync
public class AsyncConfig implements AsyncConfigurer {
@Override
public Executor getAsyncExecutor() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(10);
executor.setMaxPoolSize(100);
executor.setQueueCapacity(10);
executor.setThreadNamePrefix("AsyncThread-");
executor.initialize();
return executor;
}
@Override
public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
return new CustomAsyncExceptionHandler();
}
private static class CustomAsyncExceptionHandler implements AsyncUncaughtExceptionHandler {
@Override
public void handleUncaughtException(Throwable ex, Method method, Object... params) {
// Custom exception handling code here
}
}
}
在上面的示例中,getAsyncExecutor()
方法返回一个ThreadPoolTaskExecutor
对象,该对象配置了线程池的核心大小、最大大小、队列容量和线程名称前缀。initialize()
方法用于初始化线程池。
三、使用其他线程池
@Async还可以使用其他的线程池,只需在@Async的属性中添加其他线程池的@Bean方法名
其他线程池
@Configuration
public class AppConfig {
@Bean
public Executor taskExecutor() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
executor.setCorePoolSize(10);
executor.setMaxPoolSize(100);
executor.setQueueCapacity(10);
executor.setThreadNamePrefix("MyAsyncThread-");
executor.initialize();
return executor;
}
}
然后,在异步执行的方法上,使用@Async注解来标注,并指定线程池名称:
@Service
public class MyService {
// 指定其他线程池,输入方法名
@Async("taskExecutor")
public void doSomethingAsync() {
// 异步执行的代码
}
}
这里@Async("taskExecutor")指定了异步执行使用的线程池名称为"taskExecutor",与上面配置的Bean名称一致。