如何从单个Stream中收集多列表并优雅合并？附现有实现

当然有更优雅的方式啦！你当前的实现需要遍历三次原始列表，不仅重复操作，效率也不够高。我们可以通过一次遍历就完成所有id的收集，同时保持代码简洁易读。

基础优化方案：用flatMap合并多值流 #

最直接的优化是利用flatMap，把每个MyClass实例对应的三个id转换成一个小流，再将所有小流合并成一个整体流，最后直接收集到Set中：

private Set<Long> getAllIds(List<MyClass> fetchedElements) {
    return fetchedElements.stream()
        .flatMap(element -> Stream.of(element.getBlueId(), element.getRedId(), element.getGreenId()))
        .collect(Collectors.toSet());
}

处理可能的null值 #

如果你的id字段可能为null，可以在收集前加上过滤逻辑，避免把null值加入Set：

private Set<Long> getAllIds(List<MyClass> fetchedElements) {
    return fetchedElements.stream()
        .flatMap(element -> Stream.of(element.getBlueId(), element.getRedId(), element.getGreenId()))
        .filter(Objects::nonNull)
        .collect(Collectors.toSet());
}

高扩展性方案：提取id方法列表 #

如果以后需要新增更多id字段（比如黄色id、紫色id），可以把所有提取id的方法引用放到一个列表里，这样后续扩展只需要新增方法引用，不需要修改流的核心逻辑：

private Set<Long> getAllIds(List<MyClass> fetchedElements) {
    // 集中管理所有id提取逻辑
    List<Function<MyClass, Long>> idExtractors = Arrays.asList(
        MyClass::getBlueId,
        MyClass::getRedId,
        MyClass::getGreenId
    );
    
    return fetchedElements.stream()
        .flatMap(element -> idExtractors.stream().map(extractor -> extractor.apply(element)))
        .filter(Objects::nonNull)
        .collect(Collectors.toSet());
}

这些方案都只需要遍历一次原始列表，既提升了效率，又让代码更符合Stream API的函数式编程风格，可读性和维护性都更强。

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