Java线程池的抛弃策略

Java线程池的抛弃策略

Java线程池是Java并发编程中非常重要的一个组件。它通过重用已创建的线程来减少线程创建和销毁的开销，从而提高应用程序的性能和响应速度。然而，当线程池中的任务数量超过其处理能力时，就需要一种机制来处理新提交的任务，这就是线程池的抛弃策略。

线程池的抛弃策略（Rejection Policy）是指当线程池无法接受新的任务时，如何处理这些任务的策略。通常，线程池在以下两种情况下会拒绝新的任务：

1. 当线程池已达到最大线程数，且所有线程都在忙碌中。
2. 当任务队列已满，无法容纳更多的任务。

常见的抛弃策略

Java提供了几种内置的抛弃策略，分别是：

AbortPolicy

AbortPolicy 是默认的抛弃策略。当任务被拒绝时，线程池会抛出一个 RejectedExecutionException 异常。这个策略通常用于希望调用者能够意识到任务被拒绝，并且可以进行相应处理的场景。

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

DiscardPolicy

DiscardPolicy 直接丢弃被拒绝的任务，不予任何处理或通知。这种策略适用于对丢弃任务不敏感的场景。

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

DiscardOldestPolicy

DiscardOldestPolicy 将丢弃队列中最旧的任务，然后尝试重新提交被拒绝的任务。这种策略适用于需要优先处理新任务的场景。

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

CallerRunsPolicy

CallerRunsPolicy 由调用者线程处理被拒绝的任务。这种策略可以降低新任务提交的速度，从而减少任务被拒绝的可能性，适用于希望尽量不丢弃任务的场景。

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

自定义抛弃策略

除了内置的抛弃策略，开发者还可以实现 RejectedExecutionHandler 接口来自定义抛弃策略。例如，可以将被拒绝的任务记录到日志中，或者放入一个备用队列中以备后续处理。

public class CustomRejectedExecutionHandler implements RejectedExecutionHandler {
    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        // 自定义处理逻辑，例如记录日志或放入备用队列
        System.out.println("Task " + r.toString() + " rejected from " + executor.toString());
    }
}

ExecutorService executor = new ThreadPoolExecutor(
    corePoolSize, 
    maximumPoolSize, 
    keepAliveTime, 
    timeUnit, 
    workQueue, 
    new CustomRejectedExecutionHandler()
);

选择合适的抛弃策略

选择合适的抛弃策略需要根据应用程序的具体需求和场景来定。例如：

• 如果希望调用者能够处理被拒绝的任务，可以选择 AbortPolicy 或 CallerRunsPolicy。
• 如果任务可以被安全地丢弃，可以选择 DiscardPolicy 或 DiscardOldestPolicy。
• 如果需要对被拒绝的任务进行特殊处理，可以实现自定义的抛弃策略。