线程池管理多线程

一 .创建线程池配置

package com.web.config;


import java.util.concurrent.ThreadPoolExecutor;

import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

/**
 * 线程池配置
 * @author zhh
 *
 */
@Configuration
@EnableAsync
@Slf4j
public class ThreadPoolTaskConfig {

    /** 核心线程数（默认线程数） */
    private static final int CORE_POOL_SIZE = 8;
    /** 最大线程数 */
    private static final int MAX_POOL_SIZE = 16;
    /** 允许线程空闲时间（单位：默认为秒） */
    private static final int KEEP_ALIVE_TIME = 30;
    /** 缓冲队列大小 */
    private static final int QUEUE_CAPACITY = 200;
    /** 线程池名前缀 */
    private static final String THREAD_NAME_PREFIX = "Async-Service-";

    /**
     * bean的名称
     */
    @Bean("taskExecutor")
    public ThreadPoolTaskExecutor taskExecutor(){
        log.info("线程池创建");
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(CORE_POOL_SIZE);
        executor.setMaxPoolSize(MAX_POOL_SIZE);
        executor.setQueueCapacity(QUEUE_CAPACITY);
        executor.setKeepAliveSeconds(KEEP_ALIVE_TIME);
        executor.setThreadNamePrefix(THREAD_NAME_PREFIX);

        // 线程池对拒绝任务的处理策略 直接抛出异常
        // CallerRunsPolicy：由调用线程（提交任务的线程）处理该任务
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.AbortPolicy());
        // 初始化
        executor.initialize();
        return executor;
    }
	@Async("taskExecutor")
    public void funcation(Param param, CountDownLatch countDownLatch) {
        try {
            //业务逻辑
        } catch (Exception e) {
            log.error("SkuEditPrice抖店返回结果异常:{}", e);
        } finally {
            countDownLatch.countDown();
        }
    }

}

拒绝策略:

handler：表示当拒绝处理任务时的策略

AbortPolicy:丢弃任务并抛出RejectedExecutionException异常。

DiscardPolicy：也是丢弃任务，但是不抛出异常。

DiscardOldestPolicy：丢弃队列最前面的任务，然后重新尝试执行任务（重复此过程）

CallerRunsPolicy：由调用线程处理该任务

2. Spring中启用@Async注解调用线程池

@EnableAsync
@EnableTransactionManagement

@SpringBootApplication(scanBasePackages = {"com.xx.xx"})
@MapperScan("com.xx.xx.**.mapper")
public class XxMicroappWebApplication {
    public static void main(String[] args) {
        SpringApplication.run(XxMicroappWebApplication.class, args);
    }

}

3. 调用

@Autowired
ThreadPoolTaskConfig threadPoolTaskConfig;

public Ret  funcation(Param param){
		List list = new ArrayList<>();
		CountDownLatch countDownLatch=new CountDownLatch(list.size);
		for (int i = 0; i < list.size(); i++) {
            threadPoolTaskConfig.funcation(param, countDownLatch);
        }
        //等待所以线程执行完毕统一返回
        try {
            countDownLatch.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return Ret.success;
}

4.等待所以线程执行完毕统一返回

        CountDownLatch countDownLatch=new CountDownLatch(updateSkuStockParam.size());
        countDownLatch.countDown();
        try {
            countDownLatch.await();
        } catch (Exception e) {
            e.printStackTrace();
        }