java -- 通用线程池

浅浅的花香味﹌ 2024-03-23 14:08 66阅读 0赞

### 背景： ###

最近工作中使用到了线程池，所以写一个通用线程池工具类，后面需要使用线程池直接复用

### 代码： ###

import lombok.extern.slf4j.Slf4j;
    import java.util.concurrent.*;
    /**
    *   通用线程池
    * */
    @Slf4j
    public class PushThreadPoolUtils {
        public static volatile ThreadPoolExecutor pool=null;
    
    
        private final static int CORE_POOL_SIZE =10 ;//核心线程数
        private final static int MAXIMUM_POOL_SIZE =50 ;//最大线程数
        private final static long KEEP_ALIVE_TIME =5 ;//非核心线程空闲时间
        private final static int MAXIMUM_WORK_QUEUE = 400;//消息队列最大任务数
    
        private PushThreadPoolUtils(){
            pool=new ThreadPoolExecutor(CORE_POOL_SIZE,// 1. 核心线程数
                    MAXIMUM_POOL_SIZE,// 2. 最大线程数
                    KEEP_ALIVE_TIME, // 3. 非核心线程空闲时间
                    TimeUnit.SECONDS, // 4. 时间单位
                    new ArrayBlockingQueue<>(MAXIMUM_WORK_QUEUE),// 5. 阻塞队列
                    Executors.defaultThreadFactory(),// 6. 创建线程工厂
                    new ThreadPoolExecutor.AbortPolicy());// 7. 拒绝策略
            log.info("PushThreadPoolUtils 线程池创建完成");
        }
    
        public static ThreadPoolExecutor getInstance(){
            //双重if单例
            if (null == pool){
                synchronized (PushThreadPoolUtils.class){
                    if (pool == null){
                        new PushThreadPoolUtils();
                        return pool;
                    }
                }
            }
            return pool;
        }
    
        // 无响应执行
        public static void execute(Runnable runnable){
            getInstance().execute(runnable);
        }
    
        // 有响应执行
        public static<T> Future<T> submit(Callable<T> callable){
            return getInstance().submit(callable);
        }
    
    }

### 测试： ###

main方法测试代码，创建20个Runnable，模拟实际业务任务，放到线程池执行

public static void main(String[] args) {
            System.out.println("测试执行20个任务");
            for (int i=0; i<20; i++){
                String aa = "执行任务：" + (i+1);
                PushThreadPoolUtils.execute( new Runnable() {
                    public void run(){
                        System.out.println(aa);
                    }
                });
            }
        }

### 测试结果： ###

可以看到，线程池正常创建并执行了线程任务，任务的执行顺序也是无序的，因为创建线程去执行任务时，多个就绪的线程会竞争cpu，故而不是先创建就可以先执行，证明线程池的使用让串行处理变成了并行

测试执行20个任务
    14:59:28.188 [main] INFO  c.r.q.u.PushThreadPoolUtils - [<init>,28] - PushThreadPoolUtils 线程池创建完成
    执行任务：2
    执行任务：1
    执行任务：4
    执行任务：3
    执行任务：5
    执行任务：6
    执行任务：7
    执行任务：8
    执行任务：11
    执行任务：9
    执行任务：12
    执行任务：13
    执行任务：14
    执行任务：15
    执行任务：17
    执行任务：18
    执行任务：19
    执行任务：20
    执行任务：10
    执行任务：16

线程池的阻塞队列，线程工厂，还有超负荷后的拒绝策略都是可选的，可根据自身业务情况进行不同选择