线程池源码之addWorker方法解析

一切都在源码中

private boolean addWorker(Runnable firstTask, boolean core) {
    retry:
    // 每次for循环都需要获取最新的ctl值
    for (int c = ctl.get();;) {
        /*
        * 这个地方的判断为true时可以分成三种情况
        * 1、线程池的状态是大于等于 STOP的 这个判断为true
        * 2、线程池状态是SHUTDOWN 但是 firstTask 是null 对应execute 的 else if (!addWorker(command, false)) 状态大于等于SHUTDOWN时不接受任务
        * 3、线程池状态是SHUTDOWN 工作队列已经空了 对应execute的 在状态大于等于SHUTDOWN时用addWorker(null, false) 执行队列里面的任务
        */
        if (runStateAtLeast(c, SHUTDOWN)
                && (runStateAtLeast(c, STOP) || firstTask != null || workQueue.isEmpty()) )
            return false;

        for (;;) {
            // 如果core是true使用的核心线程数配置 否则使用maximumPoolSize
            if (workerCountOf(c)
                >= ((core ? corePoolSize : maximumPoolSize) & COUNT_MASK))
                return false;
            // 使用CAS的方法给ctl的worker的数量加1 成功则跳出最外层循环
            if (compareAndIncrementWorkerCount(c))
                break retry;
            // CAS不成功则重新获取ctl的值 因为不成功ctl的值一定变了 CAS嘛 
            c = ctl.get();
            // 如果不成功的原因是状态变了 就重新进行外层循环
            if (runStateAtLeast(c, SHUTDOWN))
                continue retry;
        }
    }

    // 能走到这里一定是说明CAS成功了 那么就可以进行创建worker执行任务了
    // woker是否执行了start
    boolean workerStarted = false;
    // worker 是否添加到workers成功
    boolean workerAdded = false;
    Worker w = null;
    try {
        // 构建一个worker thread的target是worker所以调用t.start会执行worker的run方法，最后调用到runWorker方法
        w = new Worker(firstTask);
        // worker 的thread是 线程工厂的newThread方法创建的
        final Thread t = w.thread;
        if (t != null) {
            // 需要操作workers 加锁执行
            final ReentrantLock mainLock = this.mainLock;
            mainLock.lock();
            try {
                int c = ctl.get();

                // 如果线程池状态是running 或者 是shutdown但是firstTask是null(kanexecute方法 防止没有worker执行队列里面的任务)
                if (isRunning(c) ||
                    (runStateLessThan(c, STOP) && firstTask == null)) {
                    // 如果线程的状态不是NEW说明线程不是经过线程池开启的 抛异常
                    if (t.getState() != Thread.State.NEW)
                        throw new IllegalThreadStateException();
                    // 把worker添加到workers里面
                    workers.add(w);
                    workerAdded = true;
                    // 更新当前最大线程数量 maximumPoolSize 和 corePoolSize可以在线程池创建之后动态修改的
                    int s = workers.size();
                    if (s > largestPoolSize)
                        largestPoolSize = s;
                }
            } finally {
                mainLock.unlock();
            }
            if (workerAdded) {
                /*
                * 如果添加成功 就启动这个线程
                * 上面线程池状态判断没通过 或者 线程的状态不是NEW 就不会执行 workerAdded = true;
                * 线程就不会启动 workerStarted 就是false
                */
                t.start();
                workerStarted = true;
            }
        }
    } finally {
        if (! workerStarted)
            // 如果没有执行过t.start() 就要把这个woker从workers里面剔除并且 ctl里面worker数量减一
            addWorkerFailed(w);
    }
    return workerStarted;
}