返回

玩转Spring Boot中的@Async线程池拒绝策略，让异步任务稳如泰山

后端

Spring Boot 中@Async 的拒绝策略：打造稳如泰山的异步任务

简介

在 Spring Boot 的世界中，@Async 注解可谓是异步编程的利器，它可以轻松将方法标记为异步任务，让它们在独立线程中自由驰骋。这样的操作，不仅可以显著提升应用性能，尤其是在处理大量耗时任务时，更能发挥其独到之处。

然而，在使用 @Async 注解时，你可能会遭遇一个棘手的问题——线程池队列已满。当队列爆满时，默认的拒绝策略会毫不留情地抛出 RejectedExecutionException 异常，宣告无法接受新任务。这种毫不留情的拒绝，可能会导致任务丢失，进而动摇应用的稳定根基。

自定义拒绝策略：掌控异步任务的命运

为了化解队列满载的危机，我们可以祭出自定义拒绝策略这把利器。自定义拒绝策略赋予我们掌控权，让我们能够决定队列爆满时如何处理新任务。

默认的拒绝策略

Spring Boot 为 @Async 注解提供了默认的拒绝策略——AbortPolicy。它的行事风格简单粗暴，一旦任务提交至线程池，队列已满，它就会毫不犹豫地抛出 RejectedExecutionException 异常，无情地拒绝新任务的加入。

ThreadPoolExecutor executor = new ThreadPoolExecutor(10, 20, 10, TimeUnit.SECONDS, new LinkedBlockingQueue<>(100), new AbortPolicy());

自定义拒绝策略的诞生

要打造属于自己的拒绝策略，需要遵循 RejectedExecutionHandler 接口的指引。RejectedExecutionHandler 接口只有一个方法——rejectedExecution。每当任务提交至线程池，队列已满时，该方法就会粉墨登场，由它来决定新任务的去留。

public class CustomRejectedExecutionHandler implements RejectedExecutionHandler {

    @Override
    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
        // 自定义拒绝策略的逻辑
    }
}

打造好自定义拒绝策略后，就可以在 ThreadPoolExecutor 中指定它的存在。

ThreadPoolExecutor executor = new ThreadPoolExecutor(10, 20, 10, TimeUnit.SECONDS, new LinkedBlockingQueue<>(100), new CustomRejectedExecutionHandler());

常见的拒绝策略

除了 AbortPolicy，Spring Boot 还贴心地提供了几款常见的拒绝策略，以备不时之需：

  • CallerRunsPolicy ：当队列爆满时，任务会礼貌地提交至调用线程。这样一来，虽然可能会导致调用线程暂时卡壳，但至少可以避免任务丢失。
  • DiscardOldestPolicy ：当队列爆满时，它会毫不留情地抛弃队列中年纪最大的任务。虽然这可以确保新任务的顺利加入，但也存在遗漏重要任务的风险。
  • DiscardPolicy ：当队列爆满时，它会毫不犹豫地丢弃新任务。虽然这可能会导致任务丢失，但也能够确保线程池不会爆仓。

选择拒绝策略：因需而异

在选择拒绝策略时，需要结合以下因素仔细斟酌：

  • 任务重要性 ：如果任务至关重要，那么应当选择能够避免任务丢失的拒绝策略，比如 CallerRunsPolicy。
  • 线程池饱和程度 ：如果线程池经常处于饱和状态，那么应当选择能够防止线程池爆仓的拒绝策略，比如 DiscardPolicy。
  • 应用性能 ：如果应用对性能要求苛刻，那么应当选择既能避免任务丢失，又不会导致调用线程阻塞的拒绝策略，比如 CallerRunsPolicy。

结语

通过自定义拒绝策略，我们可以巧妙地掌控队列爆满时的任务处理方式，避免任务丢失，确保任务顺利执行，从而为应用稳定性和性能保驾护航。

常见问题解答

  1. 为什么需要自定义拒绝策略？
    答：自定义拒绝策略可以让我们灵活地控制队列爆满时的任务处理方式，避免任务丢失，确保线程池平稳运行。

  2. 有哪些常见的拒绝策略可供选择？
    答：Spring Boot 提供了多种拒绝策略，包括 AbortPolicy、CallerRunsPolicy、DiscardOldestPolicy 和 DiscardPolicy。

  3. 如何选择合适的拒绝策略？
    答：拒绝策略的选择需要根据任务重要性、线程池饱和程度和应用性能要求等因素综合考虑。

  4. 自定义拒绝策略时，需要遵循哪些原则？
    答：自定义拒绝策略时，需要遵循 RejectedExecutionHandler 接口，并根据具体场景制定相应的逻辑。

  5. 如何配置自定义拒绝策略？
    答：在创建 ThreadPoolExecutor 时，可以指定自定义拒绝策略作为其参数之一。