如何对Java Stream逐层过滤直至仅剩一个匹配元素？

核心需求回顾 #

按指定顺序应用多层过滤条件处理对象列表，每一步过滤后执行以下检查：

• 若结果仅含1个元素，直接返回该元素
• 若结果为空，抛出异常
• 全部过滤完成后仍不符合（数量不为1），同样抛出异常
  注：原代码中第三步过滤基于第一步的结果而非第二步，需注意该逻辑是否为预期需求

优化方案：提取通用逻辑减少冗余 #

原代码存在大量重复的过滤、检查代码，我们可以将这部分逻辑封装为通用方法，提升代码简洁性和可维护性。

1. 封装通用过滤检查方法

import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.function.Predicate;
import java.util.stream.Collectors;

// 自定义异常用于提前返回结果，避免重复检查逻辑
class EarlyReturnException extends RuntimeException {
    private final Object result;

    public EarlyReturnException(Object result) {
        this.result = result;
    }

    public Object getResult() {
        return result;
    }
}

private List<Object> filterAndCheck(List<Object> source, Predicate<Object> filter) throws Exception {
    List<Object> filtered = Optional.ofNullable(source)
                                   .orElse(Collections.emptyList())
                                   .stream()
                                   .filter(filter)
                                   .collect(Collectors.toList());

    if (filtered.isEmpty()) {
        throw new Exception("过滤后无匹配元素");
    }
    if (filtered.size() == 1) {
        throw new EarlyReturnException(filtered.get(0));
    }
    return filtered;
}

2. 简化主逻辑

public Object findMatchingObject() throws Exception {
    List<Object> initialList = getAllObjects();

    try {
        // 第一步过滤
        List<Object> step1 = filterAndCheck(initialList, filter1());
        // 第二步过滤（基于第一步结果）
        filterAndCheck(step1, filter2());
        // 第三步过滤（原逻辑基于第一步结果，而非第二步）
        filterAndCheck(step1, filter3());
    } catch (EarlyReturnException e) {
        return e.getResult();
    }

    // 全部过滤后仍未得到唯一元素
    throw new Exception("无唯一匹配元素");
}

其他优化建议 #

• 替换自定义emptyIfNull：用Java标准库的Optional.ofNullable(source).orElse(Collections.emptyList())替代，无需额外实现工具方法。
• 使用具体异常类型：避免抛出通用的Exception，自定义NoMatchingElementException、MultipleMatchingElementsException等异常，提升代码可读性和异常处理精准度。
• 验证过滤逻辑顺序：原代码第三步过滤基于第一步结果，若为笔误（应为第二步结果），只需将第三步的step1改为step2即可。

内容的提问来源于stack exchange，提问作者Link