线程池如何实现的线程复用及源码分析

目录

线程池创建流程图​

源码分析

总结【精华】

常见问题


线程池创建流程图

源码分析

 

 线程池核心类:ThreadPoolExecutor,其构造方法维护了很多线程池关键参数,比如:corePoolSize、workQueue、handler等等,其都是ThreadPoolExecutor类的成员变量。

public ThreadPoolExecutor(int corePoolSize,
                              int maximumPoolSize,
                              long keepAliveTime,
                              TimeUnit unit,
                              BlockingQueue<Runnable> workQueue,
                              ThreadFactory threadFactory,
                              RejectedExecutionHandler handler) {
        if (corePoolSize < 0 ||
            maximumPoolSize <= 0 ||
            maximumPoolSize < corePoolSize ||
            keepAliveTime < 0)
            throw new IllegalArgumentException();
        if (workQueue == null || threadFactory == null || handler == null)
            throw new NullPointerException();
        this.acc = System.getSecurityManager() == null ?
                null :
                AccessController.getContext();
        this.corePoolSize = corePoolSize;
        this.maximumPoolSize = maximumPoolSize;
        this.workQueue = workQueue;
        this.keepAliveTime = unit.toNanos(keepAliveTime);
        this.threadFactory = threadFactory;
        this.handler = handler;
    }

另外还有不可忽视的一个属性:workers,这个Set集合就是我们所说的线程池,里面的每一个Worker就是一个工作线程,而线程为什么可以复用,都与这个Worker息息相关。

private final HashSet<Worker> workers = new HashSet<Worker>();

下面来看Worker类

  • 发现其继承了AQS
  • 实现了Runnable接口,代表Worker中一定有一个run方法来执行任务
  • 构造方法中this.thread = getThreadFactory().newThread(this);代表启动线程工厂创建一个线程,而入参this代表Worker,于是只要该thread被启动,Worker的run方法就会被执行。
public class ThreadPoolExecutor extends AbstractExecutorService {
    private final class Worker extends AbstractQueuedSynchronizer implements Runnable {
        // 当前Worker的具体工作线程
        final Thread thread;

        // 刚提交过来的Runnable任务,可能是核心线程的也可能是最大线程的
        // 此Worker之后可能会消费队列中的任务
        Runnable firstTask;

        // 构造方法初始化以上两个参数
        Worker(Runnable firstTask) {
            setState(-1); // inhibit interrupts until runWorker
            this.firstTask = firstTask;
            this.thread = getThreadFactory().newThread(this);
        }
        ...
    }
}

下面来看threadPoolExecutor.execute方法

public void execute(Runnable command) {
        if (command == null)
            throw new NullPointerException();
        int c = ctl.get();

        // 如果当前工作线程数 < 核心线程数,则创建一个核心Worker,并把任务传递给Worker
        if (workerCountOf(c) < corePoolSize) {
            // addWorker方法第二个参数如果是true代表是核心线程
            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);
        }

        // 当队列也满了,则开始创建最大线程数(临时Worker)
        else if (!addWorker(command, false))
            // addWorker = false说明最大线程数满了,则开始执行丢弃策略
            reject(command);
    }

addWorker方法,用于new Worker,一个Worker维护一个Thread。每来一个Runnable任务,都会赋值一个firstTask,因为只有创建核心线程和最大线程的Worker才会走此方法,因此firstTask可能是核心线程的也可能是最大线程的Runnable,创建出来后添加到workers的Set集合,然后调用Worker维护的Thread.start开启线程。

/**
     * 添加Worker,一个Worker维护一个Thread
     *
     * @param firstTask 每来一个Runnable任务,都会赋值一个firstTask
     *                  当任务数量少于corePoolSize时,Worker会发生复用
     *
     * @param core 是否创建核心线程
     * @return true if successful
     */
    private boolean addWorker(Runnable firstTask, boolean core) {
        /**
         * 上半部分:通过自旋+CAS方式增加Worker数量
         */
        retry:
        // 外层自旋 检测线程的可用状态
        for (;;) {
            int c = ctl.get();
            int rs = runStateOf(c);

            if (rs >= SHUTDOWN &&
                    ! (rs == SHUTDOWN &&
                            firstTask == null &&
                            ! workQueue.isEmpty()))
                return false;

            // 内层自旋
            for (;;) {
                int wc = workerCountOf(c);// 获取当前工作线程的数量
                // worker数量超过容量直接返回false
                // core=true看是否大于核心线程数,否则看是否大于最大线程数;大于了返回false
                if (wc >= CAPACITY ||
                        wc >= (core ? corePoolSize : maximumPoolSize))
                    return false;
                // CAS增加Worker数量
                if (compareAndIncrementWorkerCount(c))
                    break retry;
                c = ctl.get();
                // 线程状态发生改变,外层重新自旋
                if (runStateOf(c) != rs)
                    continue retry;
            }
        }

        /**
         * 下半部分:ReentrantLock加锁,创建Worker
         */
        boolean workerStarted = false;
        boolean workerAdded = false;
        Worker w = null;
        try {
            // Worker继承了AQS,实现了Runnable,因此重写run方法
            // 而Runnable的run方法执行是通过start方法
            w = new Worker(firstTask);
            final Thread t = w.thread;
            if (t != null) {
                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的Set集合
                        workers.add(w);
                        int s = workers.size();
                        if (s > largestPoolSize)
                            largestPoolSize = s;
                        workerAdded = true;
                    }
                } finally {
                    mainLock.unlock();
                }
                if (workerAdded) {
                    t.start();// 启动Worker线程,执行Worker类的run方法
                    workerStarted = true;
                }
            }
        } finally {
            if (! workerStarted)
                addWorkerFailed(w);
        }
        return workerStarted;
    }

thread.start调用的Worker中的run方法,run方法又调用了runWorker方法

public void run() {
    runWorker(this);
}

死循环中是线程复用的关键参数!

while (task != null || (task = getTask()) != null)遵循短路原则

task是从Worker中获取,而Worker中包含核心线程及非核心线程,但不包含队列中的线程。因此首先执行核心线程及非核心线程中的任务,当这俩执行完了,才会从队列中取。

第一次执行完task后,finally中会将task清空,于是getTask方法从队列中获取任务,因此只要线程池中或者队列中有任务,Worker就会一直执行。而第二次从getTask中获取任务,依然使用的上一次Worker中的task,从而实现了线程复用。

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 ((runStateAtLeast(ctl.get(), STOP) ||
                     (Thread.interrupted() &&
                      runStateAtLeast(ctl.get(), STOP))) &&
                    !wt.isInterrupted())
                    wt.interrupt();
                try {
                    beforeExecute(wt, task);
                    Throwable thrown = null;
                    try {
                        // Runnable task调用run方法而不是start方法,实际上并没有开启新的线程
                        // 最终调用的就是我们自己代码中重写的run方法
                        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);
        }
    }

getTask方法从任务队列中获取任务,

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

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

            // 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;

           
            if ((wc > maximumPoolSize || (timed && timedOut))
                && (wc > 1 || workQueue.isEmpty())) {
                // CAS操作-1
                if (compareAndDecrementWorkerCount(c))
                    return null;
                continue;
            }

            try {
                // 从队列中获取线程,如果已经超过了核心线程数,则在一定时间内获取任务,否则阻塞获取
                Runnable r = timed ?
                    workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                    workQueue.take();
                if (r != null)
                    return r;
                timedOut = true;
            } catch (InterruptedException retry) {
                timedOut = false;
            }
        }
    }

总结【精华】

  1. 线程池任务提交流程:execute-->创建Worker-->worker.thread.start()-->Wroker.run()-->runWorker-->worker.firstTask.run()
  2. ThreadPoolExecutor维护了一个Set的workers集合,就是我们所说的线程池,每一个Worker维护了一个工作线程,workers.size即当前threadPool的size
  3. Worker类实现了Runnable接口,Worker的构造方法中this.thread = getThreadFactory().newThread(this);代表启动线程工厂创建一个线程,而入参this代表Worker,于是只要该thread被启动,Worker的run方法就会被执行。
  4. 最终执行Worker会首先执行其firstTask,执行之后把task清空,然后从队列中获取,仍然用Worker维护的task来执行run方法,因为run方法不是start,不会开启新的线程,且Worker未做改变,实现了线程复用
  5. 关于执行顺序:首先addWroker核心线程并执行,超过之后开始入队,刚刚的Worker执行完核心任务后从队列中继续获取任务执行。此时如果继续入队,并且队列已满,则会addWorker临时线程并执行(临时Worker相较于先前的队列中任务属于插队执行,但并不是严格的插队,取决于队首元素进入核心Worker的速度),刚刚执行最大线程的Worker继续从队列中获取任务,因此执行的先后顺序并不绝对,但大概顺序是:核心线程任务-->最大线程任务-->队列任务

常见问题

若corePoolSize=10,maxinumPoolSize=20,queueSize=10;一次性来了30个线程,其执行顺序是如何?

1-10进入核心线程,11-20进入队列,21-30进入最大线程;

首先执行1-10的核心线程,其次执行21-30的最大线程,最后执行11-20队列中的线程。【具体见总结5】

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值