SpringBoot中ThreadPoolTaskExecutor的使用

1 配置自己的线程池

@Configuration
@EnableAsync
public class ThreadPoolConfig
{
    
    
	// 核心线程池大小
    private int corePoolSize = 50;

    // 最大可创建的线程数
    private int maxPoolSize = 200;

    // 队列最大长度
    private int queueCapacity = 1000;

    // 线程池维护线程所允许的空闲时间
    private int keepAliveSeconds = 300;

    @Bean(name = "threadPoolTaskExecutor")
    public ThreadPoolTaskExecutor threadPoolTaskExecutor()
    {
    
    
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        // 设置最大线程池数量
        executor.setMaxPoolSize(maxPoolSize);
        // 核心线程数
        executor.setCorePoolSize(corePoolSize);
        // 线程池维护线程所允许的空闲时间
        executor.setKeepAliveSeconds(keepAliveSeconds);
        // 队列最大长度
        executor.setQueueCapacity(queueCapacity);
        // 线程前缀名称
        executor.setThreadNamePrefix("ThreadPoolTaskExecutor======>");
        // 设置线程池关闭时，等待所有任务都完成后，再继续销毁其他的bean
        // 这样这些异步任务的销毁就会优先于 数据库连接池对象的销毁
        // 也就是说当其他bean都完成任务销毁后，数据库连接池对象才会销毁
        executor.setWaitForTasksToCompleteOnShutdown(true);
        // 任务的等待时间，如果超过这个时间还没有销毁，就强制销毁，以确保应用最后能够被关闭，而不是阻塞住
        executor.setAwaitTerminationSeconds(5);
        // 线程池对拒绝任务(无线程可用)的处理策略
        // 线程不够用时由调用的线程处理该任务
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        executor.initialize();
        return executor;
    }
}

• 一定要自己配置，不要使用SpringBoot自带的,Spring默认的线程池没有上限，并发多时，可能会OOM。并且不是线程重用，每次都会新建一个线程。ThreadPoolTaskExecutor 会重用线程，节省创建线程的资源。因此，在使用ThreadLocal时，也要及时清理缓存，否则，会获取到上一个任务的参数。

2 使用

2.1 在Service层使用

• Controller层直接返回结果，具体的业务，由其他线程执行。

	@Override
    @Async("threadPoolTaskExecutor")
    public void concatAB() {
    
    
    	......
    }

2.2 多线程中使用事务的写法

• 事务放到其他方法@Transactional，然后调用。

	@Override
    @Async("threadPoolTaskExecutor")
    public void concatAB() {
    
    
    	addChange()
    	......
    }

 @Transactional(rollbackFor = Exception.class,isolation = Isolation.SERIALIZABLE)
    public void addChange(){
    
    
		.......
	}

2.3 方法内多线程

• 同一个类下调用内部方法无法实现多线程，下面这种方法是错误的。因为，@Async是通过AOP实现的，方法没调用，没有进入切面。所以，无法实现多线程。

	@Override
    public void concatAB() {
    
    
    	addChange()
    	......
    }

    @Async("threadPoolTaskExecutor")
    public void addChange(){
    
    
		.......
	}

• 正确写法

	@Autowired
    private ThreadPoolTaskExecutor threadPoolTaskExecutor;
	
	@Override
    public void concatAB() {
    
    
    	threadPoolTaskExecutor.execute(() -> {
    
    
            ....
        });
    }

3 线程池与并行流的选取

• IO密集型的业务，选取线程池来完成。（例如，连接数据库之类的操作）
• cpu密集型的业务，选取并行流allelesvs.parallelStream()的方式处理。（例如，计算类、过滤、求和、分组等）