ThreadPoolExecutor拒绝策略

ThreadPoolExecutor 的拒绝策略（RejectedExecutionHandler）在线程池无法处理新的任务时会被触发。下面介绍 Java 中 ThreadPoolExecutor 提供的几种内置拒绝策略，并展示如何自定义拒绝策略。

1. 内置拒绝策略

1. AbortPolicy (默认策略)

   • 当任务无法被接受时，抛出 RejectedExecutionException 异常。

   new ThreadPoolExecutor.AbortPolicy()
   

2. CallerRunsPolicy

   • 当任务无法被接受时，调用线程池的线程会执行该任务。这种策略提供了一种“回退”机制，从而降低新任务提交速度。

   new ThreadPoolExecutor.CallerRunsPolicy()
   

3. DiscardPolicy

   • 当任务无法被接受时，直接丢弃任务，不做任何处理。

   new ThreadPoolExecutor.DiscardPolicy()
   

4. DiscardOldestPolicy

   • 当任务无法被接受时，丢弃工作队列中最旧的未处理任务，然后重新尝试提交新任务。

   new ThreadPoolExecutor.DiscardOldestPolicy()
   

2. 自定义拒绝策略

可以通过实现 RejectedExecutionHandler 接口来创建自定义的拒绝策略。例如，记录被拒绝任务的日志或进行其他处理。

import java.util.concurrent.*;

public class CustomRejectedExecutionHandler implements RejectedExecutionHandler {
    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        System.out.println("Task " + r.toString() + " rejected");
    }

    public static void main(String[] args) {
        ThreadPoolExecutor executor = new ThreadPoolExecutor(
            2, // corePoolSize
            4, // maximumPoolSize
            60, // keepAliveTime
            TimeUnit.SECONDS, // timeUnit
            new ArrayBlockingQueue<>(2), // workQueue
            Executors.defaultThreadFactory(), // threadFactory
            new CustomRejectedExecutionHandler() // rejectedExecutionHandler
        );

        for (int i = 0; i < 10; i++) {
            executor.execute(new Task(i));
        }

        executor.shutdown();
    }

    static class Task implements Runnable {
        private final int taskId;

        public Task(int taskId) {
            this.taskId = taskId;
        }

        @Override
        public void run() {
            System.out.println("Executing task " + taskId);
            try {
                Thread.sleep(2000); // Simulate task execution time
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }
            System.out.println("Task " + taskId + " completed");
        }

        @Override
        public String toString() {
            return "Task{" + "taskId=" + taskId + '}';
        }
    }
}

• CustomRejectedExecutionHandler：自定义拒绝策略，实现 RejectedExecutionHandler 接口。在任务被拒绝时，打印被拒绝任务的信息。

• ThreadPoolExecutor 配置：

  • 核心线程数为 2。
  • 最大线程数为 4。
  • 工作队列大小为 2。
  • 自定义拒绝策略。

• 任务提交：提交了 10 个任务。前两个任务由核心线程处理，接下来的两个任务进入工作队列。由于工作队列和线程池达到最大容量，后续任务将触发自定义拒绝策略，打印被拒绝任务的信息。

3. 总结

通过使用内置拒绝策略或实现自定义拒绝策略，可以灵活处理线程池无法接受新任务的情况，从而提高系统的健壮性和可控性。选择哪种策略取决于具体的应用场景和需求。