Java Stream中如何实现中间条件检查？

Java Stream本身没有原生的中间操作可以直接在流为空时抛出异常——这是因为Stream采用惰性求值，所有中间操作只有在终端操作触发时才会执行，无法在中间步骤直接判断流是否为空。不过我们可以通过几种方式模拟Reactor中switchIfEmpty的中间检查效果：

方法1：拆分流做阶段性终端检查 #

最直接的方式是把每一步过滤后的流单独做终端检查，确认非空后再继续处理。缺点是会遍历集合多次，数据量大时效率较低：

public String interimConditionCheckStream(Collection<String> input) {
    // 第一步过滤并检查是否为空
    input.stream()
            .filter(s -> s.startsWith("A"))
            .findAny()
            .orElseThrow(() -> new IllegalArgumentException("No string starts with 'A'"));
    
    // 继续后续过滤和处理
    return input.stream()
            .filter(s -> s.startsWith("A"))
            .filter(s -> s.length() > 10)
            .findAny()
            .orElseThrow(() -> new IllegalArgumentException("No string is long enough"));
}

方法2：自定义工具方法实现中间非空校验 #

通过自定义工具方法，在每一步过滤后将元素收集到临时列表，检查非空后再返回新流。这种方式只需要遍历一次，但会产生中间列表的内存开销：

import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class StreamHelper {
    public static <T> Stream<T> filterAndEnsureNotEmpty(Collection<T> source, 
                                                       java.util.function.Predicate<? super T> predicate, 
                                                       RuntimeException exception) {
        List<T> filteredList = source.stream()
                .filter(predicate)
                .collect(Collectors.toList());
        
        if (filteredList.isEmpty()) {
            throw exception;
        }
        return filteredList.stream();
    }
}

使用时直接链式调用：

public String interimConditionCheckStream(Collection<String> input) {
    return StreamHelper.filterAndEnsureNotEmpty(input, s -> s.startsWith("A"), 
                    new IllegalArgumentException("No string starts with 'A'"))
            .filter(s -> s.length() > 10)
            .findAny()
            .orElseThrow(() -> new IllegalArgumentException("No string is long enough"));
}

方法3：利用原子变量跟踪元素（注意局限性） #

可以尝试用AtomicBoolean跟踪是否有元素通过过滤，但因为Stream的惰性特性，异常只会在最终终端操作时抛出，无法严格实现「中间步骤立即终止」的效果，仅适合不需要严格中间终止的场景：

import java.util.Collection;
import java.util.concurrent.atomic.AtomicBoolean;

public String interimConditionCheckStream(Collection<String> input) {
    AtomicBoolean hasStartA = new AtomicBoolean(false);
    
    return input.stream()
            .filter(s -> {
                boolean matches = s.startsWith("A");
                if (matches) hasStartA.set(true);
                return matches;
            })
            .filter(s -> s.length() > 10)
            .findAny()
            .orElseThrow(() -> {
                if (!hasStartA.get()) {
                    return new IllegalArgumentException("No string starts with 'A'");
                } else {
                    return new IllegalArgumentException("No string is long enough");
                }
            });
}

总结 #

如果追求严格的中间步骤异常抛出，推荐使用方法2；如果数据量极小且追求代码简洁，方法1也可以接受；方法3仅适合不需要严格中间终止的场景。

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