如何按优先级从Java Stream中获取首个匹配值？

问题原因 #

你遇到的IllegalStateException是因为Java Stream是一次性资源：调用findFirst()这类终端操作后，流会被消耗或关闭，无法再对同一个流对象执行后续操作。原代码中第一次stream.filter(...).findFirst()已经耗尽了流，后续的stream.filter()自然会触发报错。

正确实现方案 #

方案1：单次遍历跟踪最高优先级元素（推荐） #

通过一次遍历完成所有条件检查，找到第一个符合最高优先级的元素，效率最高且完全避免流重复操作：

import java.util.stream.Stream;

private static String findBestValue(Stream<String> stream) {
    String result = null;
    var iterator = stream.iterator();
    while (iterator.hasNext()) {
        String current = iterator.next();
        if (result != null) {
            continue; // 已找到高优先级元素，跳过后续检查
        }
        if (current.equals("a")) {
            result = current;
        } else if (current.equals("b")) {
            result = current;
        } else if (current.equals("c")) {
            result = current;
        }
    }
    return result;
}

方案2：函数式风格的reduce实现 #

利用reduce操作聚合流元素，保持Java Stream的函数式编程风格：

import java.util.stream.Stream;

private static String findBestValue(Stream<String> stream) {
    return stream.reduce(null, (best, current) -> {
        if (best != null) {
            return best; // 保留已找到的高优先级元素
        }
        if (current.equals("a")) {
            return current;
        } else if (current.equals("b")) {
            return current;
        } else if (current.equals("c")) {
            return current;
        }
        return best; // 无匹配条件，返回原有值（null）
    });
}

方案3：可扩展的优先级条件列表 #

如果需要灵活调整优先级规则，可以将条件按顺序存入列表，通过一次遍历找到最高优先级的匹配元素：

import java.util.Arrays;
import java.util.List;
import java.util.function.Predicate;
import java.util.stream.Stream;

private static String findBestValue(Stream<String> stream) {
    // 按优先级从高到低定义条件列表
    List<Predicate<String>> priorityConditions = Arrays.asList(
        str -> str.equals("a"),
        str -> str.equals("b"),
        str -> str.equals("c")
    );

    return stream.filter(str -> priorityConditions.stream().anyMatch(pred -> pred.test(str)))
                 .min((s1, s2) -> {
                     // 比较元素的优先级索引，返回优先级更高的元素
                     int p1 = priorityConditions.indexOf(pred -> pred.test(s1));
                     int p2 = priorityConditions.indexOf(pred -> pred.test(s2));
                     return Integer.compare(p1, p2);
                 })
                 .orElse(null);
}

验证结果 #

以上三种方案均符合你的示例预期：

• findBestValue(stream0) 返回 "a"
• findBestValue(stream1) 返回 "a"
• findBestValue(stream2) 返回 "b"
• findBestValue(stream3) 返回 null

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