Go iter.Pull 怎么把推送序列改成按需读取

Go 1.23 引入了标准迭代器。iter.Seq[V] 是推送式接口：迭代器主动调用 yield；当业务需要手动取一个、比较两个序列，或在读到目标值后立即停止时，可以用 iter.Pull 转成按需读取的 next 函数。最稳妥的写法是拿到 next, stop 后马上 defer stop()，然后依据布尔值判断是否还有数据。

一、先分清 push 和 pull 的调用方向

iter.Seq[V] 的形态是 func(yield func(V) bool)。调用方把处理函数交给序列，序列再把每个值推给它；yield 返回 false 时，序列应停止继续产生数据。这种形式适合普通的 for v := range seq。

但如果调用方要把两个序列一项项对齐，或者只想检查前几项，控制权应该在调用方手里。iter.Pull(seq) 返回 next func() (V, bool) 与 stop func()，前者拉取一个值，后者结束尚未读完的序列。

二、用 iter.Pull 改成按需读取

下面的示例把一个推送序列转换为拉取接口，并在读到第二个元素时提前退出。代码中的序列故意保留了一个清理动作，用来说明 stop 的责任。

package main

import (
    "fmt"
    "iter"
)

func records() iter.Seq[string] {
    return func(yield func(string) bool) {
        defer fmt.Println("source cleanup") // 模拟底层资源回收
        for _, item := range []string{"a", "b", "c"} {
            if !yield(item) { // 调用方不再需要时立即停止推送
                return
            }
        }
    }
}

func main() {
    next, stop := iter.Pull(records())
    defer stop() // 提前 return 或 break 时仍要结束迭代

    for {
        value, ok := next()
        if !ok { // false 表示已经到达序列尾部
            break
        }
        fmt.Println(value)
        if value == "b" { // 只消费到目标值，剩余值不再读取
            break
        }
    }
}

这里 next 不会一次性把序列收集成切片，因此适合逐个消费或尽早停止。stop 会让底层序列的 yield 在后续被调用时返回停止信号，给序列留下完成清理的机会。

三、stop 为什么要紧跟 defer

读到 ok == false 后，序列已经结束，再调用 stop 也是合法的；实际代码不必为“读完”和“提前退出”写两套清理分支。创建 pull 迭代器后立刻 defer，能覆盖错误返回、条件 break 和未来新增的 return。

不要把 next 发给多个 goroutine 并发调用。iter.Pull 的接口约束是同一个 next 或 stop 不能被多个 goroutine 同时调用；如果业务要并发处理，应由当前消费者读取后把普通值发送给 worker，而不是并发拉取同一个迭代器。

四、用两个 next 判断序列是否相等

按需读取最典型的场景是同步比较两个 iter.Seq。两个序列都要注册清理函数；每轮分别读取，任一序列先结束，只有两边同时结束才算长度一致。

func same[E comparable](left, right iter.Seq[E]) bool {
    nextLeft, stopLeft := iter.Pull(left)
    defer stopLeft() // 左序列提前结束时释放它
    nextRight, stopRight := iter.Pull(right)
    defer stopRight() // 右序列也必须独立释放

    for {
        lv, lok := nextLeft()
        rv, rok := nextRight()
        if !lok || !rok {
            return lok == rok // 只有同时结束，长度才相同
        }
        if lv != rv {
            return false // 值不同即可停止后续读取
        }
    }
}

判断顺序有两个要点：先比较有效标记，再比较值；当一侧结束时，不要把另一侧返回的零值误认为真实数据。若只比较前缀，则应在业务条件满足时直接返回，让两个 defer stop 负责收尾。

相关问题

iter.Pull 适合替代所有 for range 吗？

不适合。普通顺序遍历优先用 for range；只有需要调用方逐项控制、并行对齐或提前停止时，pull 形式才更直接。

next 返回 false 后还需要 stop 吗？

文档允许在序列结束后调用 stop。工程代码统一 defer stop 更简单，也能覆盖尚未读完的分支。

iter.Pull 能否跨 goroutine 共享？

不要让多个 goroutine 同时调用同一个 next 或 stop。由一个读取方串行拉取，再把值交给并发 worker 更安全。