源碼解讀--線程池TheadPoolExecutor深入講解

     在使用線程的,我們常使用線程池,線程的複用可以減少線程的開銷,很好的提高cpu資源利用率。這篇文章信息量會很大,做好準備哦~ java源碼系列

    有對線程池沒有直觀感覺的同學可以看看這篇文章。在jdk8中,從ThreadPoolExecutor從發,來逐步講解它的源碼實現。

1.線程池定義初始值

private final AtomicInteger ctl = new AtomicInteger(ctlOf(RUNNING, 0));  
private static final int COUNT_BITS = Integer.SIZE - 3;  
private static final int CAPACITY   = (1 << COUNT_BITS) - 1;  //容量限制在2^29-1

//線程池中線程有五種狀態
private static final int RUNNING    = -1 << COUNT_BITS;  //接收新任務並且處理隊列的中任務
private static final int SHUTDOWN   =  0 << COUNT_BITS;  //不接收新任務,但是處理隊列中的任務
private static final int STOP       =  1 << COUNT_BITS;  // 不接收新任務,不處理隊列中的任務,中斷正在處理的任務
private static final int TIDYING    =  2 << COUNT_BITS;  //所有的任務都已經停止了,工作線程的數目爲0,線程在轉換到改狀態是會調用中斷方法
private static final int TERMINATED =  3 << COUNT_BITS;  //調用了中斷方法

private static int runStateOf(int c)     { return c & ~CAPACITY; }  //獲取線程運行的狀態
private static int workerCountOf(int c)  { return c & CAPACITY; }  //當前工作的線程
private static int ctlOf(int rs, int wc) { return rs | wc; }     //獲取ctl
private static boolean runStateAtLeast(int c, int s) {  //比較最後的狀態值
    return c >= s;
}
//判斷線程是否正在執行
private static boolean isRunning(int c) {
    return c < SHUTDOWN;
}

初始化了一定基本變量及方法,下面會經常使用到。

2.執行方法

public void execute(Runnable command) {  //提交線程的執行方法
    if (command == null)  //如果放入線程池中的線程爲null,會包空指針異常
        throw new NullPointerException();
    int c = ctl.get();  //獲取clt的當前值,這個值會經常被修改
    if (workerCountOf(c) < corePoolSize) { //工作線程個數小於核心線程個數
        if (addWorker(command, true))  //新建一個線程來執行任務
            return;
        c = ctl.get();  //添加線程失敗,重新回去ctl的值,下面isRunning方法需要判斷線程是否是運行狀態
    }
    if (isRunning(c) && workQueue.offer(command)) {  //如果線程的個數大於核心線程池的個數,並且當前線程是運行狀態,它就會被放到緩存隊列中
        int recheck = ctl.get();
        if (! isRunning(recheck) && remove(command))  //如果放到隊列成功了,但是線程池被關了,需要移除掉該線程,沒問題的話就拒絕這個線程的執行
            reject(command);
        else if (workerCountOf(recheck) == 0)  //如果工作線程的數量變成0了,添加空的任務,之後其實知道從隊列中取任務時會跳過null的
            addWorker(null, false);
    }
    else if (!addWorker(command, false))  //隊列滿了,如果小於最大線程池的大小,可以新建線程執行任務,如果大於最大線程池大小,就會拒絕任務
        reject(command);
}

總結,總有有四步:

第一步,判斷線程是不是爲空,如果是空的話,就拋出空指針異常。

第二步,如果線程的數量小於核心線程池的個數,創建新的線程執行任務。

第三步,如果線程的數量大於核心線程的數量,就將線程放到緩存隊列中。

第四步,緩存隊列的滿了,還沒有到達最大核心線程池的大小,這個時候就也是創建線程執行任務,如果大於最大的核心線程池的大小,那麼就拒絕任務的執行。這裏所說執行任務的線程就是Worker.

2.1添加Woker線程

先來看下Worker的定義

private final class Worker
    extends AbstractQueuedSynchronizer
    implements Runnable
{
    private static final long serialVersionUID = 6138294804551838833L;

    final Thread thread;
    Runnable firstTask;
    volatile long completedTasks;
    Worker(Runnable firstTask) {
        setState(-1); // inhibit interrupts until runWorker
        this.firstTask = firstTask;
        this.thread = getThreadFactory().newThread(this);
    }

    /** Delegates main run loop to outer runWorker  */
    public void run() {
        runWorker(this);
    }
.......
}
繼承了aqs,同時實現了Runable,說明是一個線程.

private boolean addWorker(Runnable firstTask, boolean core) {
    retry:  //retry:  不明白是什麼的可以看這篇文章
    for (;;) {
        int c = ctl.get();  //獲取到最新的ctl
        int rs = runStateOf(c);  //查看當前線程的狀態

        //如果線程的狀態是關閉,第一個任務時空,隊列是空,直接返回添加任務失敗
        if (rs >= SHUTDOWN &&
            ! (rs == SHUTDOWN &&
               firstTask == null &&
               ! workQueue.isEmpty()))
            return false;

        for (;;) {
            int wc = workerCountOf(c);
            if (wc >= CAPACITY ||  //需要小於容量
                wc >= (core ? corePoolSize : maximumPoolSize))  //core爲true,工作線程要小於核心線程數,core爲false,工作線程數要小於最大線程數目。不然,也放回添加任務失敗
                return false;
            if (compareAndIncrementWorkerCount(c))  //如果任務個數添加成功,就跳出循環,這個用到了CAS,不明白的可以看這篇文章
                break retry;
            c = ctl.get();  // Re-read ctl
            if (runStateOf(c) != rs)  //運行狀態不等的話,需要不斷的重試,直到成功
                continue retry;
            // else CAS failed due to workerCount change; retry inner loop
        }
    }
    //下面方法是真正的創建線程,上面乾的事,你可以理解爲對環境的校驗
    boolean workerStarted = false;
    boolean workerAdded = false;
    Worker w = null;
    try {
        w = new Worker(firstTask);  //初始化一個任務
        final Thread t = w.thread;
        if (t != null) {  //當前線程不爲空的情況下
            final ReentrantLock mainLock = this.mainLock;
            mainLock.lock();  //獲取鎖
            try {
                int rs = runStateOf(ctl.get());  //需要重新獲取到線程的運行狀態,在獲取鎖之前,他的狀態可能被別的線程改變。
                if (rs < SHUTDOWN ||
                    (rs == SHUTDOWN && firstTask == null)) {  //在線程池是運行狀態,或者線程池被關閉了並且第一個線程是空的情況下
                    if (t.isAlive()) // 如果線程是活躍狀態,那麼說明先狀態有問題,直接拋出異常
                        throw new IllegalThreadStateException();
                    workers.add(w);  //將任務加到工作線程集裏面
                    int s = workers.size();
                    if (s > largestPoolSize)  //largestPoolSize是記錄線程池中出現的最大的線程的數量
                        largestPoolSize = s;
                    workerAdded = true;
                }
            } finally {
                mainLock.unlock();  //解鎖
            }
            if (workerAdded) {
                t.start();  //線程啓動,相應的調用run方法
                workerStarted = true;
            }
        }
    } finally {
        if (! workerStarted)  //如果工作線程啓動失敗,就將它從工作線程集中移除
            addWorkerFailed(w);  
    }
    return workerStarted;
}

總結一下,先校驗線程的狀態,沒有問題就創建一個工作線程,如果線程創建沒有問題就啓動了,如果有問題就將這個工作線程從工作線程集中移除。

3.運行線程

前面已經講過了將線程建立起來,現在需要使線程跑起來了。大家都知道要使線程跑起來,肯定要執行start方法。沒錯,線程池裏面也是這樣乾的。

public void run() {
    runWorker(this);
}

繼續往下看,woker裏面的run方法調用的是線程池裏面的runWorker方法。

final void runWorker(Worker w) {
    Thread wt = Thread.currentThread();  //取到當前線程
    Runnable task = w.firstTask;  //拿到第一個工作線程的第一個線程
    w.firstTask = null;
    w.unlock(); // 允許出現多箇中斷
    boolean completedAbruptly = true;
    try {
        while (task != null || (task = getTask()) != null) {  //不斷循環取任務
            w.lock();  //加鎖,別的線程不能進入
            //如果線程線程池被關閉了,線程沒沒有關閉的話,需要中斷
            if ((runStateAtLeast(ctl.get(), STOP) ||
                 (Thread.interrupted() &&
                  runStateAtLeast(ctl.get(), STOP))) &&
                !wt.isInterrupted())
                wt.interrupt();
            try {
                beforeExecute(wt, task);  //在線程執行之前的處理
                Throwable thrown = null;
                try {
                    task.run();  //線程執行
                } catch (RuntimeException x) {  //有錯拋出來
                    thrown = x; throw x;
                } catch (Error x) {
                    thrown = x; throw x;
                } catch (Throwable x) {  
                    thrown = x; throw new Error(x);
                } finally {
                    afterExecute(task, thrown);  //線程執行之後的處理
                }
            } finally {
                task = null;  //將任務置成null
                w.completedTasks++;  //完成的任務加一
                w.unlock();
            }
        }
        completedAbruptly = false;  //是否完全結束,false表示 完全結束了
    } finally {
        processWorkerExit(w, completedAbruptly);  //清除任務
    }
}

總結:不斷的取任務執行,如果執行完成了記錄任務的執行數加一,將這個線程從工作任務(工作線程)集中刪除。

進一步查看是怎麼獲取任務的getTask()

private Runnable getTask() {
    boolean timedOut = false; 

    for (;;) {
        int c = ctl.get();
        int rs = runStateOf(c);

        // 如果線程池關了,返回的任務時null
        if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
            decrementWorkerCount();
            return null;
        }

        int wc = workerCountOf(c);

        // 是否允許核心線程數超時,當滿足allowCoreThreadTimeOut被設置爲true 或當前線程數大於核心線程數時,timed=true
        boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;

        if ((wc > maximumPoolSize || (timed && timedOut))  //超過最大線程數或超時或任務隊列爲空
            && (wc > 1 || workQueue.isEmpty())) {
            if (compareAndDecrementWorkerCount(c))  //如果線程數減到了-1 ,就返回null
                return null;
            continue;  //跳過下面的任務執行
        }

        try { 
            Runnable r = timed ?  //設置了超時,按超時的,沒有超時的話,就直接從隊列的拿出任務來執行
                workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                workQueue.take();
            if (r != null)
                return r;
            timedOut = true;
        } catch (InterruptedException retry) {
            timedOut = false;
        }
    }
}

到此差不多已經基本上講完了線程池的運行流程,還有一個添加進工作隊列的方法沒有講,涉及到了LinkedBlockQueue,已經專門寫了一篇文章來講解。

其實講了這麼多,大家只要知道線程池的使用和運行流程就可以了。它的使用demo在此就不在講解了,網上也有很多例子。再來溫習一下它的運行流程。

第一步,判斷線程是不是爲空,如果是空的話,就拋出空指針異常。

第二步,如果線程的數量小於核心線程池的個數,創建新的線程執行任務。

第三步,如果線程的數量大於核心線程的數量,就將線程放到緩存隊列中。

第四步,緩存隊列的滿了,還沒有到達最大核心線程池的大小,這個時候就也是創建線程執行任務,如果大於最大的核心線程池的大小,那麼就拒絕任務的執行。這裏所說執行任務的線程就是Worker.

當然如果設置了超時,在時間範圍內,任務肯定是可以被執行的,超時返回的就是一個空的任務,不會執行之前的任務。有一個地方需要注意的是:只有在線程數大於核心線程數的時候,任務纔會被放到任務隊列裏面,小於的話,任務時直接執行的。


發佈了60 篇原創文章 · 獲贊 122 · 訪問量 17萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章