Java中的ThreadPoolExecutor线程池

今天简单来和大家一起学习一下,java中的ThreadPoolExecutor线程池。

线程池简介

背书中,线程池是一个并发框架,在初始化一个多线程应用程序过程中创建一个线程集合,然后在需要执行新的任务时重用这些线程而不是新建一个线程。

合理使用线程池有以下几个好处

1.线程池可以利用已经创建好的线程重复执行任务,避免了线程创建、销毁带来的系统资源上的开支。所以线程池减少了系统资源消耗
2.有任务时,线程池可以利用空闲的线程去执行任务,提高了系统的响应速率
3.使用线程可以提高系统的并发处理能力,但是不合理的创建、使用线程会严重损耗系统的资源,线程的上下文环境切换也会带来系统资源上的开支,使用线程池可以对线程进行统一的分配、调优、监控

线程池的实现原理

前面我已经在《并行执行任务的Fork/Join框架》一文中,给大家介绍了ForkJoinPool线程池,今天就以ThreadPoolExecutor为我们的猪脚吧,先举个栗子,然后在根据ThreadPoolExecutor使用,和大家一起分析源码。

举个栗子

/**
 * @Description: .
 * @Author: ZhaoWeiNan .
 * @CreatedTime: 2017/8/20 .
 * @Version: 1.0 .
 */
public class Demo {

    /**
     * 初始化线程池
     */
    private ThreadPoolExecutor threadPool = new ThreadPoolExecutor(2, 10, 20, TimeUnit.SECONDS, new ArrayBlockingQueue(10),
            new ThreadPoolExecutor.DiscardOldestPolicy());

    public static void main(String[] args) {
        Demo demo = new Demo();
        for (int i = 0; i < 20; i++) {
            final int time = i;
            //把线程添加到线程池中
            demo.threadPool.execute(new Runnable() {
                @Override
                public void run() {
                    System.out.println("任务索引是:" + time);
                    try {
                        Thread.sleep(2000);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            });
        }
    }
}
1.构造函数

通过构造函数来看看ThreadPoolExecutor的成员变量。

/**
     * 创建新线程的工厂。
     *
     */
    private volatile ThreadFactory threadFactory;

    /**
     * 当线程池中线程数量或者线程池停止时被调用的处理类.
     */
    private volatile RejectedExecutionHandler handler;

    /**
     * 空闲线程等待工作的超时时间
     */
    private volatile long keepAliveTime;

    /**
     * 默认为false: 核心线程即使在空闲的时候也保持存活
     *       true: 核心线程使用keepAliveTime去等待工作
     */
    private volatile boolean allowCoreThreadTimeOut;

    /**
     * 核心线程池的大小,保持存活的工作线程的最小数量。
     */
    private volatile int corePoolSize;

    /**
     * 线程池的最大容量
     */
    private volatile int maximumPoolSize;

    /**
     * 线程池中的任务队列,是用了阻塞队列来实现(有机会为大家介绍)
     */
    private final BlockingQueue<Runnable> workQueue;

    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.corePoolSize = corePoolSize;
        this.maximumPoolSize = maximumPoolSize;
        this.workQueue = workQueue;
        this.keepAliveTime = unit.toNanos(keepAliveTime);
        this.threadFactory = threadFactory;
        this.handler = handler;
    }
2.看看把线程加入线程池的execute方法
 public void execute(Runnable command) {
        if (command == null)
            throw new NullPointerException();

        /**
         * 按3个步骤走:
         */
        int c = ctl.get();
        /**
         * 1.如果少于核心线程池大小corePoolSize的线程处于RUNNING状态,
         * 开一个新的线程,把command对象作为该线程的第一个任务。
         * 调用addWorker方法,原子性操作判断command的状态和worker的数量
         * 防止在线程池不能添加线程的时候添加线程。
         */
        if (workerCountOf(c) < corePoolSize) {
            if (addWorker(command, true))
                return;
            c = ctl.get();
        }
        /**
         * 2.如果一个任务成功加入到阻塞队列中,添加一个线程,还要进行第二次检查。
         * 因为一已经存在的线程可以能已经死亡,或者这个线程池已经关闭当要添加线程时。
         * 所以我们进行了第二次检查状态,进行必要的入队回滚,或者池中没有线程时去
         * 创建一个新的线程。
         */
        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);
        }
        /**
         * 3.如果任务不能入队,我们尝试创建一个新的线程,如果创建线程失败可,
         * 我们知道了,线程池现在是关闭状态或者是饱和状态,所以我们拒绝了这个任务。
         */
        else if (!addWorker(command, false))
            reject(command);
    }
3.看看addWorker方法
private boolean addWorker(Runnable firstTask, boolean core) {
        retry:
        for (;;) {
            //获取状态
            int c = ctl.get();
            int rs = runStateOf(c);

            // Check if queue empty only if necessary.
            // 检查状态,检查队列和任务是否为空
            if (rs >= SHUTDOWN &&
                    ! (rs == SHUTDOWN &&
                            firstTask == null &&
                            ! workQueue.isEmpty()))
                return false;

            for (;;) {
                //获取worker的数量
                int wc = workerCountOf(c);
                //判断worker的数量
                if (wc >= CAPACITY ||
                        wc >= (core ? corePoolSize : maximumPoolSize))
                    return false;
                if (compareAndIncrementWorkerCount(c))
                    break retry;
                //重新获取一次状态
                c = ctl.get();
                if (runStateOf(c) != rs)
                    continue retry;
                //由于worker数量改变导致CAS(原子操作)失败,重新进入内部循环
            }
        }

        boolean workerStarted = false;
        boolean workerAdded = false;
        //顶一个一个worker对象
        Worker w = null;
        try {
            //创建一个可重入锁
            final ReentrantLock mainLock = this.mainLock;
            //利用传入的任务对象,初始化worker对象
            w = new Worker(firstTask);
            //创建一个对象
            final Thread t = w.thread;
            if (t != null) {
                //加锁
                mainLock.lock();
                try {
                    // 在获取到锁的情况下,再次检查状态
                    int c = ctl.get();
                    int rs = runStateOf(c);

                    if (rs < SHUTDOWN ||
                            (rs == SHUTDOWN && firstTask == null)) {
                        //先检查线程t是否已经start
                        if (t.isAlive())
                            throw new IllegalThreadStateException();
                        //加入到workers中
                        //看一眼 workers
                        // private final HashSet<Worker> workers = new HashSet<Worker>();
                        // workers保存了所有已经获取到了mainLock锁的worker对象
                        workers.add(w);
                        int s = workers.size();
                        if (s > largestPoolSize)
                            largestPoolSize = s;
                        workerAdded = true;
                    }
                } finally {
                    mainLock.unlock();
                }
                if (workerAdded) {
                    t.start();
                    workerStarted = true;
                }
            }
        } finally {
            if (! workerStarted)
                addWorkerFailed(w);
        }
        return workerStarted;
    }
4.看看Worker这个类
worker

Worker是ThreadPoolExecutor中的一个内部类,实现了Runnable接口,是一个线程类,主要使用来,执行加入到线程池中的任务

线程池的监控

线程池有一些属性,可以支持我们对线程池进行了一些监控,在出现问题的时候可以很方便的进行问题的定位:

    /**
     * 线程池里曾经创建过的最大线程数,如果该数等于线程池的大小,证明该线程池曾经被打满过
     */
    private int largestPoolSize;

    /**
     * 线程中已经完成的任务数量
     */
    private long completedTaskCount;

    /**
     * 获取线程池中活动的线程数
     */
    public int getActiveCount() {
        final ReentrantLock mainLock = this.mainLock;
        mainLock.lock();
        try {
            int n = 0;
            for (ThreadPoolExecutor.Worker w : workers)
                if (w.isLocked())
                    ++n;
            return n;
        } finally {
            mainLock.unlock();
        }
    }

    /**
     * 线程池的线程数量
     */
    public int getPoolSize() {
        final ReentrantLock mainLock = this.mainLock;
        mainLock.lock();
        try {
            // Remove rare and surprising possibility of
            // isTerminated() && getPoolSize() > 0
            return runStateAtLeast(ctl.get(), TIDYING) ? 0
                    : workers.size();
        } finally {
            mainLock.unlock();
        }
    }

另外,线程池中,beforeExecute、afterExecute、terminated方法,都是空方法,可以重写这三个方法,在执行任务前、执行任务后、线程池关闭时执行代码来进行线程池的监控,例如可以统计任务执行时间等

    protected void beforeExecute(Thread t, Runnable r) { }

    protected void afterExecute(Runnable r, Throwable t) { }

    protected void terminated() { }

合理配置线程池

线程池的怎么配置,可能是一直困扰着我和大家的一个问题,具体合理配置我一直没有实践的做过实验,都是使用了度娘到的大神给的配置建议

CPU密集型任务

CPU密集型任务尽量配置少的线程数,一般为CPU个数 + 1

IO密集型任务

IO密集型任务线程并不是一直在执行任务,线程数可以设置多一点,一般为CPU个数的两倍

最后套用阿里巴巴Java开发手册中的一句话

阿里开发规约

ThreadPoolExecutor就为大家简单的说到这,欢迎大家来交流,指出文中一些说错的地方,让我加深认识,愿大家没有bug,谢谢!

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,189评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,577评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,857评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,703评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,705评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,620评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,995评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,656评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,898评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,639评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,720评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,395评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,982评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,953评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,195评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,907评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,472评论 2 342

推荐阅读更多精彩内容