Java Stream中flatMap生成的Iterator为何首次hasNext会遍历首个完整子流

• 内置flatMap的消费粒度与预期存在差异：Java Stream默认的flatMap实现不支持跨层逐元素透传。在迭代器模式下，当触发首次元素拉取时，会把单个输入元素经flatMap展开后的全部输出元素一次性生成并写入内部缓冲队列，不会在拿到第一个有效输出元素后暂停流程。这是首次调用hasNext()就完整遍历首个子流的直接原因。
• iterator()的底层实现与预期存在差异：iterator()终端操作返回的迭代器，并非对管线操作的逐元素穿透，而是基于WrappingSpliterator的缓冲队列实现。当内部缓冲为空时，会从源头拉取一个输入元素，经过全链路处理（包括子流展开、后续filter/map转换），将该元素生成的所有输出一次性全部灌入缓冲，再把第一个元素返回给调用方。这套设计最初是为了兼容并行流下flatMap的分片处理逻辑，在顺序流+嵌套子流的场景下，会出现非预期的提前消费。

对照实际运行输出可以验证上述逻辑：首次hasNext()触发时，A2对应的子流中所有B、C对象已经全部生成并写入缓冲，因此才会在第一次打印after hasNext()之前，完成整段子流的对象创建。

调整逻辑的核心是绕开内置flatMap的缓冲机制，自行实现扁平化迭代器，实现每步拉取仅生成一个元素的效果：

1. 调整流的迭代边界：将父流的迭代范围提前到生成A对象的步骤，不把flatMap及后续操作纳入原流管线。
2. 手动维护两层迭代状态：父级A对象的迭代器、当前正在消费的B子流迭代器，以及单个C元素的缓存位。
3. 自定义迭代器逻辑：仅在缓存为空时才逐级向下拉取元素，每次只处理到拿到第一个符合过滤条件的B、转换为C并缓存后就暂停流程，不提前遍历剩余的子流元素。

示例实现代码：

AtomicInteger x = new AtomicInteger();
// 提前拆分迭代边界，拿到A对象的迭代器
Iterator<A> parentIter = Stream.generate(() -> null)
        .takeWhile(y -> x.incrementAndGet() < 5)
        .filter(y -> x.get() % 2 == 0)
        .map(n -> new A("A" + x.get()))
        .iterator();

Iterator<C> customIterator = new Iterator<>() {
    private Iterator<B> currentBSubIter = Collections.emptyIterator();
    private C cachedNext;

    @Override
    public boolean hasNext() {
        // 缓存已有有效元素，直接返回
        if (cachedNext != null) {
            return true;
        }
        while (true) {
            // 当前子流迭代器耗尽时，从父迭代器拉取下一个A
            if (!currentBSubIter.hasNext()) {
                if (!parentIter.hasNext()) {
                    return false;
                }
                A nextA = parentIter.next();
                currentBSubIter = nextA.getBStream().iterator();
            }
            // 逐元素拉取子流，直到拿到非null的B
            B nextB = currentBSubIter.next();
            if (nextB != null) {
                cachedNext = nextB.toC();
                return true;
            }
        }
    }

    @Override
    public C next() {
        if (!hasNext()) {
            throw new NoSuchElementException();
        }
        C result = cachedNext;
        cachedNext = null;
        return result;
    }
};

使用该自定义迭代器替换原流管线生成的迭代器，运行输出将完全匹配预期的执行时序，不会在首次调用hasNext()时提前遍历完整个子流。

内容的提问来源于stack exchange，提问作者cigno5.5