如何用Java 8 Stream实现两层循环并从内层返回外层对象？

当然可以用Stream优雅地实现这个需求，你的两个方案都差了关键一步——核心是要保留R对象和它内部C元素的关联关系，不能像方案二那样直接flatMap丢了R的引用，也别用方案一的forEach这种没法提前终止流的方式。

先回顾下你原代码的核心逻辑：找到第一个包含符合条件C（C的V属性是V类型）的R对象，然后返回这个R的Xyz；如果遍历完所有都找不到，就返回null。

你的两个方案问题在哪？ #

• 方案一：用forEach遍历R对象，虽然能找到对应的V，但forEach无法提前终止流——哪怕已经找到符合条件的R，它还是会把所有R都遍历完。而且给外部变量xyz赋值的写法，完全违背了Stream的函数式设计思想，不够优雅也容易出问题。
• 方案二：flatMap把所有R的C元素拉平成一个流，确实能找到符合条件的V，但此时已经丢失了对应的R对象，自然拿不到我们需要的Xyz属性，这是最致命的问题。

正确的Stream实现方式 #

我们需要给每个R对象做“是否包含符合条件C”的判断，找到第一个满足条件的R后直接返回它的Xyz，代码如下：

public static Xyz getXyz(P p) {
    // 先处理空值情况，和原逻辑一致
    if (p == null || CollectionUtils.isEmpty(p.getB())) {
        return null;
    }

    return p.getB().stream()
            // 过滤出R类型的对象并强制转换
            .filter(b -> b instanceof R)
            .map(b -> (R) b)
            // 筛选出内部包含符合条件C的R：C的V属性是V类型
            .filter(r -> r.getB().stream()
                    .filter(cObj -> cObj instanceof C)
                    .map(cObj -> (C) cObj)
                    .map(C::getV)
                    // anyMatch是短路操作，找到第一个符合条件的就停止遍历
                    .anyMatch(v -> v instanceof V))
            // 取第一个符合条件的R，获取它的Xyz
            .map(R::getXyz)
            // 找不到就返回null，完美匹配原逻辑
            .findFirst()
            .orElse(null);
}

代码解释 #

1. 空值处理：合并了原代码的两次空判断，更简洁。
2. R对象过滤与转换：先把p的B列表转换成流，筛选出R类型的对象并转成R。
3. 内部C的判断：用anyMatch检查每个R的C列表——只要存在一个C的V是V类型，这个R就符合条件。anyMatch是短路操作，和原循环一样，找到第一个符合条件的元素就停止遍历，效率完全一致。
4. 结果提取：找到第一个符合条件的R后，直接映射到它的Xyz属性，最后用findFirst().orElse(null)处理找不到的情况，完全复现原代码的逻辑。

如果你的项目已经升级到Java 16+，还可以用instanceof的模式匹配简化类型转换，让代码更清爽，但上面的写法完全兼容Java 8。

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