ThreadPoolExecutor源码剖析

ThreadPoolExecutor提交任务入口代码如下:

    public void execute(Runnable command) {
        if (command == null)
            throw new NullPointerException();
        // ctl高3位表示线程池状态,低29位表示线程数目,对ctl的访问需要进行位运算
        int c = ctl.get(); 
        // 如果worker线程数目小于corePoolSize,增加一个worker线程
        if (workerCountOf(c) < corePoolSize) { 
            if (addWorker(command, true))
                return;
            c = ctl.get();
        }
        // 如果worker线程不小于corePoolSize,并且线程池正在运行,则把任务添加到workQueue中
        // workQueue是一个BlockingQueue<Runnable>
        if (isRunning(c) && workQueue.offer(command)) {
            int recheck = ctl.get();
            // 再次获取线程池状态码并检测线程池是否运行,如果没有运行,则移除刚才提交的任务,调用reject方法
            // reject方法可以由RejectedExecutionHandler指定
            if (! isRunning(recheck) && remove(command))
                reject(command);
            // 如果worker线程数目为0,以maximumPoolSize为限制增加一个worker线程
            else if (workerCountOf(recheck) == 0)
                addWorker(null, false);
        }
        // 如果任务添加到workQueue(有可能是有界队列)失败,
        // 以maximumPoolSize为限制增加一个worker线程
        // 如果增加线程失败,调用reject方法
        else if (!addWorker(command, false))
            reject(command);
    }

看完任务如何提交,继续看addWorker方法是怎么运行的,只留下了方法的主干:

    private boolean addWorker(Runnable firstTask, boolean core) {
        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线程数目大于CAPACITY(ctl的低29位全为1,500多万)
                // 或者worker线程大于corePoolSize,maximumPoolSize(用core开关控制)
                // 添加worker线程失败
                if (wc >= CAPACITY ||
                    wc >= (core ? corePoolSize : maximumPoolSize))
                    return false;
                // CAS操作,增加worker线程数目
                if (compareAndIncrementWorkerCount(c))
                    break retry;
                c = ctl.get();  // Re-read ctl
                if (runStateOf(c) != rs)
                    continue retry;
            }
        }
        // 增加worker线程测试条件通过,真正开始增加worker线程
        boolean workerStarted = false;
        boolean workerAdded = false;
        Worker w = null;
        try {
            // new一个Worker,Worker是对线程和任务的一个封装,下面会讲到
            w = new Worker(firstTask);
            final Thread t = w.thread;
            if (t != null) {
                final ReentrantLock mainLock = this.mainLock;
                mainLock.lock();
                try {
                        // 在workers里面新增worker
                        workers.add(w);
                        int s = workers.size();
                        // 更新largestPoolSize
                        if (s > largestPoolSize)
                            largestPoolSize = s;
                        workerAdded = true;
                    }
                } finally {
                    mainLock.unlock();
                }
                // 如果worker添加成功,则启动这个worker线程
                if (workerAdded) {
                    t.start();
                    workerStarted = true;
                }
            }
        } finally {
            // 如果启动worker失败,调用addWorkerFailed方法
            // 这个方法主要是减少worker数目,从workers里面移除刚才添加的worker
            // 并在线程池中尝试中断一个idle worker
            if (! workerStarted)
                addWorkerFailed(w);
        }
        return workerStarted;
    }

下面接着看Worker是怎么对线程与任务封装的,下面是Worker类的代码:

    // Worker类本身也是一个Runnable
    // AQS可以这样理解,它内部持有一个状态,并发的线程可以原子性的去修改这个状态
    private final class Worker
        extends AbstractQueuedSynchronizer
        implements Runnable
    {
        // 运行任务的线程
        final Thread thread;
        // 初始化运行的任务,可以为null
        Runnable firstTask;
        // 完成任务的计数
        volatile long completedTasks;

        Worker(Runnable firstTask) {
            setState(-1); // 状态设置为-1,禁止没有start线程前去中断这个线程
            this.firstTask = firstTask;
            // 传的是this对象,所以线程start会调用this的run方法
            this.thread = getThreadFactory().newThread(this);
        }
        
        // run方法委托给runWorker方法,参数传的是this
        public void run() {
            runWorker(this);
        }

        // 关于锁的方法
        // 通过lock与unlock,可以知道worker是不是idle worker
        // 0代表没有加锁状态
        // 1代表加锁状态

        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) {
                }
            }
        }
    }

下面来看看runWorker方法,只留下了方法的主干:

  final void runWorker(Worker w) {
    Thread wt = Thread.currentThread(); //获取到worker的thread
    Runnable task = w.firstTask;
    w.firstTask = null;
    w.unlock(); // unlock worker,允许中断thread
    boolean completedAbruptly = true;
    // 死循环,当task为null或者从任务队列获取任务为null时,worker的thread会退出
    while (task != null || (task = getTask()) != null) {
      w.lock(); // lock表示这个worker不是idle worker
      // 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 {
        Throwable thrown = null;
        try {
          // 运行任务
          task.run();
        } catch (Throwable x) {
          thrown = x;
          throw new Error(x);
        }
      } finally {
        task = null;
        w.completedTasks++;
        w.unlock();
      }
    }
  }

下面接着看worker线程是如何从任务队列获取任务的:

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

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

            // 如果线程池状态至少为shutdown,并且
            // 线程池状态至少为stop或者工作队列为空
            // 减少worker的数目,并返回null,当worker线程检测到任务为null时,会自己退出
            if (rs >= SHUTDOWN && (rs >= STOP || workQueue.isEmpty())) {
                decrementWorkerCount();
                return null;
            }

            int wc = workerCountOf(c);
            // 当allowCoreThreadTimeOut为true或者worker线程数目大于corePoolSize时,允许超时
            boolean timed = allowCoreThreadTimeOut || wc > corePoolSize;
            // 如果worker线程数目大于maximumPoolSize,或者
            // 允许超时并且上一次poll也超时,并且
            // worker线程数目大于1,或者工作队列为空
            // 减少worker线程数目返回null
            if ((wc > maximumPoolSize || (timed && timedOut))
                && (wc > 1 || workQueue.isEmpty())) {
                if (compareAndDecrementWorkerCount(c))
                    return null;
                continue;
            }

            try {
                // 允许超时则用poll,否则take
                Runnable r = timed ?
                    workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) :
                    workQueue.take();
                if (r != null)
                    return r;
                timedOut = true;
            } catch (InterruptedException retry) {
                timedOut = false;
            }
        }
    }

以上就是整个ThreadPoolExecutor主要的方法,还有另一个比较重要的方法,是用来清除idle worker的:

    private void interruptIdleWorkers(boolean onlyOne) {
        final ReentrantLock mainLock = this.mainLock;
        mainLock.lock();
        try {
            for (Worker w : workers) {
                Thread t = w.thread;
                // 如果worker的thread没有中断,并且尝试去获取worker的锁
                // 成功则表示worker没有运行任务,那么中断这个idle worker
                // 失败则表示worker正在运行任务,不中断这个worker
                if (!t.isInterrupted() && w.tryLock()) {
                    try {
                        t.interrupt();
                    } catch (SecurityException ignore) {
                    } finally {
                        w.unlock();
                    }
                }
                if (onlyOne)
                    break;
            }
        } finally {
            mainLock.unlock();
        }
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,519评论 5 468
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,842评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,544评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,742评论 1 271
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,646评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,027评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,513评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,169评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,324评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,268评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,299评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,996评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,591评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,667评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,911评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,288评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,871评论 2 341