Java 线程池拒绝策略：你必须知道的那些事

在 Java 开发中，线程池是管理并发任务的重要工具之一。线程池通过重用现有线程来减少创建和销毁线程的开销，从而提高了系统的性能和资源利用率。然而，当线程池中的任务数量超过了其处理能力时，线程池会采用拒绝策略来处理这些额外的任务。本文将详细介绍 Java 线程池的拒绝策略，以及它们在实际应用中的使用场景。

线程池的基本概念

首先，让我们回顾一下线程池的基本概念。线程池（ThreadPoolExecutor）在 Java 中主要由以下几个参数定义：

corePoolSize：核心线程池大小。
maximumPoolSize：最大线程池大小。
keepAliveTime：线程池中超过核心线程数的空闲线程的存活时间。
workQueue：用于保存等待执行的任务的阻塞队列。
threadFactory：用于创建新线程的工厂。
handler：当执行被拒绝时的处理策略。

拒绝策略的种类

当线程池和工作队列都已满时，线程池会根据预设的拒绝策略来处理新提交的任务。Java 提供了四种标准的拒绝策略：

AbortPolicy：直接抛出 RejectedExecutionException 异常，阻止系统正常运行。

ExecutorService executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.MILLISECONDS, workQueue, new ThreadPoolExecutor.AbortPolicy());

CallerRunsPolicy：由调用线程（提交任务的线程）来运行该任务。

ExecutorService executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.MILLISECONDS, workQueue, new ThreadPoolExecutor.CallerRunsPolicy());

DiscardPolicy：直接丢弃任务，不做任何处理。

ExecutorService executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.MILLISECONDS, workQueue, new ThreadPoolExecutor.DiscardPolicy());

DiscardOldestPolicy：丢弃队列中最旧的任务，然后尝试再次提交当前任务。

ExecutorService executor = new ThreadPoolExecutor(corePoolSize, maximumPoolSize, keepAliveTime, TimeUnit.MILLISECONDS, workQueue, new ThreadPoolExecutor.DiscardOldestPolicy());

应用场景

AbortPolicy：适用于对任务执行有严格要求的场景，如金融交易系统，任何任务的失败都可能导致严重后果。
CallerRunsPolicy：适用于系统负载较高时，允许调用者线程执行任务，减轻线程池的压力，适用于一些非关键任务。
DiscardPolicy：适用于任务可以被忽略的场景，如日志记录系统，丢失一些日志记录不会影响系统的整体功能。
DiscardOldestPolicy：适用于需要优先处理新任务的场景，如实时数据处理系统，旧数据可能不再重要。

自定义拒绝策略

除了标准的拒绝策略，Java 还允许开发者自定义拒绝策略。通过实现 RejectedExecutionHandler 接口，可以根据具体需求来处理被拒绝的任务。例如：

class CustomRejectPolicy implements RejectedExecutionHandler {
    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        // 自定义处理逻辑
        System.out.println("Task Rejected: " + r.toString());
    }
}

总结

Java 线程池的拒绝策略是线程池管理中的一个关键环节，选择合适的拒绝策略可以有效地控制系统的负载和稳定性。在实际应用中，根据业务需求选择或自定义拒绝策略，不仅能提高系统的健壮性，还能优化资源的使用效率。希望本文能帮助大家更好地理解和应用线程池的拒绝策略，提升系统的并发处理能力。