ThreadPool原理分析

ThreadPool在实际应用中提升响应与执行效率,避免线程在创建,销毁占用jvm的性能。接下来深入探索ThreadPool的底层原理。
jdk提供两个类创建线程池Executors和原生ThreadPoolExecutor,实际Executors在创建线程池调用的还是ThreadPoolExecutor类中的构造器,由于Executors在使用消息入队的队列大小是Integer.MAX_VALUE方式,如果任务多的话极易导致内存泄漏,在实际开发中禁止使用Executors创建线程池,使用ThreadPoolExecutor手动配置合理的参数创建线程池。

ThreadPoolExecutor构造函数的参数说明:

  • int corePoolSize,
    线程核心数大小,线程池中无任务时候存活的线程个数
  • int maximumPoolSize,
    非核心线程数与核心线程数总和,线程池中最多可存活的线程个数
  • long keepAliveTime,
    无任务时候非核心线程数存活的时间
  • TimeUnit unit,
    无任务时候非核心线程存活时间单位
  • BlockingQueue<Runnable> workQueue,
    阻塞队列,当核心线程都在工作,又有任务过来先入队列,用于存放执行任务的队列
  • ThreadFactory threadFactory,
    创建线程的工厂默认使用Executors.DefaultThreadFactory 用于存放
  • RejectedExecutionHandler handler
    拒绝策略
    1. AbortPolicy:当队列满了,线程池中线程数量已到最大线程数,当有任务提交直接拒绝,并抛出异常
      2.CallerRunsPolicy:当队列满了,线程池中线程数量已到最大线程数,当有任务提交由调用线程执行
      3.DiscardOldestPolicy:当队列满了,线程池中线程数量已到最大线程数,当有任务提交直接丢弃最老的任务,不抛异常
      4.DiscardPolicy:当队列满了,线程池中线程数量已到最大线程数,当有任务提交直接丢弃不抛异常

ThreadPoolExecutor几个重要的属性

1.ctl正在运行的线程数量
2.Worker存放线程对象

启动线程执行任务

有两种提交任务submit和excute,不同在于submit提交会有返回值,excute提交没有返回值(以后在写文章介绍这两种不同点)
execute源码

public void execute(Runnable command) {
    if (command == null)
        throw new NullPointerException();
    //c初始化是-536870912
    int c = ctl.get();
     //获取正在工作的线程数量
    //有任务提交判断正在工作的线程数量是否小于核心线程数
    //  如果小于将线程添加到工作队列中启动线程执行任务
    if (workerCountOf(c) < corePoolSize) {
        if (addWorker(command, true))
            return;
        c = ctl.get();
    }
     //isRunning(c)为true,
    //如果线程池处于RUNNING状态,则添加任务到阻塞队列
    if (isRunning(c) && workQueue.offer(command)) {
        int recheck = ctl.get();
        //再做一次检查
       //如果当前线程池状态不是RUNNING则从队列删除任务,并执行拒绝策略
        if (! isRunning(recheck) && remove(command))
            reject(command);
        //如果当前线程池线程空,则添加一个线程
        else if (workerCountOf(recheck) == 0)
            addWorker(null, false);
    }
    //添加到任务队列失败,尝试创建非核心线程如果
    //非核心线程已满直接拒绝
    else if (!addWorker(command, false))
        reject(command);
}

1.判断工作线程是否小于核心线程
如果小于直接创建线程添加到工作对列中(Worker)
2.如果不小于核心线程数
判断线程池是否运行状态,如果是将任务添加到阻塞队列中
如果入队成功做二次检查,线程池非运行状态将任务删除使用拒绝策略
如果当前线程池为空添加一个线程
3如果入队失败尝试创建非核心线程数,创建非核心线程数失败使用拒绝策略

addWorker(Runable,Boolean)方法

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 (;;) {
                int wc = workerCountOf(c);
                //判断工作线程是否大于最大限制
                //或者是否大于=核心线程or最大线程数 如果是返回false
                if (wc >= CAPACITY ||
                    wc >= (core ? corePoolSize : maximumPoolSize))
                    return false;
                //cas将工作线程个数加1,如果失败重新循环校验,成功将工作线程加1
                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 {
            //创建一个Worker将任务和线程赋值给该Worker
            w = new Worker(firstTask);
            final Thread t = w.thread;
            if (t != null) {
                //加锁将worker添加到HashSet中
                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();
                        //将创建好的线程和任务的worker添加到hashset中
                        workers.add(w);
                        int s = workers.size();
                        if (s > largestPoolSize)
                            largestPoolSize = s;
                        workerAdded = true;
                    }
                } finally {
                    mainLock.unlock();
                }
                //如果创健worker成功启动当前线程执行任务
                if (workerAdded) {
                    t.start();
                    workerStarted = true;
                }
            }
        } finally {
            if (! workerStarted)
                addWorkerFailed(w);
        }
        return workerStarted;
    }

addWorker方法主要作用就check检查和创建worker,启动线程执行任务,worker类实现了Runable接口重写了run方法,runWorker方法才是真正处理核心任务,如何线程复用以及当非核心线程空闲时关闭非核心线程。

着重分析runWorker方法

final void runWorker(Worker w) {
        //获取当前线程
        Thread wt = Thread.currentThread();
        //获取当前任务
        Runnable task = w.firstTask;
        //将worker的任务置空,下次取任务从队列取
        w.firstTask = null;
        w.unlock(); // allow interrupts
        boolean completedAbruptly = true;
        try {
             //任务不为空 getTask()也是获取任务获取的是队列中的任务,稍后我们详细讲解getTask()
            while (task != null || (task = getTask()) != null) {
                w.lock();
                // If pool is stopping, ensure thread is interrupted;
                // if not, ensure thread is not interrupted.  This
                // requires a recheck in second case to deal with
                // shutdownNow race while clearing interrupt
                //校验线程池是否停止,停止就中断线程
               //如果线程池没有停止,确保线程没用中断
                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 {
            //将worker从hashset中删除
            processWorkerExit(w, completedAbruptly);
        }
    }

getTask()方法从队列取数据,如果任务没了返回空,核心线程数阻塞在workQueue.take()方法处

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.
             //检查队列是否为空,线程池是否停止 如果停止或者为空将工作线程进行-1
            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())) {
                if (compareAndDecrementWorkerCount(c))
                    return null;
                continue;
            }
            try {
                //如果线程池工作线程数大于corPoolSize使用poll方式后获取任务返回null
                //否则线程池数量<=corPoolSize数量进行take()方式获取任务
                //如果队列为空则当前线程阻塞在这里
                Runnable r = timed ?
                    workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                    workQueue.take();
                if (r != null)
                    return r;
                timedOut = true;
            } catch (InterruptedException retry) {
                timedOut = false;
            }
        }
    }

ThreadPoolExecutor源码已经分析完毕,进行一个小的总结

总结:
1.任务执行时首先判断当前线程池线程数量是否小于核心线程数,
       小于的核心线程就添加到工作worker中将worker添加到HashSet<worker>中,并启动线程执行worker中的任务,执行完之后处于while循环取队列的任务执行。
       大于核心线程将任务添加到阻塞队列中,让已启动的核心线程去队列中取任务执行,但是如果队列满了又来新的任务,会创建非核心线程数来执行任务,当核心线程数和非核心线程数总和等于线程池最大线程数进行一个执行拒绝策略。
2.当没有任务提交时,线程池中线程执行队列中的任务,在取任务时候会判断当前线程池中的线程数量是否大于核心线程数如果大于那么线程取队列的任务使用poll+时间参数,如果队列为空了线程取任务超时取不到数据会返回null,接下来执行processWorkerExit移除worker工作对象,线程池保留的是核心线程数,而这些线程阻塞在取队列任务的take()方法处,等待线程将任务offer进队列时候唤醒进行取任务。

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