springboot @Async 配置异步线程池

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的主要特点如下：

1. 每次执行任务都会创建一个新的线程，任务执行完毕后线程会被销毁。
2. 可以设置线程名称前缀，方便调试和监控。
3. 可以设置线程优先级，但是在大多数情况下，不建议使用线程优先级。
4. 可以设置线程是否为守护线程，如果设置为守护线程，当所有非守护线程执行完毕后，守护线程会自动退出。
5. 可以设置线程池大小，但是由于每次执行任务都会创建一个新的线程，因此线程池大小的设置实际上没有意义。

SimpleAsyncTaskExecutor的使用示例：

// 创建SimpleAsyncTaskExecutor
TaskExecutor executor = new SimpleAsyncTaskExecutor();

// 执行异步任务
executor.execute(new Runnable() {
    @Override
    public void run() {
        // 异步任务的代码
    }
});

需要注意的是，SimpleAsyncTaskExecutor不适用于高并发的场景，因为每次执行任务都会创建一个新的线程，线程的创建和销毁会带来一定的开销。

二、配置@Async的线程池

为了适应高并发的场景，我们可以自己配置@async的异步线程池，建议使用ThreadPoolTaskExecutor等线程池实现。

Spring Boot的AsyncConfigurer接口是一个回调接口，用于提供异步执行器的自定义配置。通过实现该接口，可以自定义异步执行器的线程池大小、线程名称前缀、任务拒绝处理等相关参数。

它提供了两个方法来配置异步任务执行的线程池。

1. getAsyncExecutor方法：返回一个TaskExecutor类型的bean，它提供了默认的线程池配置，比如线程池大小、队列大小、拒绝策略等等。
2. 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名称一致。