Worker
先前,笔者讲解到ThreadPoolExecutor.addWorker(Runnable firstTask, boolean core),在这个方法中工作线程可能创建成功,也可能创建失败,具体视线程池的边界条件,以及当前内存情况而定。
那么,如果线程池当前的状态,是允许创建Worker对象的,那么创建Worker的内部流程又是怎样呢?线程池为何要使用Worker包装Thread来创建一个线程,为何不直接使用原生的Thread来创建线程?如果创建Worker的firstTask不为空,那么Worker理所当然应该优先执行firstTask任务,如果firstTask为空,那Worker又要如何获取任务来执行呢?我们还有一堆亟待解决的问题。
首先我们来解决前两个问题,Worker的创建流程,以及为什么不使用原生Thread代替Worker?首先,Doug Lea用Worker包装Thread,意味着Worker比Thread拥有更多的功能。例如:Worker会统计它所对应的线程执行了多少任务、通过Worker可以知道线程是否已启动、线程是否正在执行任务?而这些信息都是原生Thread所没有的,所以需要一个Worker类来扩展Thread。
在创建Worker时,会先设置其state的值为-1,代表Worker所对应的线程尚未启动,即还没有调用Worker.thread.start(),之后会进行firstTask的赋值,向线程工厂申请创建线程,创建完毕后,等待外部调用Worker.thread.start()启动一个线程执行Worker.run()方法。
private final class Worker extends AbstractQueuedSynchronizer implements Runnable { /* * 当初始化一个Worker时,会向线程工厂申请创建一个Thread对象 * 用来执行任务,为null代表线程工厂创建失败。 */ final Thread thread; //首要执行任务,该字段可能为null。 Runnable firstTask; //thread已完成任务数。 volatile long completedTasks; Worker(Runnable firstTask) { /* * state初始为-1,代表还未调用Worker.thread.start(), * Worker对应的线程尚未被创建,还不能中断。线程启动后, * 如果线程正在执行任务,state为1,如果线程启动后没在 * 执行任务的状态则state为0。 */ setState(-1);//<1> this.firstTask = firstTask; /* * 创建Thread对象的时候,会把Worker对象本身传入,而Worker * 本身实现了Runnable接口,当调用thead.start()启动一个线程 * 执行thread.run()时,会进而调用Worker.run()方法。 */ this.thread = getThreadFactory().newThread(this); } //Worker对象将任务的执行委托给ThreadPoolExecutor.runWorker(Worker w). public void run() { runWorker(this); } //判断Worker是否处于被某个线程持有状态。 protected boolean isHeldExclusively() { return getState() != 0;//<2> } /* * 线程尝试持有worker对象,如果worker没有被某个线程 * 持有,则state为0,则用CAS的方式将worker的state * 改为1,并设置exclusiveOwnerThread为当前线程。 */ protected boolean tryAcquire(int unused) { if (compareAndSetState(0, 1)) {//<3> setExclusiveOwnerThread(Thread.currentThread());//<4> return true; } return false; } /* * 线程释放worker对象,将state改为0,exclusiveOwnerThread * 改为null。 */ protected boolean tryRelease(int unused) { setExclusiveOwnerThread(null); setState(0); return true; } /* * 调用父类acquire(int arg)时,会进而调用到Worker本身实现的 * tryAcquire(int unused)。 */ public void lock() { acquire(1);//<5> } public boolean tryLock() { return tryAcquire(1); } /* * 调用父类的release(int arg)时,会进而调用到Worker本身实现的 * tryRelease(int unused)。 */ public void unlock() { release(1);//<6> } public boolean isLocked() { return isHeldExclusively(); } /* * 尝试中断worker的对应线程,如果线程已经启动。创建 * worker时,state为-1,直到调用worker.thread.start() * 后,worker的state为0,如果worker的state>=0,则尝试 * 中断线程。 */ void interruptIfStarted() { Thread t; if (getState() >= 0 && (t = thread) != null && !t.isInterrupted()) { try { t.interrupt(); } catch (SecurityException ignore) { } } } }
从上面的代码我们可以注意到,<1>、<2>、<3>、<4>、<5>、<6>处的方法并不是Worker本身有的方法,而是Worker继承自父类AbstractQueuedSynchronizer的方法。那么Worker为什么需要继承AbstractQueuedSynchronizer(AQS)?AQS又是何方神圣呢?
这里先简单介绍下AQS,它定义了若干接口交由程序员实现,诸如:lock()、unlock()、tryAcquire(int arg)、tryRelease(int arg)……等,以保证多个线程不会同时访问同一资源。ThreadPoolExecutor中的字段mainLock为可重入锁ReentrantLock,某种程度上来说也是实现了AQS,ThreadPoolExecutor通过mainLock的lock()、unlock()以保证线程池内一些非线程安全的对象不会出现并发读写,如:workers、completedTaskCount……等。
public class ReentrantLock implements Lock, java.io.Serializable { private final Sync sync; abstract static class Sync extends AbstractQueuedSynchronizer {//...} static final class NonfairSync extends Sync {//...} public ReentrantLock() { sync = new NonfairSync(); } //... }
那么Worker继承了AQS,Worker也实现了lock()、unlock()方法,说明Worker本身也存在多线程访问的可能,那是什么时候会出现多线程访问Worker呢?这里我们先按下这个问题,在介绍完Worker.run()之后你就会明白为何Worker要继承AQS以保证线程访问的顺序性。
我们知道当调用Worker.thread.start()方法时,会进而调用Worker.run()方法,而Worker.run()方法会进而调用ThreadPoolExecutor.runWorker(Worker w)。下面,我们来看看runWorker(Worker w)的执行流程:
final void runWorker(Worker w) { Thread wt = Thread.currentThread(); Runnable task = w.firstTask; /* * 将worker.firstTask设置为null,因为worker可能会存在较长的一段时间, * 而task可能很快执行完毕,避免worker长时间引用已完成task,以便GC回收 * 已完成task。 */ w.firstTask = null; /* * 新创建的worker对象state为-1,调用worker.unlock()会进而调用 * worker.tryRelease(int unused),将state设置为0,代表worker * 对应工作线程已启动,线程处于可中断状态。 */ w.unlock(); // allow interrupts /* * 标志worker是否异常完成,如果在<1>处task为空,且无法通过 * getTask()从任务队列中获取新的任务,则会跳出循环,并在<3>处 * 赋值为false,代表worker并没有异常完成。 * 不管worker是正常完成还是异常完成,最后都会将completedAbruptly的结果 * 传给processWorkerExit(...),如果是worker是正常退出,则将workerCount-1, * 并将worker从workers集合中移除。如果是异常退出,则不减少workerCount,仅仅 * 是将异常worker从workers集合中移除,并尝试新增一个worker。 */ boolean completedAbruptly = true; try { /* * 如果task不为空,或者调用getTask()能任务队列中获取到新的任务, * 则进入while块的代码。如果任务队列中没有待执行的任务,调用getTask() * 会让当前线程陷入阻塞,直到超时或者有新任务进入任务队列。 */ while (task != null || (task = getTask()) != null) {//<1> /* * 如果能进入循环,代表worker准备开始执行任务,但在执行任务 * 前会先上锁,等到任务执行结束又会在<2>处释放锁,然而线程池 * 又不会让多个线程同时执行同一个任务,那么为什么在执行任务前 * 要让worker先上锁,执行完毕再释放锁呢? * 我们假设有一个线程池有5个线程,其中A、B线程正在执行任务, * C、D、E处于空闲状态。所以我们能确定A、B两个worker已经 * 获得了锁,而C、D、E还阻塞在getTask()方法中。现在线程池 * 执行shutdown()方法,该方法会进而调用interruptIdleWorkers() * 中断处于空闲状态的工作线程。而在interruptIdleWorkers()方法中 * 判断一个worker是否处于空闲,会调用worker.tryLock(),如果能 * 成功获取到锁,则代表该worker处于空闲状态,则中断该worker对应的 * 线程。因此,一个worker是有可能被多个线程访问的,比如worker本身 * 对应的线程,又或者关闭线程池的线程。 */ w.lock(); /* * 如果线程池的运行状态>=STOP,则中断当前线程。如果运行状态<STOP, * 则确保线程没有被中断。 */ if ((runStateAtLeast(ctl.get(), STOP) || (Thread.interrupted() && runStateAtLeast(ctl.get(), STOP))) && !wt.isInterrupted()) wt.interrupt(); try { //空方法,在执行任务前执行 beforeExecute(wt, task); try { task.run();//开始执行任务 afterExecute(task, null);//空方法,在执行任务完毕后执行。 } catch (Throwable ex) { afterExecute(task, ex); throw ex; } } finally { /* * 设置当前任务为空,这样就可以在下一次循环中获取新的任务。 * 对worker执行的任务数+1,并释放锁。 */ task = null; w.completedTasks++; w.unlock();//<2> } } completedAbruptly = false;//<3> } finally { processWorkerExit(w, completedAbruptly); } }
在上面的runWorker(Worker w)中如果执行完worker的首要任务,或者首要任务为null,便会调用getTask()尝试从任务队列中获取任务,但调用getTask()可能会使当前线程陷入等待或者阻塞直到有任务入队。getTask()也有可能返回null导致当前worker对应的线程退出,有以下几个原因可能导致工作线程退出:
- 如果调用setMaximumPoolSize(int maximumPoolSize)改小最大线程数,导致工作线程数大于maximumPoolSize。
- 线程池运行状态为STOP。
- 线程池运行状态为SHUTDOWN,且队列为空。
- 线程等待任务超时后,如果线程池当前存在可回收的空闲线程(即allowCoreThreadTimeOut为true或者工作线程数大于核心线程数),如果队列为空,则可直接退出,如果队列不为空,工作线程数必须大于1,即线程池中最少两个工作线程,如果只有一个线程还退出的话,就会存在队列不为空,但线程池中没有一个工作线程的尴尬情况。
private Runnable getTask() { //超时标志,默认为false,获取任务如果超时则会在<5>赋值为true。 boolean timedOut = false; for (; ; ) { int c = ctl.get(); /* * 如果线程池处于RUNNING状态,则runStateAtLeast(c, SHUTDOWN) * 为false,不会再判断之后的逻辑为true或者false。 * 如果线程池处于SHUTDOWN状态,且workQueue.isEmpty()为true,即 * 任务队列为空,则直接返回。如果任务队列不为空,则无法进入if分支, * 依然要返回任务,按照SHUTDOWN的要求不再接受新任务,但仍要处理队列 * 中的任务。 * 如果线程池处于STOP状态,即便任务队列不为空,也不再处理,则直接进入 * if分支后返回。 * 所以总结一下,只有两种情况不会进入此分支: * 1.线程池处于RUNNING状态。 * 2.线程池处于SHUTDOWN状态且任务队列不为空。 */ if (runStateAtLeast(c, SHUTDOWN) && (runStateAtLeast(c, STOP) || workQueue.isEmpty())) {//<1> decrementWorkerCount(); return null; } int wc = workerCountOf(c); /* * timed决定是如果任务队列没有任务的话,是以无限期的方式 * 等待任务入队,或者一旦等待时间超过keepAliveTime,则 * 返回null。 * 如果allowCoreThreadTimeOut为true,则核心线程等待 * 任务时间超过keepAliveTime后会被回收。 * 如果当前工作线程数量workerCount大于核心线程数,也会在 * 线程等待任务超过keepAliveTime后回收线程。 */ boolean timed = allowCoreThreadTimeOut || wc > corePoolSize; /* * 我们先看进入此分支后会做的事,再分析如何进入这个分支。进入 * 此分支后,会用CAS减少workerCount的数量,成功则返回null。 * 否则continue重新开始新一轮的for循环。 * 现在,我们来分析下进入此分支的逻辑: * 首先是wc > maximumPoolSize,一般workerCount不会大于 * maximumPoolSize,除非线程池运行期间通过 * setMaximumPoolSize(int maximumPoolSize)将线程池 * 的最大线程数改小。 * (timed && timedOut)在第一轮for循环永远为false,因为 * timedOut要为true的条件,首先是timed为true,即线程池内存在空闲后 * 可回收的线程,不管是线程池允许回收核心线程,或者线程数大于核心线程数。 * 只有在<4>获取任务超时后workQueue返回null,才有可能到达<5>处将 * timedOut赋值为true,并且开始新一轮的循环。 * 之后的两个判断wc>1和workQueue.isEmpty(),判断队列为空还好理解, * 为什么要判断wc>1?首先我们要知道maximumPoolSize必须大于等于1,当我们 * 往线程池传入的maximumPoolSize<=0会抛出异常。其次,如果我们将<2>处改为: * (wc >= 1 || workQueue.isEmpty()),有可能出现线程池内只有一个线程, * 但任务队列不为null。依旧会进入此分支内部执行<3>的代码以CAS的方式对 * workerCount-1,从而出现一个尴尬的情况,任务队列中有任务,但工作线程数 * 为0。所以<2>处必须保证wc>1。 * 思考一种情况:假设一个线程池核心线程数为3,最大线程数为5, * allowCoreThreadTimeOut为false,线程池当前工作线程数量也为5,5个线程 * 同时完成任务,并执行getTask()获取任务,可想而知timed为true,因为工作 * 线程数(5)大于核心线程数(3),5个工作线程都是调用<4>处workQueue.poll(...) * 等待任务,超时则返回null。如果超时时间到达,3个核心线程如何重新进入等待状态, * 剩余2个线程如何被回收? * 当5个线程超时返回后,会将timedOut赋值为true,然后重新开始新一轮的for循环,一直 * 执行到此分支,此时(timed && timedOut)都为true,队列也都为null,所以5个线程会 * 进入此分支。用CAS成功对workerCount-1的线程将被回收,失败的线程则continue又开始 * 新一轮的for循环,直到wc<=corePoolSize,timed为false,最后剩余的工作线程数调用 * workQueue.take()无限期地等待任务的到来,除非线程被中断。 */ if ((wc > maximumPoolSize || (timed && timedOut)) && (wc > 1 || workQueue.isEmpty())) {//<2> if (compareAndDecrementWorkerCount(c))//<3> return null; continue; } try { /* * workQueue.poll(...)和workQueue.take()都有可能使当前线程陷入 * 等待,直到返回任务,只不过前者相比后者多了一个超时时间,到达超时时间 * 如果有任务入队,则r不为null,直接返回任务。 * 线程等待期间,如果线程被中断,则会抛出InterruptedException异常。 * 一般关闭线程池时,会尝试中断空闲线程,而处于等待任务的空闲线程会跳到<6>处, * 重新开始新一轮的for循环,并且在<1>处判断线程池处于SHUTDOWN或者STOP, * 从而退出。 */ Runnable r = timed ? workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) ://<4> workQueue.take(); if (r != null) return r; /* * 只有等待任务超时控制流才会执行到这里,将超时标志赋值为true, * 重新开始新一轮的for循环。 */ timedOut = true;//<5> } catch (InterruptedException retry) { timedOut = false;//<6> } } }
在runWorker(Worker w)中有两种方式可以进到下面的处理线程退出processWorkerExit(Worker w, boolean completedAbruptly)方法:
- 工作线程执行任务遇到异常,从而跳出while循环。
- 工作线程获取任务超时等待,正常退出循环。
上面两种方式传递给processWorkerExit(...)的completedAbruptly是不同的,第一个方式传入的completedAbruptly为true,第二个方式为false,虽然worker是同一个。那么当completedAbruptly为true或者false,processWorkerExit(...)的流程又是怎么走的呢?
private void processWorkerExit(Worker w, boolean completedAbruptly) { /* * 从runWorker(Worker w)传递而来的变量,标志worker是否意外完成, * 当worker执行任务时抛出异常,该变量为true。如果是意外完成,则表明 * workerCount尚未-1。如果worker获取任务超时从而要让线程被回收,在 * getTask()方法中会对workerCount-1。 */ if (completedAbruptly) // If abrupt, then workerCount wasn't adjusted decrementWorkerCount(); final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { /* * 获取可重入锁后,将worker已完成的任务数加到线程池已完成任务数, * 并将worker从workers集合中移除。 */ completedTaskCount += w.completedTasks; workers.remove(w); } finally { mainLock.unlock(); } //尝试终止线程。 tryTerminate(); int c = ctl.get(); //如果线程池运行状态处于RUNNING或SHUTDOWN,则进入此分支。 if (runStateLessThan(c, STOP)) { //如果线程是正常退出,则进入此分支 if (!completedAbruptly) { /* * min为线程池创建核心线程后,允许最小的核心线程数,如果 * allowCoreThreadTimeOut为true则代表核心线程可以被回收, * 则min为0,否则min为核心线程数量。 * 如果线程池允许回收线程,且队列不为空,则判断在移除当前worker * 后,线程池工作线程的数量是否还大于等于1,避免出现队列里有任务 * 但没有线程执行的情况,如果工作线程数大于等于1,则退出线程线程。 * 否则调用addWorker(Runnable firstTask, boolean core) * 尝试增加新的线程。 */ int min = allowCoreThreadTimeOut ? 0 : corePoolSize; if (min == 0 && !workQueue.isEmpty()) min = 1; if (workerCountOf(c) >= min) return; // replacement not needed } addWorker(null, false); } }
最后,我们还需要介绍关闭线程池之后做的操作。关闭线程池会修改线程池运行状态,在advanceRunState(int targetState)会使用CAS自旋的方式,将线程池状态修改为SHUTDOWN。之后调用interruptIdleWorkers()中断空闲线程,这里我们看到中断的时候调用worker的tryLock()和unlock()。Worker之所以继承AQS就是为了方便区分哪些worker正在执行任务,哪些worker处于空闲中,以便在关闭线程池时中断所有空闲的worker。
/** * 调用此方法后不再接受新任务,但会执行现有队列任务。 * 如果调用此方法前该方法已被调用,则不会有任何效果。 * 该方法不会等待所有任务完成,需要调用: * awaitTermination(long timeout, TimeUnit unit) */ public void shutdown() { final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { checkShutdownAccess(); advanceRunState(SHUTDOWN);//设置线程池状态为SHUTDOWN interruptIdleWorkers();//中断空闲线程 onShutdown(); //钩子方法,按用户需要实现。 } finally { mainLock.unlock(); } /* * 尝试终止线程池,可能线程池有任务在执行,当前线程终止失败。 * 但随着工作线程逐个退出,最后一个工作线程将成功终止线程池。 */ tryTerminate(); } private void advanceRunState(int targetState) { for (; ; ) { int c = ctl.get(); /* * 如果线程池运行状态>=targetState,则 * runStateAtLeast(c, targetState)为true直接退出。 * 否则调用ctlOf(int rs, int wc),根据运行状态和工作 * 线程数生成的值以CAS自旋的方式set进ctl。 */ if (runStateAtLeast(c, targetState) || ctl.compareAndSet(c, ctlOf(targetState, workerCountOf(c)))) break; } } //调用shutdown()会进而调用此方法,中断空闲线程。 private void interruptIdleWorkers() { interruptIdleWorkers(false); } private void interruptIdleWorkers(boolean onlyOne) { final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { for (Worker w : workers) { Thread t = w.thread; /* * 如果线程未被中断,则尝试获取worker的锁,如果能 * 成功获取,代表worker线程处于空闲中。 */ if (!t.isInterrupted() && w.tryLock()) { try { t.interrupt(); } catch (SecurityException ignore) { } finally { w.unlock(); } } //如果onlyOne为true代表最多只中断一个线程 if (onlyOne) break; } } finally { mainLock.unlock(); } }
在中断空闲线程后shutdown()还会调用tryTerminate(),如果查看tryTerminate()的引用,可以发现不单单shutdown()有调用,像addWorkerFailed(Worker w)、processWorkerExit(...)……都有调用。这个方法旨在终止线程池,如果当前有线程关闭了线程池,线程池如果没有存活线程,或者线程都处于空闲状态,自然而然执行尝试终止线程池方法;如果关闭线程池时,线程池仍然有线程处于执行任务状态,无法终止线程池,就要靠这些工作线程在退出时终止线程池。
final void tryTerminate() { for (; ; ) { int c = ctl.get(); /* * 如果线程池运行状态处于RUNNING、TIDYING则退出, * 或者运行状态小于STOP(即处于RUNNING、SHUTDOWN) * 且队列不为空,则退出。 * 总结一下,只有当运行状态处于STOP时或者状态处于SHUTDOWN * 但队列为空,才不会进此分支。 */ if (isRunning(c) || runStateAtLeast(c, TIDYING) ||//<1> (runStateLessThan(c, STOP) && !workQueue.isEmpty())) return; //如果工作线程数不为0,则尝试最多中断一个空闲线程后退出。 if (workerCountOf(c) != 0) { interruptIdleWorkers(ONLY_ONE); return; } /* * 在processWorkerExit(...)方法中如果worker是 * 异常退出会对workerCount-1,如果是正常退出,则 * workerCount在getTask()中-1。之后在processWorkerExit(...) * 中移除worker。 * 不论worker是正常退出还是异常退出,终归workerCount会慢慢回到0。 * 而processWorkerExit(...)中在对workerCount-1后,还会调用 * tryTerminate()。因此一个被关闭的线程池,它的最后一个线程,会 * 执行到此处,处理终止线程池的工作。 */ final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { /* * 用CAS设置线程池运行状态为TIDYING,可能存在多个线程同时 * 调用shutdown()后并依次执行到这一步(因为要获得可重入锁), * 但只有一个线程可以CAS成功,其他线程CAS失败后返回<1>处后 * 判断运行状态>=TIDYING则退出。 */ if (ctl.compareAndSet(c, ctlOf(TIDYING, 0))) { try { //空函数,具体由用户实现,主要用于终止线程池后的操作。 terminated(); } finally { //最后设置线程池的状态为TERMINATED ctl.set(ctlOf(TERMINATED, 0)); termination.signalAll(); } return; } } finally { mainLock.unlock(); } } }