Spring ~ 线程池配置

默认线程池ThreadPoolTaskExecutor配置

1. 配置核心参数

   • 直接在application.properties中配置核心参数

   spring.task.execution.pool.core-size=8
   spring.task.execution.pool.max-size=12
   spring.task.execution.pool.keep-alive=60s
   spring.task.execution.pool.queue-capacity=100000
   spring.task.execution.pool.allow-core-thread-timeout=true
   spring.task.execution.thread-name-prefix=swy-task-
   

   • 创建JavaBean注入

   @Configuration
   public class ExecutorConfig {
   
       private static final Logger logger = LoggerFactory.getLogger(ExecutorConfig.class);
   
       @Bean
       public Executor asyncServiceExecutor() {
           logger.info("start asyncServiceExecutor");
           ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
           //配置核心线程数
           executor.setCorePoolSize(5);
           //配置最大线程数
           executor.setMaxPoolSize(6);
           //配置队列大小
           executor.setQueueCapacity(99999);
           //配置线程池中的线程的名称前缀
           executor.setThreadNamePrefix("swy-task-");
   
           // rejection-policy：当pool已经达到max size的时候，如何处理新任务
           // CALLER_RUNS：不在新线程中执行任务，而是有调用者所在的线程来执行
           executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
           //执行初始化
           executor.initialize();
           return executor;
       }
   }
   

2. 在配置类，或入口类开启@EnableAsync注解

   @SpringBootApplication
   @EnableAsync
   public class MultiThreadApplication {
   	public static void main(String[] args) {
   		SpringApplication.run(MultiThreadApplication.class, args);
   	}
   
   }
   

3. 在Service层或Controller层的类或方法上添加@Async注解

   @Async
   public void doSomethingAsync(){
   	logger.info("start executeAsync");
   	try{
   		Thread.sleep(5000);
   	}catch(Exception e){
   		e.printStackTrace();
   	}
   	logger.info("end executeAsync");
   }
   

自定义线程池ThreadPoolTaskExecutor配置

1. 继承ThreadPoolTaskExecutor创建新线程池类

   public class CustomThreadPoolTaskExecutor extends ThreadPoolTaskExecutor {
       private static final Logger logger = LoggerFactory.getLogger(CustomThreadPoolTaskExecutor.class);
   
       private void showThreadPoolInfo(String prefix){
           ThreadPoolExecutor threadPoolExecutor = getThreadPoolExecutor();
   
           if(null==threadPoolExecutor){
               return;
           }
   
           logger.info("{}, {},taskCount [{}], completedTaskCount [{}], activeCount [{}], queueSize [{}]",
                   this.getThreadNamePrefix(),
                   prefix,
                   threadPoolExecutor.getTaskCount(),
                   threadPoolExecutor.getCompletedTaskCount(),
                   threadPoolExecutor.getActiveCount(),
                   threadPoolExecutor.getQueue().size());
       }
   
       @Override
       public void execute(Runnable task) {
           showThreadPoolInfo("1. do execute");
           super.execute(task);
       }
   
       @Override
       public void execute(Runnable task, long startTimeout) {
           showThreadPoolInfo("2. do execute");
           super.execute(task, startTimeout);
       }
   
       @Override
       public Future<?> submit(Runnable task) {
           showThreadPoolInfo("1. do submit");
           return super.submit(task);
       }
   
       @Override
       public <T> Future<T> submit(Callable<T> task) {
           showThreadPoolInfo("2. do submit");
           return super.submit(task);
       }
   
       @Override
       public ListenableFuture<?> submitListenable(Runnable task) {
           showThreadPoolInfo("1. do submitListenable");
           return super.submitListenable(task);
       }
   
       @Override
       public <T> ListenableFuture<T> submitListenable(Callable<T> task) {
           showThreadPoolInfo("2. do submitListenable");
           return super.submitListenable(task);
       }
   }
   

2. 配置新建线程池类的核心参数

   @Configuration
   public class ExecutorConfig {
       private static final Logger logger = LoggerFactory.getLogger(ExecutorConfig.class);
       @Bean
       public Executor asyncServiceExecutor() {
           logger.info("start asyncServiceExecutor");
           ThreadPoolTaskExecutor executor = new CustomThreadPoolTaskExecutor();
           //配置核心线程数
           executor.setCorePoolSize(5);
           //配置最大线程数
           executor.setMaxPoolSize(8);
           //配置队列大小
           executor.setQueueCapacity(99999);
           //配置线程池中的线程的名称前缀
           executor.setThreadNamePrefix("async-service-");
   
           // rejection-policy：当pool已经达到max size的时候，如何处理新任务
           // CALLER_RUNS：不在新线程中执行任务，而是有调用者所在的线程来执行
           executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
           //执行初始化
           executor.initialize();
           return executor;
       }
   }
   

3. 在配置类，或入口类开启@EnableAsync注解

   @SpringBootApplication
   @EnableAsync
   public class MultiThreadApplication {
   	public static void main(String[] args) {
   		SpringApplication.run(MultiThreadApplication.class, args);
   	}
   
   }
   

4. 在Service层或Controller层的类或方法上添加@Async注解，此时需需注意一定要注明Bean方法名称。

   @Async("asyncServiceExecutor")
   public void doSomethingAsync(){
   	logger.info("start executeAsync");
   	try{
   		Thread.sleep(5000);
   	}catch(Exception e){
   		e.printStackTrace();
   	}
   	logger.info("end executeAsync");
   }