《java.util.concurrent 包源码阅读》11 线程池系列之ThreadPoolExecutor 第一部分
先来看ThreadPoolExecutor的execute方法,这个方法能体现出一个Task被加入到线程池之后都发生了什么:
public void execute(Runnable command) { if (command == null) throw new NullPointerException(); /* 如果运行中的worker线程数少于设定的常驻线程数,增加worker线程,把task分配给新建的worker线程 */ 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); } // 如果任务数超过了现有worker线程的承受范围,尝试新建worker线程 // 如果无法添加新的worker线程,则会拒绝该任务 else if (!addWorker(command, false)) reject(command); }
在执行任务时,需要经常检查线程池的状态,那么接下来说说线程池是如何进行状态控制的。上面的代码有个成员变量叫做ctl,它用于标记线程池状态和worker线程的数量,是一个AutomaticInteger对象。
private final AtomicInteger ctl = new AtomicInteger(ctlOf(RUNNING, 0));
ctl是一个32位的整数,最高的3位表示状态:
111为running,
000为shutdown,
001为stop,
010为tidying,
011为ternimated。
因此状态值就是这三位加上29个0,因此running的值是个负整数(最高位为1),其他状态都是正整数,后面判断状态会比较值的大小时会用到这点。
剩下的29位表示worker线程的数量(因此最大允许的线程数就是2的29方减1)。
这里是说说这几个状态的意义,这几个状态发生的顺序正好就是上面列出的顺序:
running表示正常运行状态
shutdown状态意味着发出了一个shutdown信号,类似于你点击了windows的关机按钮
stop表示shutdown信号收到,等于windows响应了这个信号,发出正在关机的信息
tidying发生在stop之后做出的响应,表示这个时候在清理一些资源,
ternimated发生在tidying完成之后,表示关闭完成。
接着来看添加一个worker线程时都发生了什么:
private boolean addWorker(Runnable firstTask, boolean core) { retry: for (;;) { int c = ctl.get(); int rs = runStateOf(c); // 返回false的情况: // 1. rs>shutdown,即shutdown和running以外的状态 // 2. shutdown的状态 // 1)firstTask不为null,即有task分配 // 2)没有task,但是workQueue(等待任务队列)为空 if (rs >= SHUTDOWN && ! (rs == SHUTDOWN && firstTask == null && ! workQueue.isEmpty())) return false; for (;;) { // 1. 如果没有设定线程数的限制,worker线程数不能大于最大值(2的29次方-1) // 2. 如果是固定尺寸的线程池,不能大于固定尺寸 // 3. 如果是可扩展的线程池,不能大于规定的线程数的上限 int wc = workerCountOf(c); if (wc >= CAPACITY || wc >= (core ? corePoolSize : maximumPoolSize)) return false; // 用CAS操作增加线程数量,如果失败,重新循环 if (compareAndIncrementWorkerCount(c)) break retry; c = ctl.get(); // Re-read ctl if (runStateOf(c) != rs) continue retry; loop } } // 新建worker线程 Worker w = new Worker(firstTask); Thread t = w.thread; final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { int c = ctl.get(); int rs = runStateOf(c); // 检查以下任一状态是否出现: // 1. 创建线程失败 // 2. rs>shutdown,即shutdown和running以外的状态 // 3. rs==shutdown,有任务分配 if (t == null || (rs >= SHUTDOWN && ! (rs == SHUTDOWN && firstTask == null))) { decrementWorkerCount(); tryTerminate(); return false; } workers.add(w); int s = workers.size(); if (s > largestPoolSize) largestPoolSize = s; } finally { mainLock.unlock(); } t.start(); // 这里考虑一种极少出现的情况,如果worker线程调用start没有完成时, // 线程池进入Stop状态,这个时候会调用Thread#interrupt中断每个 // worker线程,但是 interrupt对没有start的线程不一定起作用,这样 // 就会漏掉了对这个thread的interrupt,因此在worker线程start之后 // 检查以下,如果stop了,而这个线程却没有被interrupt,补上这个漏掉 // 的interrupt。 if (runStateOf(ctl.get()) == STOP && ! t.isInterrupted()) t.interrupt(); return true; }
这篇文章主要讲线程池如何处理任务,下一篇文章将会讲worker线程是如何工作的。