Java多線程(十九): J.U.C 之線程池:ThreadPoolExecutor

作爲Executor框架中最核心的類,ThreadPoolExecutor代表着鼎鼎大名的線程池,它給了我們足夠的理由來弄清楚它。

下面我們就通過源碼來一步一步弄清楚它。

內部狀態

線程有五種狀態:新建,就緒,運行,阻塞,死亡,線程池同樣有五種狀態:Running, SHUTDOWN, STOP, TIDYING, TERMINATED。

 

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;

// runState is stored in the high-order bits
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;
private static final int TERMINATED =  3 << COUNT_BITS;

// Packing and unpacking ctl
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定義爲AtomicInteger ,其功能非常強大,記錄了“線程池中的任務數量”和“線程池的狀態”兩個信息。共32位,其中高3位表示"線程池狀態",低29位表示"線程池中的任務數量"。

 

RUNNING            -- 對應的高3位值是111。
SHUTDOWN       -- 對應的高3位值是000。
STOP                   -- 對應的高3位值是001。
TIDYING              -- 對應的高3位值是010。
TERMINATED     -- 對應的高3位值是011。

 

RUNNING:處於RUNNING狀態的線程池能夠接受新任務,以及對新添加的任務進行處理。

SHUTDOWN:處於SHUTDOWN狀態的線程池不可以接受新任務,但是可以對已添加的任務進行處理。

STOP:處於STOP狀態的線程池不接收新任務,不處理已添加的任務,並且會中斷正在處理的任務。

TIDYING:當所有的任務已終止,ctl記錄的"任務數量"爲0,線程池會變爲TIDYING狀態。當線程池變爲TIDYING狀態時,會執行鉤子函數terminated()。terminated()在ThreadPoolExecutor類中是空的,若用戶想在線程池變爲TIDYING時,進行相應的處理;可以通過重載terminated()函數來實現。

TERMINATED:線程池徹底終止的狀態。

各個狀態的轉換如下:

線程池狀態轉換

創建線程池

我們可以通過ThreadPoolExecutor構造函數來創建一個線程池:

 

public ThreadPoolExecutor(int corePoolSize,
                          int maximumPoolSize,
                          long keepAliveTime,
                          TimeUnit unit,
                          BlockingQueue<Runnable> workQueue,
                          ThreadFactory threadFactory,
                          RejectedExecutionHandler handler) {
    if (corePoolSize < 0 ||
        maximumPoolSize <= 0 ||
        maximumPoolSize < corePoolSize ||
        keepAliveTime < 0)
        throw new IllegalArgumentException();
    if (workQueue == null || threadFactory == null || handler == null)
        throw new NullPointerException();
    this.corePoolSize = corePoolSize;
    this.maximumPoolSize = maximumPoolSize;
    this.workQueue = workQueue;
    this.keepAliveTime = unit.toNanos(keepAliveTime);
    this.threadFactory = threadFactory;
    this.handler = handler;
}

 

共有七個參數,每個參數含義如下:

corePoolSize

線程池中核心線程的數量。當提交一個任務時,線程池會新建一個線程來執行任務,直到當前線程數等於corePoolSize。如果調用了線程池的prestartAllCoreThreads()方法,線程池會提前創建並啓動所有基本線程。

maximumPoolSize

線程池中允許的最大線程數。線程池的阻塞隊列滿了之後,如果還有任務提交,如果當前的線程數小於maximumPoolSize,則會新建線程來執行任務。注意,如果使用的是無界隊列,該參數也就沒有什麼效果了。

keepAliveTime

線程空閒的時間。線程的創建和銷燬是需要代價的。線程執行完任務後不會立即銷燬,而是繼續存活一段時間:keepAliveTime。默認情況下,該參數只有在線程數大於corePoolSize時纔會生效。

unit

keepAliveTime的單位。TimeUnit

workQueue

用來保存等待執行的任務的阻塞隊列,等待的任務必須實現Runnable接口。我們可以選擇如下幾種:

threadFactory

用於設置創建線程的工廠。該對象可以通過Executors.defaultThreadFactory(),如下:

 

public static ThreadFactory defaultThreadFactory() {
    return new DefaultThreadFactory();
}

 

返回的是DefaultThreadFactory對象,源碼如下:

 

static class DefaultThreadFactory implements ThreadFactory {
    private static final AtomicInteger poolNumber = new AtomicInteger(1);
    private final ThreadGroup group;
    private final AtomicInteger threadNumber = new AtomicInteger(1);
    private final String namePrefix;

    DefaultThreadFactory() {
        SecurityManager s = System.getSecurityManager();
        group = (s != null) ? s.getThreadGroup() :
                              Thread.currentThread().getThreadGroup();
        namePrefix = "pool-" +
                      poolNumber.getAndIncrement() +
                     "-thread-";
    }

    public Thread newThread(Runnable r) {
        Thread t = new Thread(group, r,
                              namePrefix + threadNumber.getAndIncrement(),
                              0);
        if (t.isDaemon())
            t.setDaemon(false);
        if (t.getPriority() != Thread.NORM_PRIORITY)
            t.setPriority(Thread.NORM_PRIORITY);
        return t;
    }
}

 

ThreadFactory的左右就是提供創建線程的功能的線程工廠。他是通過newThread()方法提供創建線程的功能,newThread()方法創建的線程都是“非守護線程”而且“線程優先級都是Thread.NORM_PRIORITY”。

handler

RejectedExecutionHandler,線程池的拒絕策略。所謂拒絕策略,是指將任務添加到線程池中時,線程池拒絕該任務所採取的相應策略。當向線程池中提交任務時,如果此時線程池中的線程已經飽和了,而且阻塞隊列也已經滿了,則線程池會選擇一種拒絕策略來處理該任務。

線程池提供了四種拒絕策略:

  1. AbortPolicy:直接拋出異常,默認策略;
  2. CallerRunsPolicy:用調用者所在的線程來執行任務;
  3. DiscardOldestPolicy:丟棄阻塞隊列中靠最前的任務,並執行當前任務;
  4. DiscardPolicy:直接丟棄任務; 當然我們也可以實現自己的拒絕策略,例如記錄日誌等等,實現RejectedExecutionHandler接口即可。

線程池

Executor框架提供了三種線程池,他們都可以通過工具類Executors來創建。

FixedThreadPool

FixedThreadPool,可重用固定線程數的線程池,其定義如下:

 

public static ExecutorService newFixedThreadPool(int nThreads) {
    return new ThreadPoolExecutor(nThreads, nThreads,
                                  0L, TimeUnit.MILLISECONDS,
                                  new LinkedBlockingQueue<Runnable>());
}

 

corePoolSize 和 maximumPoolSize都設置爲創建FixedThreadPool時指定的參數nThreads,意味着當線程池滿時且阻塞隊列也已經滿時,如果繼續提交任務,則會直接走拒絕策略,該線程池不會再新建線程來執行任務,而是直接走拒絕策略。FixedThreadPool使用的是默認的拒絕策略,即AbortPolicy,則直接拋出異常。

keepAliveTime設置爲0L,表示空閒的線程會立刻終止。

workQueue則是使用LinkedBlockingQueue,但是沒有設置範圍,那麼則是最大值(Integer.MAX_VALUE),這基本就相當於一個無界隊列了。使用該“無界隊列”則會帶來哪些影響呢?當線程池中的線程數量等於corePoolSize 時,如果繼續提交任務,該任務會被添加到阻塞隊列workQueue中,當阻塞隊列也滿了之後,則線程池會新建線程執行任務直到maximumPoolSize。由於FixedThreadPool使用的是“無界隊列”LinkedBlockingQueue,那麼maximumPoolSize參數無效,同時指定的拒絕策略AbortPolicy也將無效。而且該線程池也不會拒絕提交的任務,如果客戶端提交任務的速度快於任務的執行,那麼keepAliveTime也是一個無效參數。

其運行圖如下(參考《Java併發編程的藝術》):

FixedThreadPoolSingleThreadExecutor

SingleThreadExecutor是使用單個worker線程的Executor,定義如下:

 

public static ExecutorService newSingleThreadExecutor() {
    return new FinalizableDelegatedExecutorService
        (new ThreadPoolExecutor(1, 1,
                                0L, TimeUnit.MILLISECONDS,
                                new LinkedBlockingQueue<Runnable>()));
}

 

作爲單一worker線程的線程池,SingleThreadExecutor把corePool和maximumPoolSize均被設置爲1,和FixedThreadPool一樣使用的是無界隊列LinkedBlockingQueue,所以帶來的影響和FixedThreadPool一樣。

SingleThreadExecutorCachedThreadPool

CachedThreadPool是一個會根據需要創建新線程的線程池 ,他定義如下:

 

public static ExecutorService newCachedThreadPool() {
    return new ThreadPoolExecutor(0, Integer.MAX_VALUE,
                                  60L, TimeUnit.SECONDS,
                                  new SynchronousQueue<Runnable>());
}

 

CachedThreadPool的corePool爲0,maximumPoolSize爲Integer.MAX_VALUE,這就意味着所有的任務一提交就會加入到阻塞隊列中。keepAliveTime這是爲60L,unit設置爲TimeUnit.SECONDS,意味着空閒線程等待新任務的最長時間爲60秒,空閒線程超過60秒後將會被終止。阻塞隊列採用的SynchronousQueue,而我們在【死磕Java併發】----J.U.C之阻塞隊列:SynchronousQueue中瞭解到SynchronousQueue是一個沒有元素的阻塞隊列,加上corePool = 0 ,maximumPoolSize = Integer.MAX_VALUE,這樣就會存在一個問題,如果主線程提交任務的速度遠遠大於CachedThreadPool的處理速度,則CachedThreadPool會不斷地創建新線程來執行任務,這樣有可能會導致系統耗盡CPU和內存資源,所以在使用該線程池是,一定要注意控制併發的任務數,否則創建大量的線程可能導致嚴重的性能問題

SouthEastuploading.4e448015.gif轉存失敗重新上傳取消CachedThreadPool

任務提交

線程池根據業務不同的需求提供了兩種方式提交任務:Executor.execute()、ExecutorService.submit()。其中ExecutorService.submit()可以獲取該任務執行的Future。 我們以Executor.execute()爲例,來看看線程池的任務提交經歷了那些過程。

定義:

 

public interface Executor {

    void execute(Runnable command);
}

 

ThreadPoolExecutor提供實現:

 

public void execute(Runnable command) {
    if (command == null)
        throw new NullPointerException();
    int c = ctl.get();
    if (workerCountOf(c) < corePoolSize) {
        if (addWorker(command, true))
            return;
        c = ctl.get();
    }
    if (isRunning(c) && workQueue.offer(command)) {
        int recheck = ctl.get();
        if (! isRunning(recheck) && remove(command))
            reject(command);
        else if (workerCountOf(recheck) == 0)
            addWorker(null, false);
    }
    else if (!addWorker(command, false))
        reject(command);
}

 

執行流程如下:

  1. 如果線程池當前線程數小於corePoolSize,則調用addWorker創建新線程執行任務,成功返回true,失敗執行步驟2。
  2. 如果線程池處於RUNNING狀態,則嘗試加入阻塞隊列,如果加入阻塞隊列成功,則嘗試進行Double Check,如果加入失敗,則執行步驟3。
  3. 如果線程池不是RUNNING狀態或者加入阻塞隊列失敗,則嘗試創建新線程直到maxPoolSize,如果失敗,則調用reject()方法運行相應的拒絕策略。

在步驟2中如果加入阻塞隊列成功了,則會進行一個Double Check的過程。Double Check過程的主要目的是判斷加入到阻塞隊裏中的線程是否可以被執行。如果線程池不是RUNNING狀態,則調用remove()方法從阻塞隊列中刪除該任務,然後調用reject()方法處理任務。否則需要確保還有線程執行。

addWorker 當線程中的當前線程數小於corePoolSize,則調用addWorker()創建新線程執行任務,當前線程數則是根據ctl變量來獲取的,調用workerCountOf(ctl)獲取低29位即可:

 

private static int workerCountOf(int c)  { return c & CAPACITY; }

 

addWorker(Runnable firstTask, boolean core)方法用於創建線程執行任務,源碼如下:

 

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;

        // 內層循環,worker + 1
        for (;;) {
            // 線程數量
            int wc = workerCountOf(c);
            // 如果當前線程數大於線程最大上限CAPACITY  return false
            // 若core == true,則與corePoolSize 比較,否則與maximumPoolSize ,大於 return false
            if (wc >= CAPACITY ||
                    wc >= (core ? corePoolSize : maximumPoolSize))
                return false;
            // worker + 1,成功跳出retry循環
            if (compareAndIncrementWorkerCount(c))
                break retry;

            // CAS add worker 失敗,再次讀取ctl
            c = ctl.get();

            // 如果狀態不等於之前獲取的state,跳出內層循環,繼續去外層循環判斷
            if (runStateOf(c) != rs)
                continue retry;
        }
    }

    boolean workerStarted = false;
    boolean workerAdded = false;
    Worker w = null;
    try {

        // 新建線程:Worker
        w = new Worker(firstTask);
        // 當前線程
        final Thread t = w.thread;
        if (t != null) {
            // 獲取主鎖:mainLock
            final ReentrantLock mainLock = this.mainLock;
            mainLock.lock();
            try {

                // 線程狀態
                int rs = runStateOf(ctl.get());

                // rs < SHUTDOWN ==> 線程處於RUNNING狀態
                // 或者線程處於SHUTDOWN狀態,且firstTask == null(可能是workQueue中仍有未執行完成的任務,創建沒有初始任務的worker線程執行)
                if (rs < SHUTDOWN ||
                        (rs == SHUTDOWN && firstTask == null)) {

                    // 當前線程已經啓動,拋出異常
                    if (t.isAlive()) // precheck that t is startable
                        throw new IllegalThreadStateException();

                    // workers是一個HashSet<Worker>
                    workers.add(w);

                    // 設置最大的池大小largestPoolSize,workerAdded設置爲true
                    int s = workers.size();
                    if (s > largestPoolSize)
                        largestPoolSize = s;
                    workerAdded = true;
                }
            } finally {
                // 釋放鎖
                mainLock.unlock();
            }
            // 啓動線程
            if (workerAdded) {
                t.start();
                workerStarted = true;
            }
        }
    } finally {

        // 線程啓動失敗
        if (! workerStarted)
            addWorkerFailed(w);
    }
    return workerStarted;
}

 

  1. 判斷當前線程是否可以添加任務,如果可以則進行下一步,否則return false;
    1. rs >= SHUTDOWN ,表示當前線程處於SHUTDOWN ,STOP、TIDYING、TERMINATED狀態
    2. rs == SHUTDOWN , firstTask != null時不允許添加線程,因爲線程處於SHUTDOWN 狀態,不允許添加任務
    3. rs == SHUTDOWN , firstTask == null,但workQueue.isEmpty() == true,不允許添加線程,因爲firstTask == null是爲了添加一個沒有任務的線程然後再從workQueue中獲取任務的,如果workQueue == null,則說明添加的任務沒有任何意義。
  2. 內嵌循環,通過CAS worker + 1
  3. 獲取主鎖mailLock,如果線程池處於RUNNING狀態獲取處於SHUTDOWN狀態且 firstTask == null,則將任務添加到workers Queue中,然後釋放主鎖mainLock,然後啓動線程,然後return true,如果中途失敗導致workerStarted= false,則調用addWorkerFailed()方法進行處理。

在這裏需要好好理論addWorker中的參數,在execute()方法中,有三處調用了該方法: 第一次:workerCountOf(c) < corePoolSize ==> addWorker(command, true),這個很好理解,當然線程池的線程數量小於 corePoolSize ,則新建線程執行任務即可,在執行過程core == true,內部與corePoolSize比較即可。 第二次:加入阻塞隊列進行Double Check時,else if (workerCountOf(recheck) == 0) ==>addWorker(null, false)。如果線程池中的線程==0,按照道理應該該任務應該新建線程執行任務,但是由於已經該任務已經添加到了阻塞隊列,那麼就在線程池中新建一個空線程,然後從阻塞隊列中取線程即可。 第三次:線程池不是RUNNING狀態或者加入阻塞隊列失敗:else if (!addWorker(command, false)),這裏core == fase,則意味着是與maximumPoolSize比較。

在新建線程執行任務時,將講Runnable包裝成一個Worker,Woker爲ThreadPoolExecutor的內部類

Woker內部類

Woker的源碼如下:

 

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

    // task 的thread
    final Thread thread;

    // 運行的任務task
    Runnable firstTask;

    volatile long completedTasks;

    Worker(Runnable firstTask) {

        //設置AQS的同步狀態private volatile int state,是一個計數器,大於0代表鎖已經被獲取
        setState(-1);
        this.firstTask = firstTask;

        // 利用ThreadFactory和 Worker這個Runnable創建的線程對象
        this.thread = getThreadFactory().newThread(this);
    }

    // 任務執行
    public void run() {
        runWorker(this);
    }

}

 

從Worker的源碼中我們可以看到Woker繼承AQS,實現Runnable接口,所以可以認爲Worker既是一個可以執行的任務,也可以達到獲取鎖釋放鎖的效果。這裏繼承AQS主要是爲了方便線程的中斷處理。這裏注意兩個地方:構造函數、run()。構造函數主要是做三件事:1.設置同步狀態state爲-1,同步狀態大於0表示就已經獲取了鎖,2.設置將當前任務task設置爲firstTask,3.利用Worker本身對象this和ThreadFactory創建線程對象。

當線程thread啓動(調用start()方法)時,其實就是執行Worker的run()方法,內部調用runWorker()。

runWorker

 

final void runWorker(Worker w) {

    // 當前線程
    Thread wt = Thread.currentThread();

    // 要執行的任務
    Runnable task = w.firstTask;

    w.firstTask = null;

    // 釋放鎖,運行中斷
    w.unlock(); // allow interrupts
    boolean completedAbruptly = true;
    try {
        while (task != null || (task = getTask()) != null) {
            // worker 獲取鎖
            w.lock();

            // 確保只有當線程是stoping時,纔會被設置爲中斷,否則清楚中斷標示
            // 如果線程池狀態 >= STOP ,且當前線程沒有設置中斷狀態,則wt.interrupt()
            // 如果線程池狀態 < STOP,但是線程已經中斷了,再次判斷線程池是否 >= STOP,如果是 wt.interrupt()
            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;
                // 完成任務數 + 1
                w.completedTasks++;
                // 釋放鎖
                w.unlock();
            }
        }
        completedAbruptly = false;
    } finally {
        processWorkerExit(w, completedAbruptly);
    }
}

 

運行流程

  1. 根據worker獲取要執行的任務task,然後調用unlock()方法釋放鎖,這裏釋放鎖的主要目的在於中斷,因爲在new Worker時,設置的state爲-1,調用unlock()方法可以將state設置爲0,這裏主要原因就在於interruptWorkers()方法只有在state >= 0時纔會執行;
  2. 通過getTask()獲取執行的任務,調用task.run()執行,當然在執行之前會調用worker.lock()上鎖,執行之後調用worker.unlock()放鎖;
  3. 在任務執行前後,可以根據業務場景自定義beforeExecute() 和 afterExecute()方法,則兩個方法在ThreadPoolExecutor中是空實現;
  4. 如果線程執行完成,則會調用getTask()方法從阻塞隊列中獲取新任務,如果阻塞隊列爲空,則根據是否超時來判斷是否需要阻塞;
  5. task == null或者拋出異常(beforeExecute()、task.run()、afterExecute()均有可能)導致worker線程終止,則調用processWorkerExit()方法處理worker退出流程。

getTask()

 

private Runnable getTask() {
    boolean timedOut = false; // Did the last poll() time out?

    for (;;) {

        // 線程池狀態
        int c = ctl.get();
        int rs = runStateOf(c);

        // 線程池中狀態 >= STOP 或者 線程池狀態 == SHUTDOWN且阻塞隊列爲空,則worker - 1,return null
        if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
            decrementWorkerCount();
            return null;
        }

        int wc = workerCountOf(c);

        // 判斷是否需要超時控制
        boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;

        if ((wc > maximumPoolSize || (timed && timedOut)) && (wc > 1 || workQueue.isEmpty())) {
            if (compareAndDecrementWorkerCount(c))
                return null;
            continue;
        }

        try {

            // 從阻塞隊列中獲取task
            // 如果需要超時控制,則調用poll(),否則調用take()
            Runnable r = timed ?
                    workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                    workQueue.take();
            if (r != null)
                return r;
            timedOut = true;
        } catch (InterruptedException retry) {
            timedOut = false;
        }
    }
}

 

timed == true,調用poll()方法,如果在keepAliveTime時間內還沒有獲取task的話,則返回null,繼續循環。timed == false,則調用take()方法,該方法爲一個阻塞方法,沒有任務時會一直阻塞掛起,直到有任務加入時對該線程喚醒,返回任務。

在runWorker()方法中,無論最終結果如何,都會執行processWorkerExit()方法對worker進行退出處理。

processWorkerExit()

 

private void processWorkerExit(Worker w, boolean completedAbruptly) {

    // true:用戶線程運行異常,需要扣減
    // false:getTask方法中扣減線程數量
    if (completedAbruptly)
        decrementWorkerCount();

    // 獲取主鎖
    final ReentrantLock mainLock = this.mainLock;
    mainLock.lock();
    try {
        completedTaskCount += w.completedTasks;
        // 從HashSet中移出worker
        workers.remove(w);
    } finally {
        mainLock.unlock();
    }

    // 有worker線程移除,可能是最後一個線程退出需要嘗試終止線程池
    tryTerminate();

    int c = ctl.get();
    // 如果線程爲running或shutdown狀態,即tryTerminate()沒有成功終止線程池,則判斷是否有必要一個worker
    if (runStateLessThan(c, STOP)) {
        // 正常退出,計算min:需要維護的最小線程數量
        if (!completedAbruptly) {
            // allowCoreThreadTimeOut 默認false:是否需要維持核心線程的數量
            int min = allowCoreThreadTimeOut ? 0 : corePoolSize;
            // 如果min ==0 或者workerQueue爲空,min = 1
            if (min == 0 && ! workQueue.isEmpty())
                min = 1;

            // 如果線程數量大於最少數量min,直接返回,不需要新增線程
            if (workerCountOf(c) >= min)
                return; // replacement not needed
        }
        // 添加一個沒有firstTask的worker
        addWorker(null, false);
    }
}

 

首先completedAbruptly的值來判斷是否需要對線程數-1處理,如果completedAbruptly == true,說明在任務運行過程中出現了異常,那麼需要進行減1處理,否則不需要,因爲減1處理在getTask()方法中處理了。然後從HashSet中移出該worker,過程需要獲取mainlock。然後調用tryTerminate()方法處理,該方法是對最後一個線程退出做終止線程池動作。如果線程池沒有終止,那麼線程池需要保持一定數量的線程,則通過addWorker(null,false)新增一個空的線程。

addWorkerFailed()

在addWorker()方法中,如果線程t==null,或者在add過程出現異常,會導致workerStarted == false,那麼在最後會調用addWorkerFailed()方法:

 

private void addWorkerFailed(Worker w) {
    final ReentrantLock mainLock = this.mainLock;
    mainLock.lock();
    try {
        // 從HashSet中移除該worker
        if (w != null)
            workers.remove(w);

        // 線程數 - 1
        decrementWorkerCount();
        // 嘗試終止線程
        tryTerminate();
    } finally {
        mainLock.unlock();
    }
}

 

整個邏輯顯得比較簡單。

tryTerminate()

當線程池涉及到要移除worker時候都會調用tryTerminate(),該方法主要用於判斷線程池中的線程是否已經全部移除了,如果是的話則關閉線程池。

 

final void tryTerminate() {
    for (;;) {
        int c = ctl.get();
        // 線程池處於Running狀態
        // 線程池已經終止了
        // 線程池處於ShutDown狀態,但是阻塞隊列不爲空
        if (isRunning(c) ||
                runStateAtLeast(c, TIDYING) ||
                (runStateOf(c) == SHUTDOWN && ! workQueue.isEmpty()))
            return;

        // 執行到這裏,就意味着線程池要麼處於STOP狀態,要麼處於SHUTDOWN且阻塞隊列爲空
        // 這時如果線程池中還存在線程,則會嘗試中斷線程
        if (workerCountOf(c) != 0) {
            // /線程池還有線程,但是隊列沒有任務了,需要中斷喚醒等待任務的線程
            // (runwoker的時候首先就通過w.unlock設置線程可中斷,getTask最後面的catch處理中斷)
            interruptIdleWorkers(ONLY_ONE);
            return;
        }

        final ReentrantLock mainLock = this.mainLock;
        mainLock.lock();
        try {
            // 嘗試終止線程池
            if (ctl.compareAndSet(c, ctlOf(TIDYING, 0))) {
                try {
                    terminated();
                } finally {
                    // 線程池狀態轉爲TERMINATED
                    ctl.set(ctlOf(TERMINATED, 0));
                    termination.signalAll();
                }
                return;
            }
        } finally {
            mainLock.unlock();
        }
    }
}

 

在關閉線程池的過程中,如果線程池處於STOP狀態或者處於SHUDOWN狀態且阻塞隊列爲null,則線程池會調用interruptIdleWorkers()方法中斷所有線程,注意ONLY_ONE== true,表示僅中斷一個線程。

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();
    }
}

 

onlyOne==true僅終止一個線程,否則終止所有線程。

線程終止

線程池ThreadPoolExecutor提供了shutdown()和shutDownNow()用於關閉線程池。

shutdown():按過去執行已提交任務的順序發起一個有序的關閉,但是不接受新任務。

shutdownNow() :嘗試停止所有的活動執行任務、暫停等待任務的處理,並返回等待執行的任務列表。

shutdown

 

public void shutdown() {
    final ReentrantLock mainLock = this.mainLock;
    mainLock.lock();
    try {
        checkShutdownAccess();
        // 推進線程狀態
        advanceRunState(SHUTDOWN);
        // 中斷空閒的線程
        interruptIdleWorkers();
        // 交給子類實現
        onShutdown();
    } finally {
        mainLock.unlock();
    }
    tryTerminate();
}

 

shutdownNow

 

public List<Runnable> shutdownNow() {
    List<Runnable> tasks;
    final ReentrantLock mainLock = this.mainLock;
    mainLock.lock();
    try {
        checkShutdownAccess();
        advanceRunState(STOP);
        // 中斷所有線程
        interruptWorkers();
        // 返回等待執行的任務列表
        tasks = drainQueue();
    } finally {
        mainLock.unlock();
    }
    tryTerminate();
    return tasks;
}

 

與shutdown不同,shutdownNow會調用interruptWorkers()方法中斷所有線程。

 

private void interruptWorkers() {
    final ReentrantLock mainLock = this.mainLock;
    mainLock.lock();
    try {
        for (Worker w : workers)
            w.interruptIfStarted();
    } finally {
        mainLock.unlock();
    }
}

 

同時會調用drainQueue()方法返回等待執行到任務列表。

 

private List<Runnable> drainQueue() {
    BlockingQueue<Runnable> q = workQueue;
    ArrayList<Runnable> taskList = new ArrayList<Runnable>();
    q.drainTo(taskList);
    if (!q.isEmpty()) {
        for (Runnable r : q.toArray(new Runnable[0])) {
            if (q.remove(r))
                taskList.add(r);
        }
    }
    return taskList;
}

 

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