Java线程池的拒绝策略实现详解

一、简介

jdk1.5 版本新增了JUC并发编程包，大大的简化了传统的多线程开发。

Java线程池，是典型的池化思想的产物，类似的还有数据库的连接池、redis的连接池等。池化思想，就是在初始的时候去申请资源，创建一批可使用的连接，这样在使用的时候，就不必再进行创建连接信息的开销了。举个生活中鲜明的例子，在去著名洋快餐某基或者某劳的时候，配餐人员是字节从一个中间的保温箱里面直接取，然后打包就好了。不用再临时的来了一个单子，又要去拿原材料，又要去进行加工。效率明显的就是提高了很多。

既然是池子，那么必然的会存在最大值、初始值以及存活值等属性。在达到某些特定条件的时候，再来请求的话，池子是如何进行请求处理的呢？这里就引出了池的拒绝策略。一般的数据库连接池在达到最大连接数的时候会默认的等待特定的设置的时间或者直接就抛出异常。而本文中要阐述的线程池却并非如此的策略，下面开始展开讲解下。

二、线程池的拒绝策略

线程池中，有三个重要的参数，决定影响了拒绝策略：corePoolSize - 核心线程数，也即最小的线程数。workQueue - 阻塞队列 。maximumPoolSize - 最大线程数

当提交任务数大于 corePoolSize 的时候，会优先将任务放到 workQueue 阻塞队列中。当阻塞队列饱和后，会扩充线程池中线程数，直到达到 maximumPoolSize 最大线程数配置。此时，再多余的任务，则会触发线程池的拒绝策略了。

总结起来，也就是一句话，当提交的任务数大于（workQueue.size() + maximumPoolSize ），就会触发线程池的拒绝策略。

三、拒绝策略定义

拒绝策略提供顶级接口 RejectedExecutionHandler ，其中方法 rejectedExecution 即定制具体的拒绝策略的执行逻辑。

jdk默认提供了四种拒绝策略：CallerRunsPolicy - 当触发拒绝策略，只要线程池没有关闭的话，则使用调用线程直接运行任务。

一般并发比较小，性能要求不高，不允许失败。但是，由于调用者自己运行任务，如果任务提交速度过快，可能导致程序阻塞，性能效率上必然的损失较大

AbortPolicy - 丢弃任务，并抛出拒绝执行 RejectedExecutionException 异常信息。线程池默认的拒绝策略。必须处理好抛出的异常，否则会打断当前的执行流程，影响后续的任务执行。

 DiscardPolicy - 直接丢弃，其他啥都没有

DiscardOldestPolicy - 当触发拒绝策略，只要线程池没有关闭的话，丢弃阻塞队列 workQueue 中最老的一个任务，并将新任务加入

四、测试代码

1、AbortPolicy

package com.cfang;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.RejectedExecutionHandler;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import lombok.extern.slf4j.Slf4j;
@Slf4j
public class T2 {
 public static void main(String[] args) throws Exception{
  int corePoolSize = 5;
  int maximumPoolSize = 10;
  long keepAliveTime = 5;
  BlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<Runnable>(10);
  RejectedExecutionHandler handler = new ThreadPoolExecutor.AbortPolicy();
  ThreadPoolExecutor executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.SECONDS, workQueue, handler);
  for(int i=0; i<100; i++) {
   try {
    executor.execute(new Thread(() -> log.info(Thread.currentThread().getName() + " is running")));
   } catch (Exception e) {
    log.error(e.getMessage());
   }
  }
  executor.shutdown();
 }
}

如果 executor.execute（）提交任务，由于会抛出 RuntimeException，没有try.catch处理异常信息的话，会中断调用者的处理流程，后续任务得不到执行（跑不完100个）。可自行测试下。

2、CallerRunsPolicy

主体代码同上，更换拒绝策略：

RejectedExecutionHandler handler = new ThreadPoolExecutor.CallerRunsPolicy();

运行后，在控制台console中能够看到的是，会有一部分的数据打印，显示的是 “main is running”，也即体现调用线程处理。

3、DiscardPolicy

更换拒绝策略

RejectedExecutionHandler handler = new ThreadPoolExecutor.DiscardPolicy();

直接丢弃任务，实际运行中，打印出的信息不会有100条。

4、DiscardOldestPolicy

同样的，更换拒绝策略：

RejectedExecutionHandler handler = new ThreadPoolExecutor.DiscardOldestPolicy();

实际运行，打印出的信息也会少于100条。

五、总结

四种拒绝策略是相互独立无关的，选择何种策略去执行，还得结合具体的业务场景。实际工作中，一般直接使用 ExecutorService 的时候，都是使用的默认的 defaultHandler ，也即 AbortPolicy 策略。

最新评论