摘要:联合ThreadPoolExecutor类的源码深度剖析线程池执行工作的整体流程。

本文分享自华为云社区《【高并发】通过ThreadPoolExecutor类的源码深度解析线程池执行工作的外围流程》,作者: 冰 河。

ThreadPoolExecutor类中存在一个workers工作线程汇合,用户能够向线程池中增加须要执行的工作,workers汇合中的工作线程能够间接执行工作,或者从工作队列中获取工作后执行。ThreadPoolExecutor类中提供了整个线程池从创立到执行工作,再到沦亡的整个流程办法。本文,就联合ThreadPoolExecutor类的源码深度剖析线程池执行工作的整体流程。

在ThreadPoolExecutor类中,线程池的逻辑次要体现在execute(Runnable)办法,addWorker(Runnable, boolean)办法,addWorkerFailed(Worker)办法和回绝策略上,接下来,咱们就深入分析这几个外围办法。

execute(Runnable)办法

execute(Runnable)办法的作用是提交Runnable类型的工作到线程池中。咱们先看下execute(Runnable)办法的源码,如下所示。

public void execute(Runnable command) {    //如果提交的工作为空,则抛出空指针异样    if (command == null)        throw new NullPointerException();    //获取线程池的状态和线程池中线程的数量    int c = ctl.get();    //线程池中的线程数量小于corePoolSize的值    if (workerCountOf(c) < corePoolSize) {        //从新开启线程执行工作        if (addWorker(command, true))            return;        c = ctl.get();    }    //如果线程池处于RUNNING状态,则将工作增加到阻塞队列中    if (isRunning(c) && workQueue.offer(command)) {        //再次获取线程池的状态和线程池中线程的数量,用于二次查看        int recheck = ctl.get();        //如果线程池没有未处于RUNNING状态,从队列中删除工作        if (! isRunning(recheck) && remove(command))            //执行回绝策略            reject(command);        //如果线程池为空,则向线程池中增加一个线程        else if (workerCountOf(recheck) == 0)            addWorker(null, false);    }    //工作队列已满,则新增worker线程,如果新增线程失败,则执行回绝策略    else if (!addWorker(command, false))        reject(command);}

整个工作的执行流程,咱们能够简化成下图所示。

接下来,咱们拆解execute(Runnable)办法,具体分析execute(Runnable)办法的执行逻辑。

(1)线程池中的线程数是否小于corePoolSize外围线程数,如果小于corePoolSize外围线程数,则向workers工作线程汇合中增加一个外围线程执行工作。代码如下所示。

//线程池中的线程数量小于corePoolSize的值if (workerCountOf(c) < corePoolSize) {    //从新开启线程执行工作    if (addWorker(command, true))        return;    c = ctl.get();}

(2)如果线程池中的线程数量大于corePoolSize外围线程数,则判断以后线程池是否处于RUNNING状态,如果处于RUNNING状态,则增加工作到待执行的工作队列中。留神:这里向工作队列增加工作时,须要判断线程池是否处于RUNNING状态,只有线程池处于RUNNING状态时,能力向工作队列增加新工作。否则,会执行回绝策略。代码如下所示。

if (isRunning(c) && workQueue.offer(command)) 

(3)向工作队列中增加工作胜利,因为其余线程可能会批改线程池的状态,所以这里须要对线程池进行二次查看,如果以后线程池的状态不再是RUNNING状态,则须要将增加的工作从工作队列中移除,执行后续的回绝策略。如果以后线程池依然处于RUNNING状态,则判断线程池是否为空,如果线程池中不存在任何线程,则新建一个线程增加到线程池中,如下所示。

//再次获取线程池的状态和线程池中线程的数量,用于二次查看int recheck = ctl.get();//如果线程池没有未处于RUNNING状态,从队列中删除工作if (! isRunning(recheck) && remove(command))    //执行回绝策略    reject(command);//如果线程池为空,则向线程池中增加一个线程else if (workerCountOf(recheck) == 0)    addWorker(null, false);

(4)如果在步骤(3)中向工作队列中增加工作失败,则尝试开启新的线程执行工作。此时,如果线程池中的线程数量曾经大于线程池中的最大线程数maximumPoolSize,则不能再启动新线程。此时,示意线程池中的工作队列已满,并且线程池中的线程已满,须要执行回绝策略,代码如下所示。

//工作队列已满,则新增worker线程,如果新增线程失败,则执行回绝策略else if (!addWorker(command, false))    reject(command);

这里,咱们将execute(Runnable)办法拆解,联合流程图来了解线程池中工作的执行流程就比较简单了。能够这么说,execute(Runnable)办法的逻辑基本上就是个别线程池的执行逻辑,了解了execute(Runnable)办法,就根本了解了线程池的执行逻辑。

留神:无关ScheduledThreadPoolExecutor类和ForkJoinPool类执行线程池的逻辑,在【高并发专题】系列文章中的后文中会具体阐明,了解了这些类的执行逻辑,就根本全面把握了线程池的执行流程。

在剖析execute(Runnable)办法的源码时,咱们发现execute(Runnable)办法中多处调用了addWorker(Runnable, boolean)办法,接下来,咱们就一起剖析下addWorker(Runnable, boolean)办法的逻辑。

addWorker(Runnable, boolean)办法

总体上,addWorker(Runnable, boolean)办法能够分为三局部,第一局部是应用CAS平安的向线程池中增加工作线程;第二局部是创立新的工作线程;第三局部则是将工作通过平安的并发形式增加到workers中,并启动工作线程执行工作。

接下来,咱们看下addWorker(Runnable, boolean)办法的源码,如下所示。

private boolean addWorker(Runnable firstTask, boolean core) {    //标记重试的标识    retry:    for (;;) {        int c = ctl.get();        int rs = runStateOf(c);        // 查看队列是否在某些特定的条件下为空        if (rs >= SHUTDOWN &&            ! (rs == SHUTDOWN &&               firstTask == null &&               ! workQueue.isEmpty()))            return false;        //上面循环的次要作用为通过CAS形式减少线程的个数        for (;;) {            //获取线程池中的线程数量            int wc = workerCountOf(c);            //如果线程池中的线程数量超出限度,间接返回false            if (wc >= CAPACITY ||                wc >= (core ? corePoolSize : maximumPoolSize))                return false;            //通过CAS形式向线程池新增线程数量            if (compareAndIncrementWorkerCount(c))                //通过CAS形式保障只有一个线程执行胜利,跳出最外层循环                break retry;            //从新获取ctl的值            c = ctl.get();              //如果CAS操作失败了,则须要在内循环中从新尝试通过CAS新增线程数量            if (runStateOf(c) != rs)                continue retry;        }    }     //跳出最外层for循环,阐明通过CAS新增线程数量胜利    //此时创立新的工作线程    boolean workerStarted = false;    boolean workerAdded = false;    Worker w = null;    try {        //将执行的工作封装成worker        w = new Worker(firstTask);        final Thread t = w.thread;        if (t != null) {            //独占锁,保障操作workers时的同步            final ReentrantLock mainLock = this.mainLock;            mainLock.lock();            try {                //此处须要从新查看线程池状态                //起因是在取得锁之前可能其余的线程扭转了线程池的状态                int rs = runStateOf(ctl.get());                 if (rs < SHUTDOWN ||                    (rs == SHUTDOWN && firstTask == null)) {                    if (t.isAlive())                        throw new IllegalThreadStateException();                    //向worker中增加新工作                    workers.add(w);                    int s = workers.size();                    if (s > largestPoolSize)                        largestPoolSize = s;                    //将是否增加了新工作的标识设置为true                    workerAdded = true;                }            } finally {                //开释独占锁                mainLock.unlock();            }            //增加新任胜利,则启动线程执行工作            if (workerAdded) {                t.start();                //将工作是否曾经启动的标识设置为true                workerStarted = true;            }        }    } finally {        //如果工作未启动或启动失败,则调用addWorkerFailed(Worker)办法        if (! workerStarted)            addWorkerFailed(w);    }    //返回是否启动工作的标识    return workerStarted;}

乍一看,addWorker(Runnable, boolean)办法还蛮长的,这里,咱们还是将addWorker(Runnable, boolean)办法进行拆解。

(1)查看工作队列是否在某些特定的条件下为空,代码如下所示。

// 查看队列是否在某些特定的条件下为空if (rs >= SHUTDOWN &&    ! (rs == SHUTDOWN &&       firstTask == null &&       ! workQueue.isEmpty()))    return false;

(2)在通过步骤(1)的校验后,则进入内层for循环,在内层for循环中通过CAS来减少线程池中的线程数量,如果CAS操作胜利,则间接退出双重for循环。如果CAS操作失败,则查看以后线程池的状态是否产生了变动,如果线程池的状态产生了变动,则通过continue关键字从新通过外层for循环校验工作队列,测验通过再次执行内层for循环的CAS操作。如果线程池的状态没有发生变化,此时上一次CAS操作失败了,则持续尝试CAS操作。代码如下所示。

for (;;) {    //获取线程池中的线程数量    int wc = workerCountOf(c);    //如果线程池中的线程数量超出限度,间接返回false    if (wc >= CAPACITY ||        wc >= (core ? corePoolSize : maximumPoolSize))        return false;    //通过CAS形式向线程池新增线程数量    if (compareAndIncrementWorkerCount(c))        //通过CAS形式保障只有一个线程执行胜利,跳出最外层循环        break retry;    //从新获取ctl的值    c = ctl.get();      //如果CAS操作失败了,则须要在内循环中从新尝试通过CAS新增线程数量    if (runStateOf(c) != rs)        continue retry;}

(3)CAS操作胜利后,示意向线程池中胜利增加了工作线程,此时,还没有线程去执行工作。应用全局的独占锁mainLock来将新增的工作线程Worker对象平安的增加到workers中。

总体逻辑就是:创立新的Worker对象,并获取Worker对象中的执行线程,如果线程不为空,则获取独占锁,获取锁胜利后,再次查看线线程的状态,这是防止在获取独占锁之前其余线程批改了线程池的状态,或者敞开了线程池。如果线程池敞开,则须要开释锁。否则将新减少的线程增加到工作汇合中,开释锁并启动线程执行工作。将是否启动线程的标识设置为true。最初,判断线程是否启动,如果没有启动,则调用addWorkerFailed(Worker)办法。最终返回线程是否起送的标识。

//跳出最外层for循环,阐明通过CAS新增线程数量胜利//此时创立新的工作线程boolean workerStarted = false;boolean workerAdded = false;Worker w = null;try {    //将执行的工作封装成worker    w = new Worker(firstTask);    final Thread t = w.thread;    if (t != null) {        //独占锁,保障操作workers时的同步        final ReentrantLock mainLock = this.mainLock;        mainLock.lock();        try {            //此处须要从新查看线程池状态            //起因是在取得锁之前可能其余的线程扭转了线程池的状态            int rs = runStateOf(ctl.get());             if (rs < SHUTDOWN ||                (rs == SHUTDOWN && firstTask == null)) {                if (t.isAlive())                    throw new IllegalThreadStateException();                //向worker中增加新工作                workers.add(w);                int s = workers.size();                if (s > largestPoolSize)                    largestPoolSize = s;                //将是否增加了新工作的标识设置为true                workerAdded = true;            }        } finally {            //开释独占锁            mainLock.unlock();        }        //增加新任胜利,则启动线程执行工作        if (workerAdded) {            t.start();            //将工作是否曾经启动的标识设置为true            workerStarted = true;        }    }} finally {    //如果工作未启动或启动失败,则调用addWorkerFailed(Worker)办法    if (! workerStarted)        addWorkerFailed(w);}//返回是否启动工作的标识return workerStarted;

addWorkerFailed(Worker)办法

在addWorker(Runnable, boolean)办法中,如果增加工作线程失败或者工作线程启动失败时,则会调用addWorkerFailed(Worker)办法,上面咱们就来看看addWorkerFailed(Worker)办法的实现,如下所示。

private void addWorkerFailed(Worker w) {    //获取独占锁    final ReentrantLock mainLock = this.mainLock;    mainLock.lock();    try {        //如果Worker工作不为空        if (w != null)            //将工作从workers汇合中移除            workers.remove(w);        //通过CAS将工作数量减1        decrementWorkerCount();        tryTerminate();    } finally {        //开释锁        mainLock.unlock();    }}

addWorkerFailed(Worker)办法的逻辑就比较简单了,获取独占锁,将工作从workers中移除,并且通过CAS将工作的数量减1,最初开释锁。

回绝策略

咱们在剖析execute(Runnable)办法时,线程池会在适当的时候调用reject(Runnable)办法来执行相应的回绝策略,咱们看下reject(Runnable)办法的实现,如下所示。

final void reject(Runnable command) {    handler.rejectedExecution(command, this);}

通过代码,咱们发现调用的是handler的rejectedExecution办法,handler又是个什么鬼,咱们持续跟进代码,如下所示。

private volatile RejectedExecutionHandler handler;

再看看RejectedExecutionHandler是个啥类型,如下所示。

package java.util.concurrent;public interface RejectedExecutionHandler {    void rejectedExecution(Runnable r, ThreadPoolExecutor executor);}

能够发现RejectedExecutionHandler是个接口,定义了一个rejectedExecution(Runnable, ThreadPoolExecutor)办法。既然RejectedExecutionHandler是个接口,那咱们就看看有哪些类实现了RejectedExecutionHandler接口。

看到这里,咱们发现RejectedExecutionHandler接口的实现类正是线程池默认提供的四种回绝策略的实现类。

至于reject(Runnable)办法中具体会执行哪个类的回绝策略,是依据创立线程池时传递的参数决定的。如果没有传递回绝策略,则默认会执行AbortPolicy类的回绝策略。否则会执行传递的类的回绝策略。

在创立线程池时,除了可能传递JDK默认提供的回绝策略外,还能够传递自定义的回绝策略。如果想应用自定义的回绝策略,则只须要实现RejectedExecutionHandler接口,并重写rejectedExecution(Runnable, ThreadPoolExecutor)办法即可。例如,上面的代码。

public class CustomPolicy implements RejectedExecutionHandler {    public CustomPolicy() { }    public void rejectedExecution(Runnable r, ThreadPoolExecutor e) {        if (!e.isShutdown()) {            System.out.println("应用调用者所在的线程来执行工作")            r.run();        }    }}

应用如下形式创立线程池。

new ThreadPoolExecutor(0, Integer.MAX_VALUE,                       60L, TimeUnit.SECONDS,                       new SynchronousQueue<Runnable>(),                       Executors.defaultThreadFactory(),               new CustomPolicy());

至此,线程池执行工作的整体外围逻辑剖析完结。

点击关注,第一工夫理解华为云陈腐技术~