异步请求中的拒绝策略陷阱，你避开了吗？

2023-02-10 03:49:22

用CompletableFuture掌控任务执行的拒绝策略

在现代应用程序设计中，多线程扮演着至关重要的角色，它可以提升并发应用的性能和响应能力。CompletableFuture是Java中一种强大的异步编程工具，它提供了便捷的方式来执行和获取异步任务的结果。然而，在使用CompletableFuture时，开发者经常会忽略处理任务执行过程中产生的拒绝策略这个关键问题。

线程池和拒绝策略

当我们创建一个CompletableFuture的线程池时，需要设定合理的线程池大小，以确保任务能够及时处理。如果任务数量超出线程池的容量，CompletableFuture提供了三种默认的拒绝策略：

AbortPolicy： 直接抛出RejectedExecutionException异常（这是默认的拒绝策略）
CallerRunsPolicy： 在线程池的调用线程中执行任务（这可能导致调用线程被阻塞）
DiscardPolicy： 直接丢弃任务（不执行也不抛出异常）

避免默认拒绝策略的陷阱

默认情况下，CompletableFuture使用AbortPolicy 作为拒绝策略。这意味着，当任务数量超过线程池的容量时，CompletableFuture会直接抛出RejectedExecutionException异常。这可能导致应用程序崩溃或出现其他不可预期的行为。

选择合适的拒绝策略

为了避免这些问题，我们需要显式地为CompletableFuture设置拒绝策略。根据应用程序的具体需求，有几种常用的拒绝策略可供选择：

AbortPolicy： 直接抛出RejectedExecutionException异常（这是默认的拒绝策略）
CallerRunsPolicy： 在线程池的调用线程中执行任务（这可能会导致调用线程被阻塞）
DiscardPolicy： 直接丢弃任务（不执行也不抛出异常）
ThreadPoolExecutor.CallerRunsPolicy： 在线程池的调用线程中执行任务，如果执行任务的线程已被中断，则抛出RejectedExecutionException异常
ThreadPoolExecutor.DiscardOldestPolicy： 丢弃队列中最旧的任务，并尝试执行当前任务，如果队列已满，则抛出RejectedExecutionException异常

在选择拒绝策略时，需要考虑应用程序的具体需求。例如，如果应用程序不能容忍任务被丢弃，那么可以选择AbortPolicy 或CallerRunsPolicy 。如果应用程序可以容忍任务被丢弃，那么可以选择DiscardPolicy 或ThreadPoolExecutor.DiscardOldestPolicy 。

自定义拒绝策略

在实际使用中，还可以根据业务需求来自定义拒绝策略。例如，我们可以编写一个自定义的拒绝策略，在任务被拒绝时发送邮件通知管理员，或者将任务保存到数据库中，以便稍后重新执行。

代码示例：自定义拒绝策略

import java.util.concurrent.RejectedExecutionException;
import java.util.concurrent.ThreadPoolExecutor;

public class CustomRejectionPolicy implements RejectedExecutionHandler {
    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        // 自定义处理逻辑，例如发送邮件通知或保存任务到数据库
    }
}