线程池源码分析-jdk1.8

openjdk下载地址

http://hg.openjdk.java.net/jdk8u/jdk8u/jdk/tags

java自带的线程池的逻辑步骤主要以下几步 伪代码实现

提交任务
if(核心线程池已满){
    if(任务队列已满){
        if(线程池已满){
            按照策略处理无法执行的任务
        }else{
            创建线程执行任务
        }
    }else{
        将任务存储到任务队列
    }
}else{
    创建线程，执行任务
}

向线程池提交任务源码

public void execute(Runnable command) {
        //运行的任务不可为空
        if (command == null)
            throw new NullPointerException();
        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);
    }

添加工作线程

private boolean addWorker(Runnable firstTask, boolean core) {
        retry:
        for (;;) {
            int c = ctl.get();
            int rs = runStateOf(c);
            //如果线程池运行状态不是正在运行状态，并且不是 【状态为shutdown，任务为null 并且工作队列不为空】 情况 则添加新的工作线程失败
            if (rs >= SHUTDOWN &&
                ! (rs == SHUTDOWN &&
                   firstTask == null &&
                   ! workQueue.isEmpty()))
                return false;

            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
                //如果线程池运行状态改变了，则跳转到执行外循环执行
                if (runStateOf(c) != rs)
                    continue retry;
                //继续内循环原子方式增加工作线程的数量
                // else CAS failed due to workerCount change; retry inner loop
            }
        }
        
        //是否成功启动
        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 {
            //添加失败，从集合移除新创建的工作线程，线程池线程数量线程安全的减1
            if (! workerStarted)
                addWorkerFailed(w);
        }
        return workerStarted;
    }

Worker内部类

private final class Worker
        extends AbstractQueuedSynchronizer
        implements Runnable
    {
        private static final long serialVersionUID = 6138294804551838833L;
        //执行任务的线程
        final Thread thread;
        //第一个执行的任务
        Runnable firstTask;
        //线程完成的任务数量
        volatile long completedTasks;
        
        Worker(Runnable firstTask) {
            setState(-1); // inhibit interrupts until runWorker
            this.firstTask = firstTask;
            //创建新的线程
            this.thread = getThreadFactory().newThread(this);
        }

        public void run() {
            //真正执行任务的方法
            runWorker(this);
        }

        protected boolean isHeldExclusively() {
            return getState() != 0;
        }

        protected boolean tryAcquire(int unused) {
            if (compareAndSetState(0, 1)) {
                setExclusiveOwnerThread(Thread.currentThread());
                return true;
            }
            return false;
        }

        protected boolean tryRelease(int unused) {
            setExclusiveOwnerThread(null);
            setState(0);
            return true;
        }

        public void lock()        { acquire(1); }
        public boolean tryLock()  { return tryAcquire(1); }
        public void unlock()      { release(1); }
        public boolean isLocked() { return isHeldExclusively(); }

        void interruptIfStarted() {
            Thread t;
            if (getState() >= 0 && (t = thread) != null && !t.isInterrupted()) {
                try {
                    t.interrupt();
                } catch (SecurityException ignore) {
                }
            }
        }
    }

执行任务

 final void runWorker(Worker w) {
        Thread wt = Thread.currentThread();
        Runnable task = w.firstTask;
        w.firstTask = null;
        w.unlock(); // allow interrupts
        boolean completedAbruptly = true;
        try {
            //任务不为空就一直执行，为空则线程执行结束，自然销毁 getTask()从阻塞队列获取任务，没有任务时会阻塞，并挂起当前线程，不会消耗CPU资源
            while (task != null || (task = getTask()) != null) {
                //获取锁，防止任务执行中被其他线程打断
                w.lock();
                
                //如果线程池状态为stop 并且线程不是中断状态 则打断执行线程
                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);
        }
    }

从阻塞队列获取执行任务

private Runnable getTask() {
        boolean timedOut = false; // Did the last poll() time out?

        for (;;) {
            int c = ctl.get();
            int rs = runStateOf(c);

            //如果状态为 STOP TIDYING TERMINATED 状态 或者队列为空 则停止执行任务，销毁线程
            //如果是SHUTDOWN状态 但是队列不为空，则继续将任务执行结束
            if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
                //更新线程池线程数量
                decrementWorkerCount();
                return null;
            }

            int wc = workerCountOf(c);

            //如果线程数量大于核心线程数量 或者 允许核心线程超时销毁 则 从阻塞队列获取任务时设置超时时间
            boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;

            //如果线程超过线程池最大线程数量，并且队列是空的 或则线程数量大于1  则销毁线程
            if ((wc > maximumPoolSize || (timed && timedOut))
                && (wc > 1 || workQueue.isEmpty())) {
                if (compareAndDecrementWorkerCount(c))
                    return null;
                continue;
            }

            try {
                //如果是超时获取，则keepAliveTime时间后，线程仍然空闲，则销毁线程，否则一直阻塞等待直到有任务执行为止
                Runnable r = timed ?
                    workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                    workQueue.take();
                if (r != null)
                    return r;
                timedOut = true;
            } catch (InterruptedException retry) {
                timedOut = false;
            }
        }
    }