多线程与短路求值：用 CompletableFuture 实现任务快速返回

Z丶T

已于 2024-10-03 10:16:05 修改

阅读量271

点赞数 7

文章标签： java 性能优化

于 2024-10-03 01:38:10 首次发布

本文链接：https://blog.csdn.net/weixin_43311068/article/details/142687513

版权

详细地址：Java多线程：逻辑表达式的短路运算
方法进阶：Java多线程：复杂逻辑表达式的短路运算

一、问题背景

在复杂业务逻辑中，我们常常需要并发处理多个任务，比如检查多个条件 A && B && C 是否都满足。为了提高效率，当某个任务返回 false 时，我们能否立刻结束其他任务并返回结果呢？

本文将通过 Java 中的 CompletableFuture 及其 anyOf 和 allOf 方法，演示如何实现这一需求。

二、方案设计

目标是并发处理多个任务，当任意一个任务返回 false 时，立即终止其他任务并返回结果。我们可以通过 allOf 来并行处理多个任务，同时结合 anyOf 来在短路条件满足时立刻返回结果。

代码结构设计

组合 anyOf 和 allOf：使用 allOf 并发执行所有任务，一旦某个任务返回 false，通过终止线程 terminationFuture 触发 anyOf 机制，提前结束未完成的任务。如果所有任务都返回 true，则正常结束。
原子类 AtomicBoolean：一个线程安全的布尔值，用来存储逻辑表达式的最终结果。
终止线程 terminationFuture：当任意任务返回 false 时，调用 complete() 方法，提前触发 anyOf 机制。

在这个实现中，一旦某个任务返回 false，其他任务将被忽略并提前返回，避免了不必要的资源消耗。以下是代码示例：

public class MultiAnd {
    // 超时时间
    public static final long TIMEOUT = 1000L;
    // 条件 A 、B 、C ... 实现
    List<Predicate<ExpressionFacts>> expressionList = new ArrayList<>();
    // 为了方便演示，使用 newFixedThreadPool 方法，进行线程池创建
    ExecutorService executor = Executors.newFixedThreadPool(10);

    /**
     * 实现 A && B && C && ... 快速失败
     *
     * @param facts 方法入参
     * @return A && B && C && ... 的最终结果
     */
    public boolean evaluateAnd(ExpressionFacts facts) {
        // result 为该方法返回的最终结果，使用 Atomic 以保证变量的原子性操作
        AtomicBoolean result = new AtomicBoolean(true);
        // 终止线程，用来根据不同任务的执行结果决定是否要提前返回
        CompletableFuture<Void> terminationFuture = new CompletableFuture<>();

        CompletableFuture.anyOf(terminationFuture, CompletableFuture.allOf(expressionList.stream()
                .map(expression -> CompletableFuture.runAsync(() -> {
                            if (!result.get()) {
                                return; // 如果已经有规则不符合，直接返回
                            }
                            boolean ruleResult = expression.test(facts); // 不同任务执行的结果
                            if (!ruleResult && result.compareAndSet(true, false)) {
                                terminationFuture.complete(null); // 完成终止线程，提前返回多线程结果
                            }
                        }, executor)
                        .orTimeout(TIMEOUT, TimeUnit.MILLISECONDS) // JDK 9 引入的异步超时处理方法
                        .exceptionally(ex -> {
                            // 捕获异常，结果算作 false
                            if (result.compareAndSet(true, false)) {
                                terminationFuture.complete(null);
                            }
                            return null;
                        })).toArray(CompletableFuture[]::new))).join();

        return result.get();
    }    
}