亲历并发编程中的陷阱：揭秘CompletableFuture的阻塞行为

后端

2023-12-05 02:07:25

CompletableFuture：避免阻塞陷阱，编写健壮的并发代码

简介

并发编程可以极大地提升程序性能，但同时也带来了隐患，最常见的莫过于阻塞问题。CompletableFuture 是 Java 中一项强大的并发编程工具，它能够处理异步任务，但其阻塞行为却潜藏着任务丢失的陷阱。

CompletableFuture 的阻塞行为

CompletableFuture.get() 方法是一个阻塞方法，会等待异步任务完成并返回结果。如果线程池饱和，CompletableFuture 的默认策略是丢弃任务，而不会抛出异常。

陷阱：线程池饱和导致任务丢失

当线程池饱和时，任务可能会被直接丢弃，而不会抛出异常，这会导致任务丢失。为了避免这种陷阱，建议使用 AbortPolicy 线程池策略，它会在线程池饱和时抛出异常。

隔离耗时异步线程

对于耗时的异步线程，建议将其与其他线程池隔离，防止其影响其他线程的执行。

案例演示

以下代码示例演示了 CompletableFuture 的阻塞行为以及如何避免任务丢失陷阱：

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

public class CompletableFutureExample {

    public static void main(String[] args) {
        ExecutorService executorService = Executors.newFixedThreadPool(2, Executors.defaultThreadFactory(), new AbortPolicy());
        CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
            try {
                Thread.sleep(5000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }, executorService);
        try {
            future.get(3, TimeUnit.SECONDS);
        } catch (Exception e) {
            System.out.println("任务完成失败，原因：" + e.getMessage());
        }
        executorService.shutdown();
    }
}

运行这段代码，我们会看到输出如下：

任务完成失败，原因：java.util.concurrent.RejectedExecutionException: Task com.example.CompletableFutureExample$1$$Lambda$1/897479909@6de023d6 rejected from java.util.concurrent.ThreadPoolExecutor@61e4933a[Running, pool size = 2, active threads = 2, queued tasks = 0, completed tasks = 0]