SpringBoot自定义线程池

使用SpringBoot构建应用时，如何使用线程处理异步任务？其实springBoot已经提供了默认的实现，通过在启动类上加上注解@EnableAsync, 然后在需要异步处理的方法上增加注解@Async即可启动一个线程进行异步处理。其实质类似于：new Thread(()-{System.out.print("处理异步任务")}).start()。但需要注意的是@Async默认使用的是SimpleAsyncTaskExecutor，每次提交任务都创建线程，要小心在使用大任务的场景下创建大量线程导致OOM异常。所以，还是自定义线程池好。自定义线程池有2种实现方式。

1. 自定义线程池，然后使用自己定义的线程池

  1.1 修改application.yml文件，增加线程池的配置参数，如下

   

 

 

 1.2 线程池配置属性类MyThreadPoolConfig .java

/**
* 线程池配置属性类
*/

@ConfigurationProperties(prefix = "mytask.pool")

public class MyThreadPoolConfig {
private int corePoolSize;

private int maxPoolSize;

private int keepAliveSeconds;

private int queueCapacity;

}

 

1.3  启动类上一定要开启线程异步支持：@EnableAsync

@EnableAsync
@EnableConfigurationProperties({MyThreadPoolConfig.class})
@SpringCloudApplication

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

 

1.4 创建自己的线程池

      

/**
 * 创建线程池
 */
@Configuration
public class MyTaskExecutePool {
 
    @Autowired
    private MyThreadPoolConfig config;
 
    @Bean("myTaskPool")
    public Executor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        //核心线程池大小
        executor.setCorePoolSize(config.getCorePoolSize());
        //最大线程数
        executor.setMaxPoolSize(config.getMaxPoolSize());
        //队列容量
        executor.setQueueCapacity(config.getQueueCapacity());
        //活跃时间
        executor.setKeepAliveSeconds(config.getKeepAliveSeconds());
        //线程名字前缀
        executor.setThreadNamePrefix("TaskExecutePool-");
 
        // setRejectedExecutionHandler：当pool已经达到maxSize的时候，如何处理新进任务
        // CallerRunsPolicy：不在新线程中执行任务，而是由调用者所在的线程来执行
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
         // 等待所有任务结束后再关闭线程池
        executor.setWaitForTasksToCompleteOnShutdown(true);
        executor.initialize();
        return executor;
    }
}

 

  需要注意的是这样定义的线程池在使用的时候要在@Async主键里指定名称，如：@Async("mytaskExecutor"), 否则会使用spingtBoot提供的默认线程池。

2. 重新spring默认的线程池

  第一种方式创建的线程池在使用的时候必须要指明异步任务要使用的线程池名称，而重新srping默认的线程池，则不需要指定名称，直接写@Asyncj即可。

  配置方式和方式一类似，唯一的区别在于配置类要实现AsyncConfigurer

  

/**
* 重新SpringBoot默认的线程池
*/
@Configuration
public class OverrideAsyncTaskExecutePool implements AsyncConfigurer{


//注入配置类
@Autowired
MyThreadPoolConfig config;

@Override
public Executor getAsyncExecutor() {
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
//核心线程池大小
executor.setCorePoolSize(config.getCorePoolSize());
//最大线程数
executor.setMaxPoolSize(config.getMaxPoolSize());
//队列容量
executor.setQueueCapacity(config.getQueueCapacity());
//活跃时间
executor.setKeepAliveSeconds(config.getKeepAliveSeconds());
//线程名字前缀
executor.setThreadNamePrefix("MyExecutor-");

// setRejectedExecutionHandler：当pool已经达到max size的时候，如何处理新任务
// CallerRunsPolicy：不在新线程中执行任务，而是由调用者所在的线程来执行
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
executor.initialize();
return executor;
}