深讀源碼-java線程系列之線程池深入解析——普通任務執行流程

簡介

前面我們一起學習了Java中線程池的體系結構、構造方法和生命週期,本章我們一起來學習線程池中普通任務到底是怎麼執行的。

建議學習本章前先去複習之前《深讀源碼-java線程系列之自己手寫一個線程池》這兩章內容,有助於理解本章的內容,且那邊的代碼比較短小,學起來相對容易一些。

問題

(1)線程池中的普通任務是怎麼執行的?

(2)任務又是在哪裏被執行的?

(3)線程池中有哪些主要的方法?

(4)如何使用Debug模式一步一步調試線程池?

使用案例

我們創建一個線程池,它的核心數量爲5,最大數量爲10,空閒時間爲1秒,隊列長度爲5,拒絕策略打印一句話。

如果使用它運行20個任務,會是什麼結果呢?

public class ThreadPoolTest01 {
    public static void main(String[] args) {
        // 新建一個線程池
        // 核心數量爲5,最大數量爲10,空閒時間爲1秒,隊列長度爲5,拒絕策略打印一句話
        ExecutorService threadPool = new ThreadPoolExecutor(5, 10,
                1, TimeUnit.SECONDS, new ArrayBlockingQueue<>(5),
                Executors.defaultThreadFactory(), new RejectedExecutionHandler() {
            @Override
            public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
                System.out.println(currentThreadName() + ", discard task");
            }
        });

        // 提交20個任務,注意觀察num
        for (int i = 0; i < 20; i++) {
            int num = i;
            threadPool.execute(()->{
                try {
                    System.out.println(currentThreadName() + ", "+ num + " running, " + System.currentTimeMillis());
                    Thread.sleep(2000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            });
        }

    }

    private static String currentThreadName() {
        return Thread.currentThread().getName();
    }
}

構造方法的7個參數我們就不詳細解釋了,有興趣的可以看看《深讀源碼-java線程系列之線程池深入解析——構造方法》這章。

我們一起來看看一次運行的結果:

pool-1-thread-1, 0 running, 1572678434411
pool-1-thread-3, 2 running, 1572678434411
pool-1-thread-2, 1 running, 1572678434411
pool-1-thread-4, 3 running, 1572678434411
pool-1-thread-5, 4 running, 1572678434411
pool-1-thread-6, 10 running, 1572678434412
pool-1-thread-7, 11 running, 1572678434412
pool-1-thread-8, 12 running, 1572678434412
main, discard task
main, discard task
main, discard task
main, discard task
main, discard task

pool-1-thread-9, 13 running, 1572678434412
pool-1-thread-10, 14 running, 1572678434412
pool-1-thread-3, 5 running, 1572678436411
pool-1-thread-1, 6 running, 1572678436411
pool-1-thread-6, 7 running, 1572678436412
pool-1-thread-2, 8 running, 1572678436412
pool-1-thread-7, 9 running, 1572678436412

注意,觀察num值的打印信息,先是打印了0~4,再打印了10~14,最後打印了5~9,竟然不是按順序打印的,爲什麼呢?

讓我們一步一步debug進去查看。

execute()方法

execute()方法是線程池提交任務的方法之一,也是最核心的方法。

// 提交任務,任務並非立即執行,所以翻譯成執行任務似乎不太合適
public void execute(Runnable command) {
    // 任務不能爲空
    if (command == null)
        throw new NullPointerException();
        
    // 控制變量(高3位存儲狀態,低29位存儲工作線程的數量)
    int c = ctl.get();
    // 1. 如果工作線程數量小於核心數量
    if (workerCountOf(c) < corePoolSize) {
        // 就添加一個工作線程(核心)
        if (addWorker(command, true))
            return;
        // 重新獲取下控制變量
        c = ctl.get();
    }
    // 2. 如果達到了核心數量且線程池是運行狀態,任務入隊列
    if (isRunning(c) && workQueue.offer(command)) {
        int recheck = ctl.get();
        // 再次檢查線程池狀態,如果不是運行狀態,就移除任務並執行拒絕策略
        if (! isRunning(recheck) && remove(command))
            reject(command);
        // 容錯檢查工作線程數量是否爲0,如果爲0就創建一個
        else if (workerCountOf(recheck) == 0)
            addWorker(null, false);
    }
    // 3. 任務入隊列失敗,嘗試創建非核心工作線程
    else if (!addWorker(command, false))
        // 非核心工作線程創建失敗,執行拒絕策略
        reject(command);
}

關於線程池狀態的內容,我們這裏不拿出來細講了,有興趣的可以看看《深讀源碼-java線程系列之線程池深入解析——生命週期》這章。

提交任務的過程大致如下:

(1)工作線程數量小於核心數量,創建核心線程;

(2)達到核心數量,進入任務隊列;

(3)任務隊列滿了,創建非核心線程;

(4)達到最大數量,執行拒絕策略;

其實,就是三道坎——核心數量、任務隊列、最大數量,這樣就比較好記了。

流程圖大致如下:

threadpool_task

任務流轉的過程我們知道了,但是任務是在哪裏執行的呢?繼續往下看。

addWorker()方法

這個方法主要用來創建一個工作線程,並啓動之,其中會做線程池狀態、工作線程數量等各種檢測。

private boolean addWorker(Runnable firstTask, boolean core) {
    // 判斷有沒有資格創建新的工作線程
    // 主要是一些狀態/數量的檢查等等
    // 這段代碼比較複雜,可以先跳過
    retry:
    for (;;) {
        int c = ctl.get();
        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))
                return false;
            // 數量加1並跳出循環
            if (compareAndIncrementWorkerCount(c))
                break retry;
            c = ctl.get();  // Re-read ctl
            if (runStateOf(c) != rs)
                continue retry;
            // else CAS failed due to workerCount change; retry inner loop
        }
    }
    
    // 如果上面的條件滿足,則會把工作線程數量加1,然後執行下面創建線程的動作
    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()) // precheck that t is startable
                        throw new IllegalThreadStateException();
                    
                    // 添加到工作線程隊列
                    workers.add(w);
                    // 還在池子中的線程數量(只能在mainLock中使用)
                    int s = workers.size();
                    if (s > largestPoolSize)
                        largestPoolSize = s;
                    
                    // 標記線程添加成功
                    workerAdded = true;
                }
            } finally {
                mainLock.unlock();
            }
            if (workerAdded) {
                // 線程添加成功之後啓動線程
                t.start();
                workerStarted = true;
            }
        }
    } finally {
        // 線程啓動失敗,執行失敗方法(線程數量減1,執行tryTerminate()方法等)
        if (! workerStarted)
            addWorkerFailed(w);
    }
    return workerStarted;
}

這裏其實還沒到任務執行的地方,上面我們可以看到線程是包含在Worker這個類中的,那麼,我們就跟蹤到這個類中看看。

Worker內部類

Worker內部類可以看作是對工作線程的包裝,一般地,我們說工作線程就是指Worker,但實際上是指其維護的Thread實例。

// Worker繼承自AQS,自帶鎖的屬性
private final class Worker
    extends AbstractQueuedSynchronizer
    implements Runnable
{
    // 真正工作的線程
    final Thread thread;
    // 第一個任務,從構造方法傳進來
    Runnable firstTask;
    // 完成任務數
    volatile long completedTasks;

    // 構造方法
    Worker(Runnable firstTask) {
        setState(-1); // inhibit interrupts until runWorker
        this.firstTask = firstTask;
        // 使用線程工廠生成一個線程
        // 注意,這裏把Worker本身作爲Runnable傳給線程
        this.thread = getThreadFactory().newThread(this);
    }

    // 實現Runnable的run()方法
    public void run() {
        // 調用ThreadPoolExecutor的runWorker()方法
        runWorker(this);
    }
    
    // 省略鎖的部分
}

這裏要能夠看出來工作線程Thread啓動的時候實際是調用的Worker的run()方法,進而調用的是ThreadPoolExecutor的runWorker()方法。

runWorker()方法

runWorker()方法是真正執行任務的地方。

final void runWorker(Worker w) {
    // 工作線程
    Thread wt = Thread.currentThread();
    // 任務
    Runnable task = w.firstTask;
    w.firstTask = null;
    // 強制釋放鎖(shutdown()裏面有加鎖)
    // 這裏相當於無視那邊的中斷標記
    w.unlock(); // allow interrupts
    boolean completedAbruptly = true;
    try {
        // 取任務,如果有第一個任務,這裏先執行第一個任務
        // 只要能取到任務,這就是個死循環
        // 正常來說getTask()返回的任務是不可能爲空的,因爲前面execute()方法是有空判斷的
        // 那麼,getTask()什麼時候纔會返回空任務呢?
        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置爲空,重新從隊列中取
                task = null;
                // 完成任務數加1
                w.completedTasks++;
                w.unlock();
            }
        }
        completedAbruptly = false;
    } finally {
        // 到這裏肯定是上面的while循環退出了
        processWorkerExit(w, completedAbruptly);
    }
}

這個方法比較簡單,忽略狀態檢測和鎖的內容,如果有第一個任務,就先執行之,之後再從任務隊列中取任務來執行,獲取任務是通過getTask()來進行的。

getTask()

從隊列中獲取任務的方法,裏面包含了對線程池狀態、空閒時間等的控制。

private Runnable getTask() {
    // 是否超時
    boolean timedOut = false;
    
    // 死循環
    for (;;) {
        int c = ctl.get();
        int rs = runStateOf(c);

        // 線程池狀態是SHUTDOWN的時候會把隊列中的任務執行完直到隊列爲空
        // 線程池狀態是STOP時立即退出
        if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
            decrementWorkerCount();
            return null;
        }
        
        // 工作線程數量
        int wc = workerCountOf(c);

        // 是否允許超時,有兩種情況:
        // 1. 是允許核心線程數超時,這種就是說所有的線程都可能超時
        // 2. 是工作線程數大於了核心數量,這種肯定是允許超時的
        // 注意,非核心線程是一定允許超時的,這裏的超時其實是指取任務超時
        boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;

        // 超時判斷(還包含一些容錯判斷)
        if ((wc > maximumPoolSize || (timed && timedOut))
            && (wc > 1 || workQueue.isEmpty())) {
            // 超時了,減少工作線程數量,並返回null
            if (compareAndDecrementWorkerCount(c))
                return null;
            // 減少工作線程數量失敗,則重試
            continue;
        }

        try {
            // 真正取任務的地方
            // 默認情況下,只有當工作線程數量大於核心線程數量時,纔會調用poll()方法觸發超時調用
            Runnable r = timed ?
                workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                workQueue.take();
            
            // 取到任務了就正常返回
            if (r != null)
                return r;
            // 沒取到任務表明超時了,回到continue那個if中返回null
            timedOut = true;
        } catch (InterruptedException retry) {
            // 捕獲到了中斷異常
            // 中斷標記是在調用shutDown()或者shutDownNow()的時候設置進去的
            // 此時,會回到for循環的第一個if處判斷狀態是否要返回null
            timedOut = false;
        }
    }
}

注意,這裏取任務會根據工作線程的數量判斷是使用BlockingQueue的poll(timeout, unit)方法還是take()方法。

poll(timeout, unit)方法會在超時時返回null,如果timeout<=0,隊列爲空時直接返回null。

take()方法會一直阻塞直到取到任務或拋出中斷異常。

所以,如果keepAliveTime設置爲0,當任務隊列爲空時,非核心線程取不出來任務,會立即結束其生命週期。

默認情況下,是不允許核心線程超時的,但是可以通過下面這個方法設置使核心線程也可超時。

public void allowCoreThreadTimeOut(boolean value) {
    if (value && keepAliveTime <= 0)
        throw new IllegalArgumentException("Core threads must have nonzero keep alive times");
    if (value != allowCoreThreadTimeOut) {
        allowCoreThreadTimeOut = value;
        if (value)
            interruptIdleWorkers();
    }
}

至此,線程池中任務的執行流程就結束了。

再看開篇問題

觀察num值的打印信息,先是打印了0~4,再打印了10~14,最後打印了5~9,竟然不是按順序打印的,爲什麼呢?

線程池的參數:核心數量5個,最大數量10個,任務隊列5個。

答:執行前5個任務執行時,正好還不到核心數量,所以新建核心線程並執行了他們;

執行中間的5個任務時,已達到核心數量,所以他們先入隊列;

執行後面5個任務時,已達核心數量且隊列已滿,所以新建非核心線程並執行了他們;

再執行最後5個任務時,線程池已達到滿負荷狀態,所以執行了拒絕策略。

總結

本章通過一個例子並結合線程池的重要方法我們一起分析了線程池中普通任務執行的流程。

(1)execute(),提交任務的方法,根據核心數量、任務隊列大小、最大數量,分成四種情況判斷任務應該往哪去;

(2)addWorker(),添加工作線程的方法,通過Worker內部類封裝一個Thread實例維護工作線程的執行;

(3)runWorker(),真正執行任務的地方,先執行第一個任務,再源源不斷從任務隊列中取任務來執行;

(4)getTask(),真正從隊列取任務的地方,默認情況下,根據工作線程數量與核心數量的關係判斷使用隊列的poll()還是take()方法,keepAliveTime參數也是在這裏使用的。

彩蛋

核心線程和非核心線程有什麼區別?

答:實際上並沒有什麼區別,主要是根據corePoolSize來判斷任務該去哪裏,兩者在執行任務的過程中並沒有任何區別。有可能新建的時候是核心線程,而keepAliveTime時間到了結束了的也可能是剛開始創建的核心線程。

Worker繼承自AQS有何意義?

前面我們看了Worker內部類的定義,它繼承自AQS,天生自帶鎖的特性,那麼,它的鎖是用來幹什麼的呢?跟任務的執行有關係嗎?

答:既然是跟鎖(同步)有關,說明Worker類跨線程使用了,此時我們查看它的lock()方法發現只在runWorker()方法中使用了,但是其tryLock()卻是在interruptIdleWorkers()方法中使用的。

private void interruptIdleWorkers(boolean onlyOne) {
    final ReentrantLock mainLock = this.mainLock;
    mainLock.lock();
    try {
        for (Worker w : workers) {
            Thread t = w.thread;
            if (!t.isInterrupted() && w.tryLock()) {
                try {
                    t.interrupt();
                } catch (SecurityException ignore) {
                } finally {
                    w.unlock();
                }
            }
            if (onlyOne)
                break;
        }
    } finally {
        mainLock.unlock();
    }
}

interruptIdleWorkers()方法的意思是中斷空閒線程的意思,它只會中斷BlockingQueue的poll()或take()方法,而不會中斷正在執行的任務。

一般來說,interruptIdleWorkers()方法的調用不是在本工作線程,而是在主線程中調用的,還記得《深讀源碼-java線程系列之線程池深入解析——生命週期》中說過的shutdown()和shutdownNow()方法嗎?

觀察兩個方法中中斷線程的方法,shutdown()中就是調用了interruptIdleWorkers()方法,這裏tryLock()獲取到鎖了再中斷,如果沒有獲取到鎖則不中斷,沒獲取到鎖只有一種情況,也就是lock()所在的地方,也就是有任務正在執行。

而shutdownNow()中中斷線程則很暴力,並沒有tryLock(),而是直接中斷了線程,所以調用shutdownNow()可能會中斷正在執行的任務。

所以,Worker繼承自AQS實際是要使用其鎖的能力,這個鎖主要是用來控制shutdown()時不要中斷正在執行任務的線程。


原文鏈接:https://www.cnblogs.com/tong-yuan/p/11787395.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章