Java提供了几种便捷的方法建立线程池,经过这些内置的api就可以很轻松的建立线程池。在java.util.concurrent
包中的Executors
类,其中的静态方法就是用来建立线程池的:java
- newFixedThreadPool():建立一个固定线程数量的线程池,并且线程池中的任务所有执行完成后,空闲的线程也不会被关闭。
- newSingleThreadExecutor():建立一个只有一个线程的线程池,空闲时也不会被关闭。
- newCachedThreadPool():建立一个可缓存的线程池,线程的数量为
Integer.MAX_VALUE
,空闲线程会临时缓存下来,线程会等待60s
仍是没有任务加入的话就会被关闭。
Executors
类中还有一些建立线程池的方法(jdk8新加的),可是如今这个触极到个人知识盲区了~~面试
上面那几个方法,其实都是建立了一个ThreadPoolExecutor
对象做为返回值,要搞清楚线程池的原理主要仍是要分析ThreadPoolExecutor
这个类。api
ThreadPoolExecutor
的构造方法:缓存
public ThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<runnable> workQueue, ThreadFactory threadFactory, RejectedExecutionHandler handler) { ... }
ThreadPoolExecutor
的构造方法包含如下几个参数:安全
- corePoolSize: 核心线程数量,常驻线程池中的线程,即时线程池中没有任务可执行,也不会被关闭。
- maximumPoolSize:最大线程数量
- keepAliveTime:空闲线程存活时间
- unit: 空闲线程存活时间的单位
- workQueue:工做队列,线程池一下忙不过来,那新来的任务就须要排队,排除中的任务就会放在workQueue中
- threadFactory:线程工厂,建立线程用的
- handler:
RejectedExecutionHandler
实例用于在线程池中没有空闲线程可以执行任务,而且workQueue
中也容不下任务时拒绝任务时的策略。
ThreadPoolExecutor
中的线程统称为工做线程,但有一个小概念是核心线程
,核心线程由参数corePoolSize
指定,如corePoolSize
设置5,那线程池中就会有5条线程常驻线程池中,不会被回收掉,可是也会有例外,若是allowCoreThreadTimeOut
为true
空闲一段时间后,也会被关闭。架构
线程的状态和工做线程数量
线程中的状态和工做线程和数量都是由ctl
表示,是一个AtomicInteger
类型的属性:oop
private final AtomicInteger ctl = new AtomicInteger(ctlOf(RUNNING, 0));
ctl的高四位为线程的状态,其余位数为工做线程的数量,因此线程中最大的工做线程数量为(2^29)-1
。学习
线程池中的状态有五种:ui
- RUNNING:接收新的任务和处理队列中的任务
- SHUTDOWN:不能新增任务,可是会继续处理已经添加的任务
- STOP:不能新增任务,不会继续处理已经添加任务
- TIDYING:全部的任务已经被终止,工做线程为0
- TERMINATED:terminated()方法执行完成
状态码的定义以下:this
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;
建立线程池
若是有面试官问:如何正确的建立线程池?千万不要说使用Executors
建立线程,虽然Executors
能很方便的建立线程池,可是他提供的静态建立方法会有一些坑。
主要的缘由是:maximumPoolSize
和workQueue
这两个参数
Executors
静态方法在建立线程池时,若是maximumPoolSize
设置为Integer.MAX_VALUE
,这样会致使线程池能够一直要以接收运行任务,可能致使cpu负载太高。
workQueue
是一个阻塞队列的实例,用于放置正在等待执行的任务。若是在建立线程种时workQueue
实例没有指定任务的容量,那么等待队列中能够一直添加任务,极有可能致使oom
。
因此建立线程,最好是根据线程池的用途,而后本身建立线程。
添加任务
调用线程池的execute
并非当即执行任务,线程池内部用通过一顿操做,如:判断核心线程数、是否须要添加到等待队列中。
下来的代码是execute
的源码,代码很简洁只有2个if
语句:
public void execute(Runnable command) { 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); }
- 第一个if,若是当前线程池中的工做线程数量小于
corePoolSize
,直接建立一个工做线程执行任务 - 第二个if,当线程池处于运行状态,调用
workQueue.offer(command)
方法将任务添加到workQueue
,不然调用addWorker(command, false)
尝试去添加一个工做线程。
整理了一张图,把线程池分为三部分Core Worker
、Worker
、workQueue
:
换一种说法,在调用execute
方法时,任务首先会放在Core Worker
内,而后才是workQueue
,最后才会考虑Worker
。
这样作的缘由能够保证Core Worker
中的任务执行完成后,能当即从workQueue
获取下一个任务,而不须要启动别的工做线程,用最少的工做线程办更多的事。
建立工做线程
在execute
方法中,有三个地方调用了addWorker
。addWorker
方法能够分为二部分:
- 增长工做线程数量
- 启动工做线程
addWorker
的方法签名以下:
private boolean addWorker(Runnable firstTask, boolean core)
- firstTask:第一个运行的任务,能够为空。若是为空任务会从
workQueue
中获取。 - core: 是不是核心工做线程
增长工做线程数量
retry: for (;;) { int c = ctl.get(); int rs = runStateOf(c); .... for (;;) { int wc = workerCountOf(c); if (wc >= CAPACITY || wc >= (core ? corePoolSize : maximumPoolSize)) return false; 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 } }
上面代码省略了一部分代码,主要代码都在for
循环中,利用CAS
锁,安全的完成线程池状态的检查与增长工做线程的数量。其中的compareAndIncrementWorkerCount(c)
调用就是将工做线程数量+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 { // Recheck while holding lock. // Back out on ThreadFactory failure or if // shut down before lock acquired. 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); 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); }
启动工做线程的流程:
- 建立一个
Worker
实例,Worker
构造方法会使用ThreadFactory
建立一个线程
w = new Worker(firstTask); final Thread t = w.thread;
就不说Worker
类的实现了,直接给出构造方法来细品:
Worker(Runnable firstTask) { setState(-1); // inhibit interrupts until runWorker this.firstTask = firstTask; this.thread = getThreadFactory().newThread(this); }
- 若是线程池状态是在运行中,或者已经关闭,但工做线程要从
workQueue
中获取任务,才能添加工做线程
if (rs < SHUTDOWN || (rs == SHUTDOWN && firstTask == null)) { if (t.isAlive()) // precheck that t is startable throw new IllegalThreadStateException(); workers.add(w); int s = workers.size(); if (s > largestPoolSize) largestPoolSize = s; workerAdded = true; }
注意::当线程池处于SHUTDOWN
状态时,它不能接收新的任务,可是能够继续执行未完成的任务。任务是否从workQueue
中获取,是根据firstTask
判断,每一个Worker
实例都有一个firstTask
属性,若是这个值为null
,工做线程启动的时候就会从workQueue
中获取任务,不然会执行firstTask
。
- 启动线程
调用线程的start
方法,启动线程。
if (workerAdded) { t.start(); workerStarted = true; }
执行任务
回过头来看一个Worker
类的定义:
private final class Worker extends AbstractQueuedSynchronizer implements Runnable{ Worker(Runnable firstTask) { setState(-1); // inhibit interrupts until runWorker this.firstTask = firstTask; this.thread = getThreadFactory().newThread(this); } ... }
Worker
类实现了Runnable
接口,同时在构造方法中会将this
传递给线程,到这里你就知道了Worker
实例中有run
方法,它会在线程启动后执行:
public void run() { runWorker(this); }
run
方法内部接着调用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) { w.lock(); // If pool is stopping, ensure thread is interrupted; // if not, ensure thread is not interrupted. This // requires a recheck in second case to deal with // shutdownNow race while clearing 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; w.completedTasks++; w.unlock(); } } completedAbruptly = false; } finally { processWorkerExit(w, completedAbruptly); } }
- 获取任务
首先将firstTask
传递给task
临时变量:
Runnable task = w.firstTask;
而后循环检查task
或者从workQueue
中获取任务:
while (task != null || (task = getTask()) != null) { ... }
getTask()
稍后再作分析。
- 运行任务
去掉一些状态检查、异常捕获、和勾子方法调用后,保留最重要的调用task.run()
:
while (task != null || (task = getTask()) != null) { ... task.run(); ... }
task
其实就是经过调用execute
方法传递进来的Runnable
实例,也就是你的任务。只不过它可能保存在Worker.firstTask
中,或者在workQueue
中,保存在哪里在前面的任务添加顺序
中已经说明。
从workQueue中获取任务
试想一下若是每一个任务执行完成,就关闭掉一个线程那有多浪费资源,这样使用线程池也没有多大的意义。因此线程的主要的功能就是线程复用,一旦任务执行完成直接去获取下一个任务,或者挂起线程等待下一个提交的任务,而后等待一段时间后仍是没有任务提交,而后才考虑是否关闭部分空闲的线程。
runWorker
中会循环的获取任务:
while (task != null || (task = getTask()) != null) { ... task.run(); ... }
上面的代码getTask()
就是从workQueue
中获取任务:
private Runnable getTask() { boolean timedOut = false; // Did the last poll() time out? for (;;) { ... int wc = workerCountOf(c); // Are workers subject to culling? boolean timed = allowCoreThreadTimeOut || wc > corePoolSize; ... try { Runnable r = timed ? workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) : workQueue.take(); if (r != null) return r; timedOut = true; } catch (InterruptedException retry) { timedOut = false; } } }
获取任务的时候会有两种方式:
- 超时等待获取任务
- 一直等待任务,直到有新任务
若是allowCoreThreadTimeOut
为true
,corePoolSize
指定的核心线程数量会被忽略,直接使用 workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS)
获取任务,不然的话会根据当前工做线程的数量,若是wc > corePoolSize
为false
则当前会被认为是核心线程,调用workQueue.take()
一直等待任务。
工做线程的关闭
仍是在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) { task.run(); } completedAbruptly = false; } finally { processWorkerExit(w, completedAbruptly); } }
- completedAbruptly变量:标记当前工做线程是正常执行完成,仍是异常完成的。completedAbruptly为
false
能够肯定线程池中没有可执行的任务了。
上面代码是简洁后的代码,一个while
循环保证不间断的获取任务,没有任务能够执行(task为null)退出循环,最后再才会调用processWorkerExit
方法:
private void processWorkerExit(Worker w, boolean completedAbruptly) { if (completedAbruptly) // If abrupt, then workerCount wasn't adjusted decrementWorkerCount(); final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { completedTaskCount += w.completedTasks; workers.remove(w); } finally { mainLock.unlock(); } tryTerminate(); int c = ctl.get(); if (runStateLessThan(c, STOP)) { if (!completedAbruptly) { int min = allowCoreThreadTimeOut ? 0 : corePoolSize; if (min == 0 && ! workQueue.isEmpty()) min = 1; if (workerCountOf(c) >= min) return; // replacement not needed } addWorker(null, false); } }
processWorkerExit
接收一个Worker
实例与completedAbruptly
变量。processWorkerExit的大体工做流程:
- 判断当前工做线程是否异常完成,若是是直接减小工做线程的数量,简单的说就是校订一下工做线程的数量。
- 增长完成的任务数量,将
Worker
从workers
中移除 - tryTerminate() 检查线程池状态,由于线程池能够延迟关闭,若是你调用
shutdown
方法后不会当即关闭,要等待全部的任务执行完成,因此这里调用tryTerminate()方法,尝试去调用terminated
方法。
工做线程完成策略
若是某个工做线程完成,线程池内部会判断是否须要从新启动一个:
//判断线程池状态 if (runStateLessThan(c, STOP)) { if (!completedAbruptly) { //获取最小工做线程数量 int min = allowCoreThreadTimeOut ? 0 : corePoolSize; //若是最小工做线程数量为0,可是workQueue中还有任务,那重置最小工做线程数量1 if (min == 0 && ! workQueue.isEmpty()) min = 1; //若是当前工做线程数数量大于或等于最小工做线程数量,则不须要启动新的工做线程 if (workerCountOf(c) >= min) return; // replacement not needed } //启动一个新的工做线程 addWorker(null, false); }
工做线程完成后有两种处理策略:
- 对于异常完成的工做线程,直接启动一个新的替换
- 对于正常完成的工做线程,判断当前工做线程是否足够,若是足够则不须要新启动工做线程
注意: 这里的完成,表示工做线程的任务执行完成,workQueue
中也没有任务能够获取了。
线程池的关闭
关闭线程池有能够经过shutdown
方法:
public void shutdown() { final ReentrantLock mainLock = this.mainLock; mainLock.lock(); try { checkShutdownAccess(); advanceRunState(SHUTDOWN); interruptIdleWorkers(); onShutdown(); // hook for ScheduledThreadPoolExecutor } finally { mainLock.unlock(); } tryTerminate(); }
shutdown
方法,第一步就是先改变线程池的状态,调用advanceRunState(SHUTDOWN)
方法,将线程池当前状态更改成SHUTDOWN
,advanceRunState代码以下:
private void advanceRunState(int targetState) { for (;;) { int c = ctl.get(); if (runStateAtLeast(c, targetState) || ctl.compareAndSet(c, ctlOf(targetState, workerCountOf(c)))) break; } }
而后当即调用interruptIdleWorkers()
方法,interruptIdleWorkers()
内部会调用它的重载方法interruptIdleWorkers(boolean onlyOne)
同时onlyOne参数传递的false
来关闭空闲的线程:
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; if (!t.isInterrupted() && w.tryLock()) { try { t.interrupt(); } catch (SecurityException ignore) { } finally { w.unlock(); } } if (onlyOne) break; } } finally { mainLock.unlock(); } }
以上代码会遍历workers
中的Worker
实例,而后调用线程的interrupt()
方法。
什么样的线程才是空闲工做线程?
前面提到过在getTask()
中,线程从workQueue
中获取任务时会阻塞,被阻塞的线程就是空闲的。
再次回到getTask()
的代码中:
private Runnable getTask() { boolean timedOut = false; // Did the last poll() time out? for (;;) { // Check if queue empty only if necessary. if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) { decrementWorkerCount(); return null; } ... int wc = workerCountOf(c); // Are workers subject to culling? boolean timed = allowCoreThreadTimeOut || wc > corePoolSize; ... try { Runnable r = timed ? workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) : workQueue.take(); if (r != null) return r; timedOut = true; } catch (InterruptedException retry) { timedOut = false; } } }
再次分析getTask()
中的代码中有一段捕获InterruptedException
的代码块,interruptIdleWorkers方法中断线程后,getTask()
会捕获中断异常,由于外面是一个for
循环,随后代码走到判断线程池状态的地方:
if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) { decrementWorkerCount(); return null; }
上面的代码的会判断当前线程池状态,若是状态大于STOP
或者状态等于SHUTDOWN
而且workQueue
为空时则返回null
,getTask()
返回空那么在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) { task.run(); } completedAbruptly = false; } finally { processWorkerExit(w, completedAbruptly); } }
shutdownNow
除了shutdown方法能关闭线程池,还有shutdownNow
也能够关闭线程池。它两的区别在于:
shutdownNow
会清空workQueue
中的任务shutdownNow
还会停止当前正在运行的任务shutdownNow
会使线程进入STOP
状态,而shutdown()
是SHUTDOWN
状态
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; }
上面代码基本流程:
- advanceRunState(STOP): 使线程池进行
STOP
状态,与shutdown()
中的一致 ,只是使用的状态码是STOP
- interruptWorkers(): 与
shutdown()
中的一致 - drainQueue(): 清空队列
任务是停止执行仍是继续执行?
调用shutdownNow()后线程池处于STOP
状态,紧接着全部的工做线程都会被调用interrupt
方法,若是此时runWorker
还在运行会发生什么?
在runWorker
有一段代码,就是工做线程停止的重要代码:
final void runWorker(Worker w) { ... while (task != null || (task = getTask()) != null) { if ((runStateAtLeast(ctl.get(), STOP) || (Thread.interrupted() && runStateAtLeast(ctl.get(), STOP))) && !wt.isInterrupted()) wt.interrupt(); task.run(); } ... }
重点关注:
if ((runStateAtLeast(ctl.get(), STOP) || (Thread.interrupted() && runStateAtLeast(ctl.get(), STOP))) && !wt.isInterrupted()) wt.interrupt();
这个if看起来有点难理解,理解下来大体意思是:若是线程池状态大于等于STOP
,当即中断线程,不然清除线程的中断标记,也就是说当线程池状态为RUNNING
和SHUTDOWN
时,线程的中断标记会被清除(线程的中断代码在interruptWorkers
方法中),能够继续执行任务。
以上代码执行完成后,紧接着就会调用task.run()
方法,这里面咱们本身就能够根据线程的中断标记来判断任务是否被中断。
总结
我的水平有限,文中若有错误,谢谢你们指正。
本文从线程池的源码入手,分析线程池的建立、添加任务、运行任务等流程,整个分析下来基本上大多数公司关于线程池面试的问题均可以回答得上来,固然还有一些小细节如:Worker
类是继承AQS
的,为何这么作其实源码中都有一些苗头,Worker
在运行时会锁住运行的代码块,而shutdown
在关闭空闲的Worker
时,首先就要去获取Worker
的同步锁才能继续操做,这样才能安全的关闭工做线程。
欢迎关注个人公众号:架构文摘,得到独家整理120G的免费学习资源助力你的架构师学习之路!
公众号后台回复
arch028
获取资料: