ThreadPoolExecutor源码研读

原文为 https://www.cnblogs.com/huangjuncong/p/10031525.html，因为博主加了注释所以会使用一些他的图片，侵删。对于博主没有讲到的部分会使用jdk源码的截图

 为什么要用一个ctl记录线程状态和线程个数？因为需要得到一个对这两个量的同一快照，如果使用两个变量则需要一个锁，而使用一个变量不需要（赋值操作是原子的，赋值后在对副本进行解析，可以保证是同一个的快照）。这种设计非常巧妙，值得学习

线程池状态：RUNNING：运行状态，接受任务且处理阻塞队列中的任务

SHUTDOWN：拒绝任务但处理阻塞队列中的任务

STOP：拒绝任务，放弃阻塞队列中的任务，且中断执行中的任务

TIDYING：所有任务都处理完成，活动线程为0，准备调用terminate方法

TERMINATED：关闭状态

调用shutdown会进入SHUTDOWN并拒绝接受任务，调用shutdownNow会进入STOP并拒绝接受任务，抛弃等待队列中的任务并返回他们，中断执行中的任务

参数：corePoolSize：核心线程数

maxPoolSize：最大线程数

ThreadFactory：线程工厂

RejectExecutionHandler：拒绝策略

workQueue：等待队列

keepAlive：存活时间

timeUnit：时间单位

 

这是execute的代码，需要注意几个点。(3)处，可以看到即便是核心线程也不是一开始就创建好的，而是按需创建的

(4)处，即使当前有空闲的线程，也直接将任务放入阻塞队列，空闲线程会从阻塞队列中取出任务；如果阻塞队列满了（offer方法返回false）才会到(5)处试图创建非核心线程。所以除了(3)处直接创建的核心线程和(5)处创建非核心线程，其他时候都是先将任务放入阻塞队列等待被提取

那么就出现一个问题，如果我传入一个capacity为0的队列是不是就会导致这个方法出问题？（每次(4)都会失败，所以创建非核心线程，然后非核心线程就会达到上限，最后进入拒绝策略，即使每个线程都空闲，但它们都无法从等待队列中得到任务，因为等待队列永远不返回任何任务）

下面是我的实验，由于ArrayBlockingQueue无法设置capacity为0，会抛出异常，所以我继承了ArrayBlockingQueue，实现它的offer方法永远返回false，模拟这个队列的capacity为0

 

可以看到当提交的任务总数超过了maximumPoolSize就会走入拒绝策略，这和我们讨论的一样，这些线程都是空闲的，但都无法从等待队列中得到任务，又不能再创建新的线程。要注意这里的keepAlive设置为100s，原因是等待一个长的时间才能让非核心线程不被释放从而沾满线程池，达到我们实验的效果

总的来说这个不能算是bug吧，就是一种设计细节，毕竟capacity为0的阻塞队列是没有意义的

 来看看addWorker的实现，这是上半部分，下图是下半部分。上半部分主要使用两个循环来检查线程池状态并增加线程个数

 

 下半部分主要是创建一个Worker并将其添加入workers集合中，然后使用t.start()执行此Worker线程

 

 

 Woker是一个Runnable继承了AQS，其tryAquire和tryRelease也非常简单，继承AQS本质不是实现一个锁，而是保持一个状态，lock状态下此Worker不会被shutdown方法中断，而unlock状态下会被中断，这个我们后面会分析。它的run方法调用了runWorker，runWorker是ThreadPoolExecutor中的方法，让我们看一看

 这里会调用task.run()执行任务，值得注意的是(10.1)是唯一一处使用woker.lock方法的地方，这里使用lock防止shutdown中断正在执行的任务

 

而这个方法的789行是唯一一处调用tryLock方法的地方，这个方法会被shutdown方法调用，从而通过tryLock中断所有不在执行的worker

再看(10)处的getTask()，这个方法是执行完一个任务后从等待队列中取另一个任务，如果getTask返回null则会跳出循环然后在processWorkerExit中将worker从workers中移除，下面我们会看一下这个方法

 

这里有两个重点，断点1的timed表示这次取任务操作是有限时间等待还是无限时间等待，如果allowCoreThreadTimeout则意味着核心线程可以等待超时，如果wc>corePoolSize则意味着存在非核心线程，在这两种情况下是有限时间的等待

断点2根据之前的timed来选择取操作的方法，如果有限时间则使用poll，否则使用take，然后如果r为null则说明有限时间的poll没取到任务，此时设置timeOut为true

设置timeOut为true会在1044行用到，这里返回null会导致调用getTask的worker被移除出workers队列，返回null有三个条件需要满足，首先线程数量超过最大值maximumPoolSize或已超时即timed和timeout为true；然后需要满足线程数超过1或等待队列为空，如果线程数为1且等待队列不为空再将这个worker移除出队列就会导致没有worker处理等待队列的任务，从而饿死这些任务；第三需要1046行CAS减少workerCount成功

这里可以看到poll方法的等待时间是keepAliveTime，以上就是线程池释放非核心线程的方式

 刚才说了shutDown，现在看一下shutDownNow，注意二者使用的中断方法不同，shutDown调用interruptIdleWorkers中断所有没有在执行的worker，而interruptWorkers会中断所有worker

 最后说一下，代码中看到的mainLock.lock和worker.lock完全不是一回事，mainLock是ReentrantLock，用来所注竞争资源如workers集合等；而worker虽然也继承自AQS，但其只是持有一个状态，lock状态为不可被shutdown中断