SpringBoot自定义线程池（ThreadPoolTaskExecutor）

Spring Boot 项目中可以直接使用 @Async 注解即可实现异步调用。

• 方法上添加 @Async 注解。
• 启动类或者配置类（@Configuration），添加 @EnableAsync 开启异步调用。

@SpringBootApplication
@EnableAsync // @Async注解能够生效
public class DailyApplication {

   public static void main(String[] args) {
       SpringApplication.run(DairlyLearnApplication.class, args);
  }
}

@Configuration // 配置类
@EnableAsync // @Async注解能够生效
public class TaskConfiguration {

}

自定义线程池

推荐使用：@Async("eventTaskExecutor")，可以进行线程池隔离。

配置文件配置默认线程池（不推荐）

# 默认线程池配置
# 线程池创建时的初始化线程数，默认为8
spring.task.execution.pool.core-size=2
# 线程池的最大线程数，默认Integer.MAX_VALUE
spring.task.execution.pool.max-size=5
# 用来缓冲执行任务的队列，默认Integer.MAX_VALUE
spring.task.execution.pool.queue-capacity=10
# 线程终止前允许保持空闲的时间，默认60L
spring.task.execution.pool.keep-alive=60s
# 是否允许核心线程超时，默认 true
spring.task.execution.pool.allow-core-thread-timeout=true
# 是否等待剩余任务完成后才关闭应用
spring.task.execution.shutdown.await-termination=false
# 等待剩余任务完成的最大时间
spring.task.execution.shutdown.await-termination-period=
# 线程名的前缀，默认 task-
spring.task.execution.thread-name-prefix=task-

配置文件的方式，是可以自定义线程池参数，但是如果想要做到线程池的隔离，就需要配合根据业务不同，使用不同的线程池。

自定义 TaskExecutor

@Configuration // 配置类
@EnableAsync // @Async注解能够生效
public class TaskConfiguration {
   @Bean("taskExecutor")
   public Executor taskExecutor() {
       ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
       // 线程池创建时候初始化的线程数
       executor.setCorePoolSize(5);
       // 线程池最大的线程数，只有在缓冲队列满了之后，才会申请超过核心线程数的线程
       executor.setMaxPoolSize(10);
       // 用来缓冲执行任务的队列
       executor.setQueueCapacity(200);
       // 当超过了核心线程之外的线程，在空闲时间到达之后会被销毁
       executor.setKeepAliveSeconds(60);
       // 可以用于定位处理任务所在的线程池
       executor.setThreadNamePrefix("taskExecutor-orderSuccess-");
       // 这里采用CallerRunsPolicy策略，当线程池没有处理能力的时候，该策略会直接在execute方法的调用线程中运行被拒绝的任务；
       // 如果执行程序已关闭，则会丢弃该任务
       executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
       // 设置 线程池关闭 的时候 等待 所有任务都完成后，再继续 销毁 其他的 Bean，
       // 这样这些 异步任务 的 销毁 就会先于 数据库连接池对象 的销毁。
       executor.setWaitForTasksToCompleteOnShutdown(true);
       // 该方法用来设置线程池中 任务的等待时间，如果超过这个时间还没有销毁就 强制销毁，以确保应用最后能够被关闭，而不是阻塞住。
       executor.setAwaitTerminationSeconds(60);
       return executor;
  }
}

实现 AsyncConfigurer（推荐使用）

@Configuration
@Slf4j
@EnableAsync // @Async注解能够生效
public class AsyncConfiguration implements AsyncConfigurer {
   @Bean("eventTaskExecutor")
   public ThreadPoolTaskExecutor executor() {
       //Spring封装的一个线程池
       ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
       executor.setCorePoolSize(10);
       executor.setMaxPoolSize(50);
       executor.setQueueCapacity(30);
       executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
       executor.setThreadNamePrefix("eventTaskExecutor--orderSuccess-");
       executor.initialize();
       return executor;
  }

   @Override
   public Executor getAsyncExecutor() {
       return executor();
  }

   /**
    * 异常处理
    * @return
    */
   @Override
   public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
       return (ex, method, params) -> log.error(String.format("[async] task{} error:", method), ex);
  }
}

继承 AsyncConfigurerSupport

已经 deprecated，建议使用上面的方式。