Java實現(xiàn)手寫線程池實例并測試詳解

更新時間：2023年02月22日 15:08:46 作者：小威要向諸佬學習呀

這篇文章主要來模擬一下線程池和工作隊列的流程，以及編寫代碼和測試類進行測試。文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下

前言

在之前的文章中介紹過線程池的核心原理，在一次面試中面試官讓手寫線程池，這塊知識忘記的差不多了，因此本篇文章做一個回顧。

希望能夠加深自己的印象以及幫助到其他的小伙伴兒們

在線程池核心原理篇介紹過線程池的核心原理，今天來模擬線程池和工作隊列的流程，以及編寫代碼和測試類進行測試。下面附下之前線程池的核心流程：

在線程池核心原理的源碼中，涉及到了一系列的流程，包括線程池隊列數(shù)量是否已滿，運用什么樣的拒絕策略等。在我們手寫線程池的代碼中，不需要考慮那么多因素，只需要模擬簡單的情景和過程，因此整體來講還是比較簡單的。

手寫線程池，必不可少的組件有任務隊列，任務的消費者線程池，線程池創(chuàng)建等。我們也可定義構造方法，創(chuàng)建指定大小的線程池線程個數(shù)。當然在使用完線程池中的線程后，我們需要考慮將其銷毀或關閉。

相關代碼和解釋如下：

package XIAOWEI;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.stream.IntStream;

public class ThreadPool {
    private static final int WorkQueueSIZE = 5;
    private BlockingQueue<Runnable> workQueue;
    private List<WorkThread> workThreads=new ArrayList<WorkThread>();

    /**
     * @author xiaowei
     * @param poolSize
     * @param workQueue
     * 構造方法，傳線程池的大小和阻塞隊列
     */
    public ThreadPool(int poolSize,BlockingQueue<Runnable> workQueue) {
        this.workQueue = workQueue;
        IntStream.range(0,poolSize).forEach((i)->{
            WorkThread workThread=new WorkThread();
            workThread.start();
            workThreads.add(workThread);
        });
    }

    /**
     * @param poolSize
     * 在ThreadPool的構造方法中傳入線程池的大小
     */
    public ThreadPool(int poolSize){
        this(poolSize,new LinkedBlockingQueue<>(WorkQueueSIZE));
    }

    /**
     * @param task
     * 通過線程池執(zhí)行任務
     */
    public void extcute(Runnable task){
        try {
            workQueue.put(task);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    public void shutdown(){
        if(workThreads!=null && workThreads.size()>0){
            workThreads.stream().forEach((workThread) -> {
                workThread.interrupt();
            });
        }
    }
    /**
     * 內部類 源源不斷的消耗workQueue中的任務
     */
    class WorkThread extends Thread {
        @Override
        public void run() {
            Thread currentThread = Thread.currentThread();
            // 死循環(huán) 不斷一直消費隊列中的任務 直到任務被消費完全
            while (true) {
                try {
                    if (currentThread.isInterrupted()) {
                        break;
                    }
                    Runnable workTasK = workQueue.take();
                    workTasK.run();
                } catch (InterruptedException e) {
                    currentThread.interrupt();
                }
            }
        }
    }
}

相關測試代碼如下：

package XIAOWEI;

import java.util.stream.IntStream;

public class ThreadPoolTest {
    public static void main(String[] args) {
        ThreadPool threadPool = new ThreadPool(5);
        IntStream.range(0,10).forEach((i)->{
            threadPool.extcute(()->{
                System.out.println(Thread.currentThread().getName()+"2023一起加油");
            });
        });
        threadPool.shutdown();
    }
}

一共在線程池中加入了五個線程和十個任務，因此每個線程會執(zhí)行兩個任務。

測試結果如下圖所示：