Java线程池实现原理详解,看完一篇,学会线程池?

一、为什么要使用线程池?

线程池提供了一种限制和管理资源(线程、任务)的方式。

这里借用《Java 并发编程的艺术》提到的来说一下使用线程池的好处

  • 降低资源消耗:通过重复利用已创建的线程降低线程创建和销毁造成的消耗。
  • 提高响应速度:当任务到达时,任务可以不需要等待创建线程就能立即执行。
  • 提高线程的可管理性:线程是稀缺资源,如果无线的创建,不仅会消耗系统资源,还会降低系统的稳定性,使用线程池可以进行统一的分配,调优和监控。

二、ThreadPoolExecutor类分析

Java线程池主要由Executor框架实现,Executor 框架不仅包括了线程池的管理,还提供了线程工厂、队列以及拒绝策略等,Executor 框架让并发编程变得更加简单。

线程池实现类ThreadPoolExecutorExecutor框架最核心的类,我们就从这个类的学习线程池的实现原理。

核心属性

public class ThreadPoolExecutor extends AbstractExecutorService {
    
    // 控制变量-存放状态和线程数 32位, 高3位存放状态, 低29位存放线程数
    private final AtomicInteger ctl = new AtomicInteger(ctlOf(RUNNING, 0));
    
    // 任务队列, 必须是阻塞队列
    private final BlockingQueue workQueue;
    
    // 工作线程集合,存放线程池中所有的(活跃的)工作线程,只有在持有全局锁mainLock的前提下才能访问此集合
    private final HashSet workers = new HashSet<>();
    
    // 全局锁
    private final ReentrantLock mainLock = new ReentrantLock();
    
    // awaitTermination方法使用的等待条件变量
    private final Condition termination = mainLock.newCondition();
    
    // 记录峰值线程数
    private int largestPoolSize;
    
    // 记录完成成功执行的任务数
    private long completedTaskCount;
    
    // 线程工厂, 用于创建新的线程实例
    private volatile ThreadFactory threadFactory;
    
    // 拒绝执行处理器, 对应不同的拒绝策略
    private volatile RejectedExecutionHandler handler;
    
    // 空闲线程等待任务的时间周期, 单位是纳秒
    private volatile long keepAliveTime;
    
    // 是否允许核心线程超时, 如果为true则keepAliveTime对核心线程也生效
    private volatile boolean allowCoreThreadTimeOut;
    
    // 核心线程数
    private volatile int corePoolSize;
    
    // 线程池容量
    private volatile int maximumPoolSize;
 	
    // 省略其他代码
}
复制代码

构造方法

参数最多的构造方法:

public ThreadPoolExecutor(int corePoolSize,
                          int maximumPoolSize,
                          long keepAliveTime,
                          TimeUnit unit,
                          BlockingQueue 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;
}
复制代码

可以根据这个构造方法自定义线程数、线程池容量(最大线程数)、空闲线程等待任务周期、任务队列、线程工厂、拒绝策略。

在《阿里巴巴Java开发手册》“并发处理”这一章节,明确指出线程资源必须通过线程池创建,不允许在应用中自行显示创建线程,这是因为使用线程池创建线程可以减少在创建和销毁线程上所消耗的时间以及系统资源的开销。

《阿里巴巴Java开发手册》中还强制不能使用Executors去创建线程池,而是通过上面的ThreadPoolExecutor的构造方式创建,这样的处理方式可以让写的同学更加明确线程池的运行规则,避免资源耗尽的风险。

下面简单分析一下每个参数的含义和作用:

  • corePoolSize:核心线程数量。
  • maximumPoolSize:最大线程数量,也就是线程池的容量。
  • keepAliveTime:线程空闲等待时间,也和工作线程的生命周期有关。
  • unit:线程空闲时间的单位,最终会转为成纳秒。
  • workQueue:等待队列或者叫任务队列。
  • ThreadFactory:创建线程的工厂,默认使用Executors.defaultThreadFactory()作为线程池工厂实例。
  • handler:线程池的执行执行处理器,更多的时候成为拒绝策略,拒绝策略执行的时机是当阻塞队列已满、没有空闲的线程(包含核心线程和非核心线程)并且继续提交任务。提供了4种拒绝策略实现:
    • AbortPolicy:直接拒绝策略,也就是不会执行任务,直接抛出RjectedExecutionExcetion错误,默认的拒绝策略。
    • DiscardPolicy:抛弃策略,也就是直接忽略提交的任务。
    • DiscardOldestPolicy:抛弃最老任务策略,也就是通过poll()方法取出任务队列头的任务抛弃,然后执行当前提交的任务。
    • CallerRunsPolicy:调用者执行策略,也就是当前调用Executor#execute()的线程直接调用任务Runnable#run()一般不希望任务丢失会选用这种策略,但从实际角度来看,原来的异步调用意图会退化成同步调用。

状态控制

状态控制主要围绕原子整数成员变量crl

private final AtomicInteger ctl = new AtomicInteger(ctlOf(RUNNING, 0));
private static final int COUNT_BITS = Integer.SIZE - 3;
private static final int COUNT_MASK = (1 << COUNT_BITS) - 1;

private static final int RUNNING    = -1 << COUNT_BITS;
private static final int SHUTDOWN   =  0 << COUNT_BITS;
private static final int STOP       =  1 << COUNT_BITS;
private static final int TIDYING    =  2 << COUNT_BITS;
private static final int TERMINATED =  3 << COUNT_BITS;

// 通过ctl值获取运行状态
private static int runStateOf(int c)     { return c & ~COUNT_MASK; }

// 通过ctl值获取工作线程数
private static int workerCountOf(int c)  { return c & COUNT_MASK; }

// 通过运行状态和工作线程数计算ctl的值,或运算
private static int ctlOf(int rs, int wc) { return rs | wc; }

private static boolean runStateLessThan(int c, int s) {
    return c < s;
}

private static boolean runStateAtLeast(int c, int s) {
    return c >= s;
}

private static boolean isRunning(int c) {
    return c < SHUTDOWN;
}

// CAS操作线程数增加1
private boolean compareAndIncrementWorkerCount(int expect) {
    return ctl.compareAndSet(expect, expect + 1);
}

// CAS操作线程数减少1
private boolean compareAndDecrementWorkerCount(int expect) {
    return ctl.compareAndSet(expect, expect - 1);
}

// 线程数直接减少1
private void decrementWorkerCount() {
    ctl.addAndGet(-1);
}
复制代码

接下来分析一下线程池的状态变量,工作线程数量位的长度是COUNT_BITS,它的值是Integer.SIZE - 3,也就是正整数29:

我们知道,整数包装类型Integer实例的大小是4byte,一共是32位,也就是一共有32位用于存放0和1。

在ThreadPoolExecutor实现中,使用32位的整数包装类型存放工作线程数和线程状态,其中低29位用于存放工作线程数,而高3位用于存放线程池状态,所以线程池的状态最多只能有2^3种,工作线程上限数量为2^29 - 1,超过5亿,这个数量在短时间内不用考虑会超限。

接着看工作线程上线数量掩码COUNT_MASK,它的值是(1 < COUNT_BITS - 1),也就是1左移29位,再减去1,如果补全32位,它的位示图如下:

Java线程池实现原理详解,看完一篇,学会线程池?_第1张图片

然后就是线程池的状态常量,比如RUNNING状态:

// -1的补码为:111-11111111111111111111111111111
// 左移29位:  111-00000000000000000000000000000
// 10进制为:-536870912 
// 高3位111的值就是表示线程池正在处于运行状态
private final static int RUNNING = -1 << COUNT_BITS;
复制代码

线程池状态的运行状态常量:

状态名称 位图 十进制值 描述
RUNNING 111-0000... -536870912 运行中状态,可以接受新的任务和执行任务队列中的任务。
SHUTDOWN 000-0000... 0 关闭状态,不再接收新的任务,但是会执行任务队列中的任务。
STOP 001-0000... 536870912 停止状态,不再接受新的任务,也不会执行任务队列中的任务,中断所有执行中的任务。
TIDYING 010-0000... 1073741824 整理中状态,所有任务都已经执行完毕,工作线程数为0,过渡到此状态的工作线程会调用钩子方法terminated()
TERMINATED 011-0000... 1610612736 终止状态,钩子方法terminated()执行完毕。

这里还有一个比较特殊的技巧,由于运行状态值存放在高3位,所以直接可以通过十进制来比较判断线程池的状态:

RUNNING < SHUTDOWN < STOP < TIDYING < TERMINATED

下面的3个方法就是使用这种技巧:

// ctl和状态常量比较,判断是否小于
private static boolean runStateLessThan(int c, int s) {
    return c < s;
}

// ctl和状态常量比较,判断是否小于或等于
private static boolean runStateAtLeast(int c, int s) {
    return c >= s;
}

// ctl和状态常量SHUTDOWN比较,判断是否处于RUNNING状态
private static boolean isRunning(int c) {
    return c < SHUTDOWN;
}
复制代码

线程状态流转关系如下图:

Java线程池实现原理详解,看完一篇,学会线程池?_第2张图片

execute方法源码分析

线程异步执行任务的方法实现是ThreadPoolExecutor#execute()方法,我们从源码的实现来学习,源码如下:

public void execute(Runnable command) {
    // 判断任务对象非空
    if (command == null)
        throw new NullPointerExcetion();
    // 获取ctl值, 用于获取线程池状态、线程池线程数量
    int c = ctl.get();
    // 如果当前线程数小于核心线程数,则创建新的核心线程数并且执行传入的任务
    if (workerCountOf(c) < corePoolSize) {
        if (addWoker(command, true))
            // 如果创建新的核心线程成功则直接返回
            return;
        // 这里说明创建新的核心线程失败,则更新ctl的临时变量c
        c = ctl.get();
    }
    // 走到这里说明创建核心线程失败,也就是当前工作的线程数大于等于核心线程数
    // 判断线程是否处于运行中状态,如果是运行状态尝试使用非阻塞方法向任务队列放入任务
    if (isRunning(c) && workQueue.offer(command)) {
        int recheck = ctl.get();
        // 这里向任务队列投放成功,对线程池的运行中状态做二次检查
        // 如果线程池二次检查状态是非运行中状态,则从任务队列移除当前的任务,调用拒绝策略处理
        if (!isRunning() && remove(command))
            // 调用拒绝策略处理任务
            reject(command);
        // 走到这里说明有以下的前提:
        // 1、待执行的任务已经成功加入任务队列
        // 2、线程池状态可能是RUNNING
        // 3、传入的任务可能从任务队列中移除失败(移除失败的唯一可能就是任务已经被执行了)
        // 如果当前工作线程数量为0,则创建一个非核心线程并且传入的对象为null
        // 也就是创建非核心线程不会马上运行,而是等待获取任务队列的任务再执行
        else if (workerCountOf(recheck) == 0)
            // 创建一个非核心线程并且传入的任务对象为null
            addWorker(null, false);
    }
    // 走到这里说明以下:
    // 1、线程池的工作线程总数已经大于等于核心线程数
    // 2、线程池可能不是RUNNING状态
    // 3、线程池可能是RUNNING状态同时任务队列已经满了
    // 如果任务队列投放任务失败,则会尝试创建非核心线程执行任务
    else if (!addWorker(command, false))
        // 如果创建非核心线程失败,执行拒绝策略
        reject(command);
}
复制代码

上面代码的流程如下:

  1. 如果当前工作线程总数小于核心线程数corePoolSize,则直接创建核心线程去执行任务(任务实例会传入直接用于构造工作线程实例)。
  2. 如果当前工作线程总是大于等于核心线程数corePoolSize,判断线程状态是否是运行中状态,如果是运行中状态则会尝试用非阻塞方法(offer())向任务队列投放任务,如果投放成功会二次检查线程池运行状态,如果线程池是非运行中状态或者从任务队列移除当前的任务失败,则会调用拒绝策略,如果当前工作线程数量为0,则创建一个非核心线程并且传入的任务对象为null
  3. 如果任务队列投放任务失败了(任务队列满了),则会创建创建非核心线程传入任务实例执行。
  4. 如果非核心线程创建失败,则会调用拒绝策略。

这里有一个疑惑点:为什么要二次检查线程池的状态,当前工作线程数量为0,尝试创建一个非核心线程并且传入的任务对象为null?这个可以看API的解释:

如果一个任务成功加入任务队列,我们依然需要二次检查是否需要添加一个工作线程,因为所有存活的工作线程有可能在最后一次检查之后就终结了或者执行当前任务的时候线程池是否已经shutdown了,所以我们需要二次检车线程池的状态,必须时要把任务从任务队列中移除或者在没有可用的工作线程的前提的下创建一个工作线程。

execute()方法执行流程图如下:

Java线程池实现原理详解,看完一篇,学会线程池?_第3张图片

addWorker方法源码分析

addWorker()方法用于添加工作线程,源码如下:

pirvate boolean addWorker(Runnable firstTask, boolean core) {
    retry;
    for (;;) {
        int c = ctl.get();
        int rs = runStateOf(c);
        
        // 判断边界情况表,当线程池的状态是shutdown状态下,不会再接受新的任务,
        // 在此前提下如果状态已经到stop状态、或者传入任务不为空、或者任务队列为空
        // 都不需要添加新的任务
        if (rs >= SHUTDOWN &&
           !(rs == SHUTDOWN &&
             firstTask == null &&
             ! workQueue.isEmpty()))
            return false;
        
        for (;;) {
            // 获取工作线程总数
            int wc = workCountOf(c);
            // 如果工作线程总数大于等于容量或者大于等于核心线程数/最大线程数,
            // 则不需要添加新的任务
            if (wc >= CAPACITY ||
                wc >= (core ? corePoolSize : maximumPoolSize))
                return false;
            // 成功通过cas添加新的线程数wc,则break到最外层的循环
            if (compareAndOIncreamentWorkerCount(c))
                breack retry;
            // 走到这里说明通过cas添加新的线程数wc失败,这个时候需要重新判断线程池的状态
            // 是否由RUNNABLE已经变成SHUTDOWN
            c = ctl.get();
            // 如果线程池状态已经由RUNNING已经变为SHUTDOWN,则重新跳出到外层循环继续执行
            if (runStateOf(c) != rs)
                continue retry;
            // 如果线程池状态依然是RUNNING, CAS更新工作线程数wc失败说明有可能是并发更新导致的失败
            // 则在内层循环即可
        }
    }
    
    // 标记工作线程是否成功启动
    boolean workerStarted = false;
    // 标记工作线程是否创建成功
    boolean workerAdded = false;
    Worker w = null;
    try {
        // 传入任务实例firstTask创建worker实例,Worker构造里面会通过线程工厂创建新的Thread对象
        w = new Worker(firstTask);
        // 获取worker的线程
        final Thread t = w.thread;
        if (t != null) {
            // 获取全局锁
            final ReentrantLock mainLock = this.mainLock;
            // 全局锁加锁,因为会改变一些指标值和非线程安全的集合
            mainLock.lock();
            try {
                // 获取线程池状态
                int rs = runStateOf(ctl.get());
                
                // 如果线程池状态不是RUNNING或者是SHUTDOWN同时传入的任务实例firstTask为null,
                // 则判断线程是否存活,不存活抛异常
                if (rs < SHUTDOWN ||
                    (rs == SHUTDOWN && firstTask == null)) {
                    if (t.isAlive())
                        throw new IllegalThreadStateException();
                    // 把创建的工作线程实例添加到工作线程集合
                    workers.add(w);
                    // 获取工作线程数量
                    int s = workers.size();
                    //  尝试更新线程池峰值容量
                    if (s > largestPoolSize)
                        largestPoolSize = s;
                    // 标记工作线程添加成功,后面才会调用Thread#start()方法启动真实的线程实例
                    workerAdded = true;
                } 
            } finally {
                mainLock.unLock();
            }
            // 如果成功添加工作线程
            if (workerAdded) {
                // 调用Worker内部的线程实例t的Thread#start()方法启动真实的线程实例
                t.start();
                // 标记线程启动成功
                workerStarted = true;
            }
        }
    } finally {
         // 线程启动失败,则需要从工作线程集合移除对应的Worker
         if (! workerStarted)
          	addWorkerFailed(w);
    }
    return workerStarted;
}

// 添加Worker失败
private void addWorkerFailed(Worker w) {
    final ReentrantLock mainLock = this.mainLock();
    mainLock.lock();
    try {
        if (w != null)
            // 从工作线程移除
            workers.remove();
        // 工作线程数-1
        decrementWorkerCount();
        // 基于状态尝试终止线程池
        tryTerminate();
    } finally {
        mainLock.unLock();
    }
}
复制代码

addWorker()方法是用来添加执行任务,这个流程可以分为两部分来看,第一部分是用于记录线程数量,第二部分是在独占锁里创建执行线程并启动。流程如下:

  • 首先判断当前线程池的状态是否是SHUTDOWNSTOPTIDYINGTERMINATED中的一个。并且当前状态为SHUTDOWN、且传入的任务为NULL、同时任务队列不为空。那么就返回false
  • 不满足上一点然后判断线程数是否超过核心线程数或者最大线程数(根据传入的core判断),如果超过则返回false
  • 然后通过CAS操作增加线程池数量,成功跳出循环体。
  • 线程池数量记录成功之后,创建工作实例,使用独占锁创建工作线程并加入到工作线程集合,并记录添加状态,添加成功则启动工作线程,记录启动状态,如果最后启动失败则调用addWorkerFailed()方法移除线程等操作。

流程图如下:

Java线程池实现原理详解,看完一篇,学会线程池?_第4张图片

内部类Worker源码分析

线程池中的每一个具体的工作线程被包装为内部类Worker实例,Worker继承与AQS,实现了Runnable接口,源码如下:

private final class Worker extends AbstractQueuedSynchronized implements Runnable {
    
    private static final long serialVersionUID = 6138294804551838833L;
    
    // 保存ThreadFactory创建的线程实例,如果创建失败为null
    final Thread thread;
    
    // 保存传入的Runnable实例
    Runnable firstTask;
    
    // 记录线程完成的任务总数
    volatile long completedTasks;
    
    // 唯一构造方法,传入任务实例firstTask, 可以为null
    Worker(Runnable firstTask) {
        // 禁止线程中断,直到runWorker方法执行
        setState(-1);
        this.firstTask = firseTask;
        this.thread = getThreadFactory().newThread(this);
    }
    
    public void run() {
        // 调用外部的runWorker方法执行真正的任务
        runWorker(this);
    }
    
    // 是否持有独占锁,state = 0表示没有获取锁,state > 0表示获取锁
    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) {
            }
        }
    }
}
复制代码

Worker的构造函数里面的逻辑十分重要,通过ThreadFactory创建Thread实例同时引入Worker实例,因为Worker本身实现了Runnable,所以可以作为任务提交到线程中执行。只要Worker持有的线程实例w调用Thread#start()方法就能执行Worker#run()。简化一下逻辑如下:

// addWorker()方法中构造
Worker worker = createWorker();
// 通过线程池构造时候传入
ThreadFactory threadFactory = getThreadFactory();
// Worker构造函数中
Thread thread = threadFactory.newThread(worker);
// addWorker()方法中启动
thread.start();
复制代码

Worker继承AQS,这里使用了AQS的独占锁模式,这里有个技巧是构造Worker的时候,把AQS资源状态通过setState(-1)设置成-1,这是因为Wokrer实例刚创建时AQSstate的默认值是0,此时线程尚未启动,不能在这个时候进行线程中断,见Worker#interruptIfStarted()方法。

runWorker方法源码分析

final void runWorker(Worker w) {
    // 获取当前线程,实际上和Wokrer持有的线程实例是相同的
    Thread wt = Thread.currentThread();
    // 获取worker中持有的初始化时传入的任务对象,这里存放临时变量task
    Runnable task = w.firstTask;
    // 设置Worker中持有的初始化时传入的任务对象为null
    w.firstTask = null;
    // 构造方式的是state设置成-1,这里解锁state设成为0,允许线程中断
    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 {
                    // 执行任务
                    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 {
                // 任务对象设为null
                task = null;
                // 完成工作数量+1
                w.completedTasks++;
                // 解锁
                w.unlock();
            }
        }
        // 正常完成任务
        completedAbruptly = false;
    } finally {
        // 处理线程退出,completedAbruptly为true说明由于用户异常导致线程非正常退出
        processWorkerExit(w, completedAbruptly);
    }
}
复制代码

runWorker方法的核心流程如下:

  1. Worker先执行解锁操作,允许线程中断。
  2. 通过while循环调用getTask()方法获取任务对象,首轮循环可能是外部传入的收个任务对象。
  3. 如果线程池状态变为STOP状态,则需要确保工作线程是中断状态并且进行中断处理,否则要保证工作线程不是中断状态。
  4. 执行任务实例Runnable#run()方法,任务执行之前和之后分别会调用beforeExecute()afterExecute()
  5. while循环跳出说明任务全部执行完毕,然后会调用processWorkerExit()方法处理工作线程退出后的工作。

千言万语不如一图,流程图如下:

Java线程池实现原理详解,看完一篇,学会线程池?_第5张图片

getTask方法源码解析

getTask()方法是工作线程在while死循环中获取任务队列中的任务对象的方法,源码如下:

private Runnable getTask() {
    
    // 记录上一次从队列中获取的时候是否超时
    boolean timeOut = false;
    // 循环
    for(;;) {
        int c = ctl.get();
        // 获取线程池状态
        int rs = runStateOf(c);
        
        // 如果线程池状态至少为SHUTDOWN,如果线程池状态STOP或者任务队列为空
        // 则工作线程数量wc减1,直接返回null
        if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())){
            decrementWorkerCount();
            return null;
        }
        
        // 获取工作线程数
        int wc = workerCountOf(c);
        
        // timed临时变量用于线程超时控制,决定是否需要通过poll()的非阻塞方法从任务队列获取任务
        // allowCoreThreadTimeOut默认为false,如果设置成true,则允许核心线程也能通过poll()方法从任务队列中拉取任务
        // 工作线程数大于核心线程数的时候,说明线程池中创建了额外的非核心线程,这些非核心线程一定是通过poll()方法从任务队列中拉取任务
        boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;
        
        // 1.工作线程数大于最大线程数或者timed && timedOut 说明线程命中了超时控制并且上一轮循环通过poll()方法从任务队列获取任务为null
        // 并且工作线程总数大于1或者任务队列为空,则通过CAS把线程数减去1,同时返回null
        if ((wc > maximumPoolSize || (timed && timedOut))
            && (wc > 1 || workQueue.isEmpty())) {
            if (compareAndDecrementWorkerCount(c))
                return null;
            continue;
        }
        
        try {
            // 如果timed为true,通过poll()方法做超时拉取,keepAliveTime时间内没有等待到有效的任务,则返回null
            // 如果timed为false,通过take()做阻塞拉取,会阻塞到有下一个有效的任务时候再返回(一般不会是null)
            Runnable r = timed ? 
                workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
            	workQueue.take();
            if (r != null)
                return r;
            timedCount  = true;
        } catch (InterruptedException retry) {
                timedOut = false;
        }
    }
}
复制代码

这个方法中,有两处十分复杂的if逻辑,先来看第一处,对于第一处if可能导致工作线程数量减去1直接返回null的场景有:

  • 线程池状态为SHUTDOWN,一般是调用了shutdown()方法,并且任务队列为空。
  • 线程池状态为STOP

对于第二处if逻辑有点复杂,先拆解一下:

// 工作线程总数大于最大线程数
boolean b1 = wc > maximumPoolSize;
// 允许线程超时同时上一轮通过poll()方法从任务队列中获取任务为null
boolean b2 = timed && timedOut;
// 作线程总数大于1
boolean b3 = wc > 1;
// 任务队列为空
boolean b4 = workQueue.isEmpty();
if (r) {
    if (compareAndDecrementWorkerCount(c)){
        return null;
    }else{
        continue;
    }
}
复制代码

这段逻辑大多数情况下是针对非核心线程的。在execute()方法中,线程总数大于核心线程并且小于最大线程数时,会调用addWorker(task, false)方法添加非核心线程,而这里的逻辑恰好是想法的操作,用于减少非核心线程数,使得工作县城总数总是接近于核心线程数。如果对于核心线程,上一轮循环获取对象为null,这一轮循环很容易满足timed && timedOuttrue,这个时候getTask()返回null导致runWorker()方法跳出循环,最后执行processWorkerExit()方法处理工作,而该非核心线程对应的Worker则变成“游离对象”,等待被JVM回收。当allowCoreThreadTimeOut设置为true的时候,这里分析的非核心线程的生命周期终结逻辑同时会适用于核心线程,那么可以总结出keepAliveTime的意义:

  • 当允许核心线程超时,也就是allowCoreThreadTimeOut设置为true的时候,此时keepAliveTime表示空闲的工作线程存活周期。
  • 默认情况下不允许核心线程超时,此时keepAliveTime表示空闲的非核心线程存活周期。

三、手写一个线程池

通过上面对ThreadPoolExecutor的学习,我们可以手写一个简单的线程池,包含了线程的核心逻辑,包含了提交任务,添加任务,获取任务,执行任务核心逻辑。

这个手写线程池的逻辑也非常简单,只体现核心流程,包括:

  1. 有n个一直执行的线程。
  2. 把线程提交给线程池运行。
  3. 如果线程池已满,则把线程放入队列中。
  4. 最后当有空闲时,则获取队列中线程进行运行。

代码实现:

public class ThreadPoolTrader implements Executor {

    private final AtomicInteger ctl = new AtomicInteger(0);

    private volatile int corePoolSize;
    private volatile int maximumPoolSize;
    private final BlockingQueue workQueue;

    public ThreadPoolTrader(int corePoolSize, int maximumPoolSize,
                            BlockingQueue workQueue) {
        this.corePoolSize = corePoolSize;
        this.maximumPoolSize = maximumPoolSize;
        this.workQueue = workQueue;
    }

    @Override
    public void execute(Runnable command) {
        int c = ctl.get();
        if (c < corePoolSize) {
            if (!addWorker(command)) {
                reject();
            }
            return;
        }
        if (!workQueue.offer(command)) {
            if (!addWorker(command)) {
                reject();
            }
        }
    }

    private boolean addWorker(Runnable firstTask) {
        if (ctl.get() >= maximumPoolSize) return false;
        Worker worker = new Worker(firstTask);
        worker.thread.start();
        ctl.incrementAndGet();
        return true;

    }

    private final class Worker implements Runnable {

        final Thread thread;
        Runnable firstTask;

        Worker(Runnable firstTask) {
            this.thread = new Thread(this);
            this.firstTask = firstTask;
        }

        @Override
        public void run() {
            Runnable task = firstTask;
            try {
                while (task != null || (task = getTask()) != null) {
                    System.out.println("当前执行任务的线程:" + Thread.currentThread().getName());
                    task.run();
                    if (ctl.get() > maximumPoolSize) {
                        break;
                    }
                    task = null;
                }
            } finally {
                ctl.decrementAndGet();
            }
        }
    }

    private Runnable getTask() {
        for (; ; ) {
            try {
                System.out.println("workQueue.size:" + workQueue.size());
                return workQueue.take();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    private void reject() {
        throw new RuntimeException("Error!ctl.count:" + ctl.get() + " workQueue.size:" + workQueue.size());
    }

    public static void main(String[] args) {
        ThreadPoolTrader threadPoolTrader = new ThreadPoolTrader(2, 2,
                new ArrayBlockingQueue(10));
        for (int i = 0; i < 10; i++) {
            int finalI = i;
            threadPoolTrader.execute(() ->{
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("任务编号:" + finalI);
            });
        }
    }
}
复制代码

上面的代码测试如下:

当前执行任务的线程:Thread-1
当前执行任务的线程:Thread-0
任务编号:1
任务编号:0
workQueue.size:8
workQueue.size:8
当前执行任务的线程:Thread-0
当前执行任务的线程:Thread-1
任务编号:3
任务编号:2
workQueue.size:6
当前执行任务的线程:Thread-1
workQueue.size:6
当前执行任务的线程:Thread-0
任务编号:5
workQueue.size:4
当前执行任务的线程:Thread-0
任务编号:4
workQueue.size:3
当前执行任务的线程:Thread-1
任务编号:6
workQueue.size:2
当前执行任务的线程:Thread-0
任务编号:7
workQueue.size:1
当前执行任务的线程:Thread-1
任务编号:8
任务编号:9
workQueue.size:0
workQueue.size:0

你可能感兴趣的:(spring,java,后端,mysql,spring,boot)