无惧拒绝，揭秘Java线程池拒绝策略的奥秘

后端

2023-05-23 13:10:52

线程池拒绝策略：保障应用程序平稳运行的幕后英雄

引言：

在 Java 并发编程的浩瀚世界中，线程池扮演着至关重要的角色。它就像交通管制员，协调任务的执行，提升应用程序的性能和稳定性。然而，当任务蜂拥而至，超过线程池的处理能力时，就需要依靠一种名为拒绝策略的机制来决定如何处理这些过剩的任务。

拒绝策略类型

Java 线程池提供了四种内置的拒绝策略，每种策略都以独特的方式处理无法执行的任务：

AbortPolicy： 以抛出异常的方式通知调用者任务无法被执行。这种策略简单直接，但可能导致应用程序崩溃。
CallerRunsPolicy： 让调用者线程自己执行任务。这种策略避免了应用程序崩溃，但可能会导致调用者线程阻塞。
DiscardOldestPolicy： 丢弃队列中最老的任务，为新任务腾出空间。这种策略防止任务丢失，但可能会影响已等待任务的执行顺序。
DiscardPolicy： 直接丢弃新任务，不会进入队列也不会执行。这种策略最简单，但会造成任务丢失。

选择合适的拒绝策略

选择合适的拒绝策略取决于应用程序的具体需求。以下是一些指导原则：

对于不允许任务丢失的应用程序，建议使用 AbortPolicy 或 CallerRunsPolicy。
对于可以容忍任务丢失的应用程序，DiscardOldestPolicy 或 DiscardPolicy 是更好的选择。
对于需要根据任务优先级进行处理的应用程序，可以使用 PriorityBlockingQueue 结合 AbortPolicy 或 CallerRunsPolicy。
对于需要超时处理任务的应用程序，可以使用 DelayedWorkQueue 结合 AbortPolicy 或 CallerRunsPolicy。

代码示例

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class ThreadPoolRejectionPolicyDemo {

    public static void main(String[] args) {
        // 创建一个使用 AbortPolicy 的线程池
        ExecutorService executorService = new ThreadPoolExecutor(
            1,
            1,
            0L,
            TimeUnit.MILLISECONDS,
            new ArrayBlockingQueue<>(1),
            new ThreadPoolExecutor.AbortPolicy()
        );

        // 向线程池提交任务
        for (int i = 0; i < 10; i++) {
            executorService.submit(() -> {
                System.out.println("任务：" + i + " 执行中");
                try {
                    TimeUnit.SECONDS.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            });
        }

        // 关闭线程池
        executorService.shutdown();
    }
}