Go iter.Pull 怎样把迭代器转换成主动拉取接口

iter.Seq 是“迭代器主动调用 yield 推送值”的接口；如果调用方更适合自己控制“什么时候拿下一个值”，可以用 iter.Pull 把它转换成 next 和 stop 两个函数。next 每次返回一个值和一个 ok 标志，ok=false 表示序列结束；调用方提前停止时要调用 stop，最稳妥的写法是拿到函数后立即 defer stop()。

先把一个小需求说清楚

假设我们拿到的是一个按顺序产生数字的 iter.Seq[int]，但业务只想读取到第一个偶数，而不是让迭代器一次性把剩余值全部交给回调。这个需求更接近“调用一次，判断一次，再决定是否继续”，正是 Pull 接口适合的地方。

官方文档把普通迭代器称为 push-style：序列函数通过 yield 把值交给消费者；iter.Pull 则提供 pull-style 的访问方式。官方入口是 https://pkg.go.dev/iter#Pull，本文配图是静态结构说明图，不是运行截图。

用 iter.Pull 拿到 next 和 stop

先用 slices.Values 得到标准的 iter.Seq[int]，再交给 iter.Pull。注意 next 的返回值必须同时接收，不能只看数字本身，因为合法数据可能恰好等于类型零值。

package main

import (
    "fmt"
    "iter"
    "slices"
)

func firstEven(values []int) (int, bool) {
    seq := slices.Values(values)
    next, stop := iter.Pull(seq)
    defer stop() // 即使提前找到结果，也让迭代器结束并释放内部状态。

    for {
        value, ok := next()
        if !ok {
            return 0, false // 零值只在 ok=false 时表示“没有更多数据”。
        }
        if value%2 == 0 {
            return value, true // 找到目标后提前结束，defer 会调用 stop。
        }
    }
}

func main() {
    value, ok := firstEven([]int{3, 5, 8, 11})
    fmt.Println(value, ok) // 8 true
}

这里的关键不是把 for range 改写成更复杂的循环，而是把“下一项”和“是否还有下一项”作为一组结果处理。图1把 slices.Values、iter.Seq、iter.Pull、next 和 stop 的静态关系放在同一张图中，方便对照代码中的接口边界。

next 的 ok 到底表示什么

next() 返回的是 (V, bool)。当 ok=true 时，值才是当前序列中的有效元素；当 ok=false 时，返回的 V 是零值，代表序列已经结束。因为零值也可能是业务数据，所以不能写成“值等于 0 就结束”这样的判断。

如果调用方已经收到过 ok=false，继续调用 next 仍会得到零值和 false；调用 stop 后再调用 next 也会保持这个结束结果。但这不意味着可以省略正常的停止约定：只要还没有收到结束信号、调用方又不再需要后续元素，就应该调用 stop。

只拉取一部分时为什么必须 stop

最容易漏掉 stop 的场景是“找到第一个匹配项就 return”。对普通切片，这个遗漏可能暂时看不出影响；对一次性数据源或带清理逻辑的迭代器，未停止会让提供序列的函数继续等待或保持内部状态。把 stop 放在 defer 中，可以同时覆盖成功返回、没有匹配项和后续逻辑返回错误三条路径。

func takeUntilLimit(seq iter.Seq[string], limit int) []string {
    next, stop := iter.Pull(seq)
    defer stop() // 无论收满、提前返回还是异常路径，都结束这次拉取。

    result := make([]string, 0, limit)
    for len(result) 

图2强调的是消费边界：next 负责读取，ok 负责告诉调用方是否还有值，defer stop() 负责处理提前结束。它们是三个不同职责，不能用“循环自然退出”替代清理动作。

几个容易踩到的边界

• 不要把 Pull 当成并发队列。next 和 stop 不能被多个 goroutine 同时调用；需要并发消费时，应在外部设计明确的同步或分发协议。
• 不要跳过 stop。完整消费到 ok=false 后再调用 stop 是安全的，使用 defer stop() 可以把这个约定固定下来。
• 区分 Seq 和 Seq2。单值序列用 iter.Pull，键值或索引值对用 iter.Pull2，不要为了凑类型把第二个值塞进结构体。
• 版本要满足迭代器 API。标准 iter 包与函数类型 range 能力从 Go 1.23 时代开始进入标准库使用方式，老项目升级时先确认工具链版本。

相关问题

问：已经读到 ok=false，还要调用 stop 吗？
答：调用是安全的，而且通常建议统一保留 defer stop()，这样不会因为未来改成提前返回而遗漏清理。

问：iter.Pull 会自动把数据放进一个可并发读取的 channel 吗？
答：不要把它当作并发安全队列。文档明确要求 next 和 stop 不要被多个 goroutine 同时调用。

问：只想用 for range，还需要 Pull 吗？
答：如果可以直接让序列推送给回调，for range seq 更简单；只有当调用方需要逐次主动决定是否继续时，Pull 才更合适。