Java线程池框架核心代码分析
<h2>前言</h2> <p>多线程编程中,为每个任务分配一个线程是不现实的,线程创建的开销和资源消耗都是很高的。线程池应运而生,成为我们管理线程的利器。Java 通过 Executor 接口,提供了一种标准的方法将任务的提交过程和执行过程解耦开来,并用 Runnable 表示任务。</p> <p>下面,我们来分析一下 Java 线程池框架的实现 ThreadPoolExecutor 。</p> <p>下面的分析基于JDK1.7</p> <h2>生命周期</h2> <p>ThreadPoolExecutor 中,使用 CAPACITY 的高3位来表示运行状态,分别是:</p> <ol> <li>RUNNING:接收新任务,并且处理任务队列中的任务</li> <li>SHUTDOWN:不接收新任务,但是处理任务队列的任务</li> <li>STOP:不接收新任务,不出来任务队列,同时中断所有进行中的任务</li> <li>TIDYING:所有任务已经被终止,工作线程数量为 0,到达该状态会执行 terminated()</li> <li>TERMINATED: terminated() 执行完毕</li> </ol> <p style="text-align:center"><img src="https://simg.open-open.com/show/521301bc9a892511d507abe3551c45a8.png"></p> <p>状态转换图</p> <p>ThreadPoolExecutor 中用原子类来表示状态位</p> <pre> <code class="language-java">private final AtomicInteger ctl = new AtomicInteger(ctlOf(RUNNING, 0));</code></pre> <h2>线程池模型</h2> <h3>核心参数</h3> <ul> <li>corePoolSize :最小存活的工作线程数量(如果设置 allowCoreThreadTimeOut ,那么该值为 0)</li> <li>maximumPoolSize :最大的线程数量,受限于 CAPACITY</li> <li>keepAliveTime :对应线程的存活时间,时间单位由TimeUnit指定</li> <li>workQueue :工作队列,存储待执行的任务</li> <li>RejectExecutionHandler :拒绝策略,线程池满后会触发</li> </ul> <p>线程池的最大容量: CAPACITY 中的前三位用作标志位,也就是说工作线程的最大容量为 (2^29)-1</p> <h3>四种模型</h3> <ul> <li>CachedThreadPool :一个可缓存的线程池,如果线程池的当前规模超过了处理需求时,那么将回收空闲的线程,当需求增加时,则可以添加新的线程,线程池的规模不存在任何的限制。</li> <li>FixedThreadPool :一个固定大小的线程池,提交一个任务时就创建一个线程,直到达到线程池的最大数量,这时线程池的大小将不再变化。</li> <li>SingleThreadPool :一个单线程的线程池,它只有一个工作线程来执行任务,可以确保按照任务在队列中的顺序来串行执行,如果这个线程异常结束将创建一个新的线程来执行任务。</li> <li>ScheduledThreadPool :一个固定大小的线程池,并且以延迟或者定时的方式来执行任务,类似于Timer。</li> </ul> <h2>执行任务 execute</h2> <p>核心逻辑:</p> <ol> <li>当前线程数量 < corePoolSize ,直接开启新的核心线程执行任务 addWorker(command, true)</li> <li>当前线程数量 >= corePoolSize ,且任务加入工作队列成功 <ol> <li>检查线程池当前状态是否处于 RUNNING</li> <li>如果否,则拒绝该任务</li> <li>如果是,判断当前线程数量是否为 0,如果为 0,就增加一个工作线程。</li> </ol> </li> <li>开启普通线程执行任务 addWorker(command, false) ,开启失败就拒绝该任务</li> </ol> <p>从上面的分析可以总结出线程池运行的四个阶段:</p> <ol> <li>poolSize < corePoolSize 且队列为空,此时会新建线程来处理提交的任务</li> <li>poolSize == corePoolSize ,此时提交的任务进入工作队列,工作线程从队列中获取任务执行,此时队列不为空且未满。</li> <li>poolSize == corePoolSize ,并且队列已满,此时也会新建线程来处理提交的任务,但是 poolSize < maxPoolSize</li> <li>poolSize == maxPoolSize ,并且队列已满,此时会触发拒绝策略</li> </ol> <h2>拒绝策略</h2> <p>前面我们提到任务无法执行会被拒绝, RejectedExecutionHandler 是处理被拒绝任务的接口。下面是四种拒绝策略。</p> <ul> <li>AbortPolicy :默认策略,终止任务,抛出RejectedException</li> <li>CallerRunsPolicy :在调用者线程执行当前任务,不抛异常</li> <li>DiscardPolicy : 抛弃策略,直接丢弃任务,不抛异常</li> <li>DiscardOldersPolicy :抛弃最老的任务,执行当前任务,不抛异常</li> </ul> <h2>线程池中的 Worker</h2> <p>Worker 继承了 AbstractQueuedSynchronizer 和 Runnable ,前者给 Worker 提供锁的功能,后者执行工作线程的主要方法 runWorker(Worker w) (从任务队列捞任务执行)。Worker 引用存在 workers 集合里面,用 mainLock 守护。</p> <pre> <code class="language-java">private final ReentrantLock mainLock = new ReentrantLock(); private final HashSet<Worker> workers = new HashSet<Worker>();</code></pre> <h3>核心函数 runWorker</h3> <p>下面是简化的逻辑,注意:每个工作线程的 run 都执行下面的函数</p> <pre> <code class="language-java">final void runWorker(Worker w) { Thread wt = Thread.currentThread(); Runnable task = w.firstTask; w.firstTask = null; while (task != null || (task = getTask()) != null) { w.lock(); beforeExecute(wt, task); task.run(); afterExecute(task, thrown); w.unlock(); } processWorkerExit(w, completedAbruptly); }</code></pre> <ol> <li>从 getTask() 中获取任务</li> <li>锁住 worker</li> <li>执行 beforeExecute(wt, task) ,这是 ThreadPoolExecutor 提供给子类的扩展方法</li> <li>运行任务,如果该worker有配置了首次任务,则先执行首次任务且只执行一次。</li> <li>执行 afterExecute(task, thrown);</li> <li>解锁 worker</li> <li>如果获取到的任务为 null,关闭 worker</li> </ol> <h2>获取任务 getTask</h2> <p>线程池内部的任务队列是一个阻塞队列,具体实现在构造时传入。</p> <pre> <code class="language-java">private final BlockingQueue<Runnable> workQueue;</code></pre> <p>getTask() 从任务队列中获取任务,支持阻塞和超时等待任务,四种情况会导致返回 null ,让 worker 关闭。</p> <ol> <li>现有的线程数量超过最大线程数量</li> <li>线程池处于 STOP 状态</li> <li>线程池处于 SHUTDOWN 状态且工作队列为空</li> <li>线程等待任务超时,且线程数量超过保留线程数量</li> </ol> <p>核心逻辑:根据 timed 在阻塞队列上超时等待或者阻塞等待任务,等待任务超时会导致工作线程被关闭。</p> <pre> <code class="language-java">timed = allowCoreThreadTimeOut || wc > corePoolSize; Runnable r = timed ? workQueue.poll(keepAliveTime, TimeUnit.NANOSECONDS) : workQueue.take();</code></pre> <p>在以下两种情况下等待任务会超时:</p> <ol> <li>允许核心线程等待超时,即 allowCoreThreadTimeOut(true)</li> <li>当前线程是普通线程,此时 wc > corePoolSize</li> </ol> <p>工作队列使用的是 BlockingQueue ,这里就不展开了,后面再写一篇详细的分析。</p> <h2>总结</h2> <ul> <li>ThreadPoolExecutor 基于生产者-消费者模式,提交任务的操作相当于生产者,执行任务的线程相当于消费者。</li> <li>Executors 提供了四种基于 ThreadPoolExecutor 构造线程池模型的方法,除此之外,我们还可以直接继承 ThreadPoolExecutor ,重写 beforeExecute 和 afterExecute 方法来定制线程池任务执行过程。</li> <li>使用有界队列还是无界队列需要根据具体情况考虑,工作队列的大小和线程的数量也是需要好好考虑的。</li> <li>拒绝策略推荐使用 CallerRunsPolicy ,该策略不会抛弃任务,也不会抛出异常,而是将任务回退到调用者线程中执行。</li> </ul> <p> </p> <p>来自:http://developer.51cto.com/art/201702/530079.htm</p> <p> </p>
本文由用户 qzbr7800 自行上传分享,仅供网友学习交流。所有权归原作者,若您的权利被侵害,请联系管理员。
转载本站原创文章,请注明出处,并保留原始链接、图片水印。
本站是一个以用户分享为主的开源技术平台,欢迎各类分享!