1. 引言
合理利用线程池能够带来三个好处。第一:降低资源消耗。通过重复利用已创建的线程降低线程创建和销毁造成的消耗。第二:提高响应速度。当任务到达时,任务可以不需要的等到线程创建就能立即执行。第三:提高线程的可管理性。线程是稀缺资源,如果无限制的创建,不仅会消耗系统资源,还会降低系统的稳定性,使用线程池可以进行统一的分配,调优和监控。但是要做到合理的利用线程池,必须对其原理了如指掌。
2.线程池的使用
线程池的创建
我们可以通过ThreadPoolExecutor来创建一个线程池。
public class ThreadPoolExecutor extends AbstractExecutorService {
.....
public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,
BlockingQueue<Runnable> workQueue);
public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,
BlockingQueue<Runnable> workQueue,ThreadFactory threadFactory);
public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,
BlockingQueue<Runnable> workQueue,RejectedExecutionHandler handler);
public ThreadPoolExecutor(int corePoolSize,int maximumPoolSize,long keepAliveTime,TimeUnit unit,
BlockingQueue<Runnable> workQueue,ThreadFactory threadFactory,RejectedExecutionHandler handler);
...
}
创建一个线程池需要输入几个参数:
-
corePoolSize(线程池的基本大小):当提交一个任务到线程池时,线程池会创建一个线程来执行任务,即使其他空闲的基本线程能够执行新任务也会创建线程,等到需要执行的任务数大于线程池基本大小时就不再创建。如果调用了线程池的prestartAllCoreThreads方法,线程池会提前创建并启动所有基本线程。
-
runnableTaskQueue(任务队列):用于保存等待执行的任务的阻塞队列。可以选择以下几个阻塞队列。
1、ArrayBlockingQueue:是一个基于数组结构的有界阻塞队列,此队列按 FIFO(先进先出)原则对元素进行排序。
2、LinkedBlockingQueue:一个基于链表结构的阻塞队列,此队列按FIFO (先进先出) 排序元素,吞吐量通常要高于ArrayBlockingQueue。静态工厂方法Executors.newFixedThreadPool()使用了这个队列。
3、SynchronousQueue:一个不存储元素的阻塞队列。每个插入操作必须等到另一个线程调用移除操作,否则插入操作一直处于阻塞状态,吞吐量通常要高于LinkedBlockingQueue,静态工厂方法Executors.newCachedThreadPool使用了这个队列。
4、PriorityBlockingQueue:一个具有优先级得无限阻塞队列 -
maximumPoolSize(线程池最大大小):线程池允许创建的最大线程数。如果队列满了,并且已创建的线程数小于最大线程数,则线程池会再创建新的线程执行任务。值得注意的是如果使用了无界的任务队列这个参数就没什么效果。
-
ThreadFactory:用于设置创建线程的工厂,可以通过线程工厂给每个创建出来的线程设置更有意义的名字,Debug和定位问题时非常又帮助。
-
RejectedExecutionHandler(饱和策略):当队列和线程池都满了,说明线程池处于饱和状态,那么必须采取一种策略处理提交的新任务。这个策略默认情况下是AbortPolicy,表示无法处理新任务时抛出异常。以下是JDK1.5提供的四种策略。
1、AbortPolicy:直接抛出异常。
2、CallerRunsPolicy:只用调用者所在线程来运行任务。
3、DiscardOldestPolicy:丢弃队列里最近的一个任务,并执行当前任务。
4、DiscardPolicy:不处理,丢弃掉。
5、当然也可以根据应用场景需要来实现RejectedExecutionHandler接口自定义策略。如记录日志或持久化不能处理的任务。 -
keepAliveTime(线程活动保持时间):线程池的工作线程空闲后,保持存活的时间。所以如果任务很多,并且每个任务执行的时间比较短,可以调大这个时间,提高线程的利用率。
-
TimeUnit(线程活动保持时间的单位):可选的单位有天(DAYS),小时(HOURS),分钟(MINUTES),毫秒(MILLISECONDS),微秒(MICROSECONDS,
千分之一毫秒)和毫微秒(NANOSECONDS, 千分之一微秒)。
线程池提交任务
我们可以使用execute提交的任务,但是execute方法没有返回值,所以无法判断任务知否被线程池执行成功。execute方法输入的任务是一个Runnable类的实例。我们也可以使用submit 方法来提交任务,它会返回一个future,那么我们可以通过这个future来判断任务是否执行成功,通过future的get方法来获取返回值,get方法会阻塞住直到任务完成,而使用get(long timeout, TimeUnit unit)方法则会阻塞一段时间后立即返回,这时有可能任务没有执行完。
线程池的关闭
我们可以通过调用线程池的shutdown或shutdownNow方法来关闭线程池,但是它们的实现原理不同,shutdown的原理是只是将线程池的状态设置成SHUTDOWN状态,然后中断所有没有正在执行任务的线程。shutdownNow的原理是遍历线程池中的工作线程,然后逐个调用线程的interrupt方法来中断线程,所以无法响应中断的任务可能永远无法终止。shutdownNow会首先将线程池的状态设置成STOP,然后尝试停止所有的正在执行或暂停任务的线程,并返回等待执行任务的列表。
只要调用了这两个关闭方法的其中一个,isShutdown方法就会返回true。当所有的任务都已关闭后,才表示线程池关闭成功,这时调用isTerminaed方法会返回true。至于我们应该调用哪一种方法来关闭线程池,应该由提交到线程池的任务特性决定,通常调用shutdown来关闭线程池,如果任务不一定要执行完,则可以调用shutdownNow。
3. 线程池的分析
流程分析:线程池的主要工作流程如下图:
从上图我们可以看出,当提交一个新任务到线程池时,线程池的处理流程如下:
- 首先线程池判断基本线程池是否已满?没满,创建一个工作线程来执行任务。满了,则进入下个流程。
- 其次线程池判断工作队列是否已满?没满,则将新提交的任务存储在工作队列里。满了,则进入下个流程。
- 最后线程池判断整个线程池是否已满?没满,则创建一个新的工作线程来执行任务,满了,则交给饱和策略来处理这个任务。
源码分析。上面的流程分析让我们很直观的了解的线程池的工作原理,让我们再通过源代码来看看是如何实现的。线程池执行任务的方法如下:
- execute源码分析
public void execute(Runnable command) {
if (command == null) // 空任务抛出异常
throw new NullPointerException();
/*
* Proceed in 3 steps:
*
* 1. If fewer than corePoolSize threads are running, try to
* start a new thread with the given command as its first
* task. The call to addWorker atomically checks runState and
* workerCount, and so prevents false alarms that would add
* threads when it shouldn't, by returning false.
*
* 2. If a task can be successfully queued, then we still need
* to double-check whether we should have added a thread
* (because existing ones died since last checking) or that
* the pool shut down since entry into this method. So we
* recheck state and if necessary roll back the enqueuing if
* stopped, or start a new thread if there are none.
*
* 3. If we cannot queue task, then we try to add a new
* thread. If it fails, we know we are shut down or saturated
* and so reject the task.
*/
int c = ctl.get();
// 1. 如果工作线程数小于核心线程数,则添加新的线程
if (workerCountOf(c) < corePoolSize) {
if (addWorker(command, true))
return; // 添加成功则返回
c = ctl.get(); // 否则获取线程池状态
}
// 2. 工作线程数大于等于核心线程数,则将任务放入缓存任务队列
// 与操作: 如果线程池正在运行,而且成功将任务插入缓存任务队列两个条件
// 都满足则进入条件语句内
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); // 添加无初始任务的线程
}
// 3. 任务入队失败,说明任务缓存任务队列已满,尝试添加新的线程处理
// 如果添加失败则以某种方式拒绝任务
else if (!addWorker(command, false))
reject(command);
}
2.内部实现addWorker源码分析
private boolean addWorker(Runnable firstTask, boolean core) {
// 对 runState进行循环获取和判断,如果不满足添加条件则返回false
retry:
for (;;) {
// 获取runState,和workerCount
int c = ctl.get();
int rs = runStateOf(c);
// 对线程池状态进行判断,是否适合添加新的线程
if (rs >= SHUTDOWN &&
! (rs == SHUTDOWN &&
firstTask == null &&
! workQueue.isEmpty()))
return false;
// 进行CAS设置workerCount,失败重试
for (;;) {
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
// 如果workerCount没有设置成功,而且runState发生变化,
// 则继续最外层的循环,对runState重新获取和判断
if (runStateOf(c) != rs)
continue retry;
}
}
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 {
// 获取锁后,对runState进行再次检查
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); // 线程回滚
}
return workerStarted;
}
/**
* Rolls back the worker thread creation.
*/
private void addWorkerFailed(Worker w) {
final ReentrantLock mainLock = this.mainLock;
// 对线程回滚需要获取全局锁
mainLock.lock();
try {
if (w != null)
workers.remove(w); // 从工作线程集合中移除添加失败的线程
decrementWorkerCount(); // 减少工作线程计数
// 因为中断异常而没有启动线程,从而回滚已入队的线程
// 这个中断异常可能是关闭线程池时发生的,所以应该将终止线程池的信号传播
tryTerminate(); // 尝试终止线程
} finally {
mainLock.unlock();
}
}
- 工作线程主循环函数源码分析
final void runWorker(Worker w) {
Thread wt = Thread.currentThread();
Runnable task = w.firstTask;
w.firstTask = null; // 将线程初始任务设为null
w.unlock(); // 执行任务前,释放锁,允许被中断
boolean completedAbruptly = true; // 因为运行异常导致线程突然终止的标志
try {
// 获取任务,如果没有任务可以获取,则此循环终止,
// 这个工作线程将结束工作,等待被清理前的登记工作
while (task != null || (task = getTask()) != null) {
w.lock(); // 执行任务之前获取工作线程锁
// 如果线程池关闭,则确保线程被中断
// 如果线程池没有关闭,则确保线程不被中断
// 这就要求在第二种情况下,进行重新检查,处理shutdownNow正在运行同时清除中断
if ((runStateAtLeast(ctl.get(), STOP) ||
(Thread.interrupted() &&
runStateAtLeast(ctl.get(), STOP))) &&
!wt.isInterrupted())
wt.interrupt();
try {
beforeExecute(wt, task); // before hook
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); after hook
}
} finally {
task = null;
w.completedTasks++;
w.unlock();
}
}
completedAbruptly = false;
} finally {
// (可能是因为没有任务可做正常结束,或者产生异常而导致线程异常结束)
// 登记信息,移除结束线程,然后根据情况添加新的线程等
processWorkerExit(w, completedAbruptly);
}
}
// 处理线程退出函数
private void processWorkerExit(Worker w, boolean completedAbruptly) {
if (completedAbruptly) // 如果是用户代码task.run()异常突然退出
decrementWorkerCount(); // 则减少 workerCount的计数
// 否则为正常退出,在getTask函数里已经进行了减少workerCount的操作
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; // allowCoreThreadTimeOut 情况下需要的最少线程数
// 如果线程数量大于等于正常工作的数量则不再添加新的线程
if (workerCountOf(c) >= min)
return; // replacement not needed
}
addWorker(null, false); // 添加worker线程
}
}
- 从队列中获取任务源码分析
private Runnable getTask() {
boolean timedOut = false; // 记录上一次获取任务是否超时的标志
for (;;) {
int c = ctl.get();
int rs = runStateOf(c);
// Check if queue empty only if necessary
if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
decrementWorkerCount(); // 因为上述条件无法获取任务,则减少workerCount数量
return null; // 返回 null,导致工作线程主循环结束,并移除该工作线程
}
int wc = workerCountOf(c);
// 根据线程数量判断是否工作线程需要被移除
boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;
// 包含对超时的判断,如果发生超时,则说明该worker已经空闲了
// keepAliveTime时间,则应该返回null,这样会使工作线程正常结束,
// 并被移除
if ((wc > maximumPoolSize || (timed && timedOut))
&& (wc > 1 || workQueue.isEmpty())) {
if (compareAndDecrementWorkerCount(c))
return null;
continue;
}
try {
// 超时获取任务
Runnable r = timed ?
workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
workQueue.take();
// 如果在keepAliveTime时间内获取到任务则返回
if (r != null)
return r;
// 否则将超时标志设置为true
timedOut = true;
} catch (InterruptedException retry) {
timedOut = false;
}
}
}
5.shutdown和shutdownNow以及awaitTermination源码分析
public void shutdown() {
final ReentrantLock mainLock = this.mainLock;
mainLock.lock(); // 获取全局锁
try {
checkShutdownAccess(); // 调用线程权限检查
advanceRunState(SHUTDOWN); // 将runState状态设为SHUTDOWN
interruptIdleWorkers(); // 中断所有空闲状态的线程,通过获取Worker的锁来实现
onShutdown(); // hook for ScheduledThreadPoolExecutor
} finally {
mainLock.unlock();
}
tryTerminate(); // 尝试终止,执行terminated hook方法
}
public List<Runnable> shutdownNow() {
List<Runnable> tasks;
final ReentrantLock mainLock = this.mainLock;
mainLock.lock();
try {
checkShutdownAccess();
advanceRunState(STOP); //将runState状态设为SHUTDOWN
interruptWorkers(); // 直接调用worker的中断方法,对所有worker进行中断
tasks = drainQueue(); // 队列 --> 集合
} finally {
mainLock.unlock();
}
tryTerminate();
return tasks;
}
public boolean awaitTermination(long timeout, TimeUnit unit)
throws InterruptedException {
long nanos = unit.toNanos(timeout);
final ReentrantLock mainLock = this.mainLock;
mainLock.lock();
try {
for (;;) {
// 如果所有提交的任务已经完成,则立刻返回true
if (runStateAtLeast(ctl.get(), TERMINATED))
return true;
// 已经超时,则返回false
if (nanos <= 0)
return false;
// 进入等待,直到被通知、中断、超时,则返回剩余的间,
// 如果返回值小于等于0,则表示是超时返回
nanos = termination.awaitNanos(nanos);
}
} finally {
mainLock.unlock();
}
}
4. 合理的配置线程池
要想合理的配置线程池,就必须首先分析任务特性,可以从以下几个角度来进行分析:
- 任务的性质:CPU密集型任务,IO密集型任务和混合型任务。
- 任务的优先级:高,中和低。
- 任务的执行时间:长,中和短。
- 任务的依赖性:是否依赖其他系统资源,如数据库连接。
任务性质不同的任务可以用不同规模的线程池分开处理。CPU密集型任务配置尽可能少的线程数量,如配置Ncpu+1个线程的线程池。IO密集型任务则由于需要等待IO操作,线程并不是一直在执行任务,则配置尽可能多的线程,如2*Ncpu。混合型的任务,如果可以拆分,则将其拆分成一个CPU密集型任务和一个IO密集型任务,只要这两个任务执行的时间相差不是太大,那么分解后执行的吞吐率要高于串行执行的吞吐率,如果这两个任务执行时间相差太大,则没必要进行分解。我们可以通过Runtime.getRuntime().availableProcessors()方法获得当前设备的CPU个数。
优先级不同的任务可以使用优先级队列PriorityBlockingQueue来处理。它可以让优先级高的任务先得到执行,需要注意的是如果一直有优先级高的任务提交到队列里,那么优先级低的任务可能永远不能执行。
执行时间不同的任务可以交给不同规模的线程池来处理,或者也可以使用优先级队列,让执行时间短的任务先执行。
依赖数据库连接池的任务,因为线程提交SQL后需要等待数据库返回结果,如果等待的时间越长CPU空闲时间就越长,那么线程数应该设置越大,这样才能更好的利用CPU。
建议使用有界队列:有界队列能增加系统的稳定性和预警能力,可以根据需要设大一点,比如几千。有一次我们组使用的后台任务线程池的队列和线程池全满了,不断的抛出抛弃任务的异常,通过排查发现是数据库出现了问题,导致执行SQL变得非常缓慢,因为后台任务线程池里的任务全是需要向数据库查询和插入数据的,所以导致线程池里的工作线程全部阻塞住,任务积压在线程池里。如果当时我们设置成无界队列,线程池的队列就会越来越多,有可能会撑满内存,导致整个系统不可用,而不只是后台任务出现问题。当然我们的系统所有的任务是用的单独的服务器部署的,而我们使用不同规模的线程池跑不同类型的任务,但是出现这样问题时也会影响到其他任务。
5. 线程池的监控
通过线程池提供的参数进行监控。线程池里有一些属性在监控线程池的时候可以使用
- taskCount:线程池需要执行的任务数量。
- completedTaskCount:线程池在运行过程中已完成的任务数量。小于或等于taskCount。
- largestPoolSize:线程池曾经创建过的最大线程数量。通过这个数据可以知道线程池是否满过。如等于线程池的最大大小,则表示线程池曾经满了。
- getPoolSize:线程池的线程数量。如果线程池不销毁的话,池里的线程不会自动销毁,所以这个大小只增不减。
- getActiveCount:获取活动的线程数。
- 通过扩展线程池进行监控。通过继承线程池并重写线程池的beforeExecute,afterExecute和terminated方法,我们可以在任务执行前,执行后和线程池关闭前干一些事情。如监控任务的平均执行时间,最大执行时间和最小执行时间等。
文章参考: