Springboot中使用线程池

一、导读

java多线程的使用很多，这里介绍一下springboot中的多线程的使用，使用线程连接池。

二、关键代码

1. CustomThreadPoolTaskExecutor 类代码

public class CustomThreadPoolTaskExecutor extends ThreadPoolTaskExecutor {

    private void showThreadPoolInfo(String prefix) {
        ThreadPoolExecutor threadPoolExecutor = getThreadPoolExecutor();
        if (null == threadPoolExecutor) {
            return;
        }
        log.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);
    }
}
复制代码

1. ThreadPoolConfig 配置类代码

@Configuration
@EnableAsync
public class ThreadPoolConfig {

    @Value("${thread.pool.coreSize}")
    private int coreSize;

    @Value("${thread.pool.maxSize}")
    private int maxSize;

    @Value("${thread.pool.queueSize}")
    private int queueSize;

    @Value("${thread.pool.threadNamePrefix}")
    private String threadNamePrefix;

    /**
     * 线程池配置bean
     * @return Executor
     */
    @Bean
    public Executor asyncServiceExecutor() {
        log.info("start asyncServiceExecutor");
        // 使用 自定义 CustomThreadPoolTaskExecutor
        ThreadPoolTaskExecutor executor = new CustomThreadPoolTaskExecutor();
        // 配置核心线程数
        executor.setCorePoolSize(coreSize);
        // 配置最大线程数
        executor.setMaxPoolSize(maxSize);
        // 配置队列大小
        executor.setQueueCapacity(queueSize);
        // 配置线程池中的线程的名称前缀
        executor.setThreadNamePrefix(threadNamePrefix);
        // rejection-policy：当pool已经达到max size的时候，如何处理新任务
        // CALLER_RUNS：不在新线程中执行任务，而是有调用者所在的线程来执行
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        // 执行初始化
        executor.initialize();
        return executor;
    }
}
复制代码

1. ThreadService 使用类

@Service
public class ThreadService {

    /**
     * 带返回值
     * @return
     */
    @Async("asyncServiceExecutor")
    public Future<String> index1() {
        log.info("start executeAsync");
        try {
            Thread.sleep(2000);
        } catch (Exception e) {
            e.printStackTrace();
        }
        log.info("end executeAsync");
        return new AsyncResult<>("1");
    }

    /**
     * 不带返回值
     * @return
     */
    @Async("asyncServiceExecutor")
    public void index2() {
        log.info("start executeAsync");
        try {
            Thread.sleep(2000);
        } catch (Exception e) {
            e.printStackTrace();
        }
        log.info("end executeAsync");
    }
}
复制代码

三、后记

1. 最后说一下线程池，引用阿里巴巴开发规范的解释

【强制】线程池不允许使用Executors去创建，而是通过ThreadPoolExecutor的方式，这样的处理方式让写的同学更加明确线程池的运行规则，规避资源耗尽的风险。

说明：Executors各个方法的弊端：

1） newFixedThreadPool和newSingleThreadExecutor: 主要问题是堆积的请求处理队列可能会耗费非常大的内存，甚至OOM。

2） newCachedThreadPool和newScheduledThreadPool: 主要问题是线程数最大数是Integer.MAX_VALUE，可能会创建数量非常多的线程，甚至OOM。

【强制】创建线程或线程池时请指定有意义的线程名称，方便出错时回溯。

1. 现实程序中是绝对要避免直接显示创建线程，而不是使用线程池。显示创建线程会导致资源不合理利用已经导致oom的问题，线程创建和销毁都会消耗巨大资源，使用线程池来管理线程是明智的选择。

作者：fds小哥
链接：
https://juejin.cn/post/7087526683430879240