spring boot使用自定义配置的线程池执行Async异步任务

Spring Boot 使用自定义配置的线程池执行 Async 异步任务 在 Spring Boot 中执行异步任务时，默认情况下使用的是默认的线程池，但是，在实际项目中，我们可能需要根据项目的需求来定制自己的线程池。下面将介绍如何在 Spring Boot 中使用自定义配置的线程池执行 Async 异步任务。 一、自定义线程池配置类 在 Spring Boot 中，我们可以使用 `@ConfigurationProperties` 注解来定义自定义的配置类。例如，我们可以创建一个 `TaskThreadPoolConfig` 类，如下所示： ```java package com.chhliu.springboot.async.configuration; import org.springframework.boot.context.properties.ConfigurationProperties; @ConfigurationProperties(prefix = "spring.task.pool") public class TaskThreadPoolConfig { private int corePoolSize; private int maxPoolSize; private int keepAliveSeconds; private int queueCapacity; // getter, setter 方法省略 } ``` 二、创建线程池 在创建线程池时，我们需要使用 `ThreadPoolTaskExecutor` 类， 并使用 `@Bean` 注解将其注入到 Spring 容器中。例如，我们可以创建一个 `TaskExecutePool` 类，如下所示： ```java package com.chhliu.springboot.async.pool; import java.util.concurrent.Executor; import java.util.concurrent.ThreadPoolExecutor; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.scheduling.annotation.EnableAsync; import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; import com.chhliu.springboot.async.configuration.TaskThreadPoolConfig; @Configuration @EnableAsync public class TaskExecutePool { @Autowired private TaskThreadPoolConfig config; @Bean public Executor myTaskAsyncPool() { ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); executor.setCorePoolSize(config.getCorePoolSize()); executor.setMaxPoolSize(config.getMaxPoolSize()); executor.setQueueCapacity(config.getQueueCapacity()); executor.setKeepAliveSeconds(config.getKeepAliveSeconds()); executor.setThreadNamePrefix("MyExecutor-"); // rejection-policy：当 pool 已经达到 max size 时，如何处理新任务 // CALLER_RUNS：不在新线程中执行任务，而是由调用者所在的线程来执行 executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); executor.initialize(); return executor; } } ``` 三、在主类中开启配置支持 在主类中，我们需要开启 `@EnableAsync` 注解，以便启用异步支持。例如： ```java package com.chhliu.springboot.async; import org.springframework.boot.SpringApplication; import org.springframework.boot.autoconfigure.SpringBootApplication; import org.springframework.scheduling.annotation.EnableAsync; @SpringBootApplication @EnableAsync public class AsyncApplication { public static void main(String[] args) { SpringApplication.run(AsyncApplication.class, args); } } ``` 四、使用自定义线程池 在使用自定义线程池时，我们可以使用 `@Async` 注解来标注异步方法，例如： ```java @Service public class MyService { @Async public void doSomethingAsync() { // 异步执行的代码 } } ``` 在上面的示例中，我们使用了 `@Async` 注解来标注 `doSomethingAsync` 方法，以便使用自定义的线程池来执行异步任务。 五、总结 在 Spring Boot 中使用自定义配置的线程池执行 Async 异步任务，可以通过定义自定义的配置类、创建线程池和开启配置支持来实现。在实际项目中，我们可以根据项目的需求来定制自己的线程池，以提高系统的性能和灵活性。