Go 怎样把迭代器转换为拉取式接口

如果一个迭代器适合 for range，但业务代码需要“拿一个、判断一个、必要时立刻停止”，可以用 iter.Pull 把推送式的 iter.Seq 转成拉取式接口。它返回 next 和 stop 两个函数：前者一次取一个值，后者负责在没有读完时结束后台迭代。

这篇只围绕一个小任务展开：从一组订单号中跳过无效值，按需读取前两个有效项，并在提前结束时正确释放迭代器。核心习惯只有一句：拿到 next 后，除非已经读到 ok == false，否则就用 defer stop() 兜底。

先看清 Seq 和 next 的方向

iter.Seq[V] 的形状是 func(yield func(V) bool)。迭代器主动调用 yield 把值推给调用方；调用方只能通过返回 false 请求停止。iter.Pull 在两者之间加了一层适配，把同一序列变成 next() (V, bool)。

这里的 bool 不是业务字段，而是“当前值是否有效”。序列结束后，next 返回元素类型的零值和 false。因此不要把零值本身当成结束标志，应该始终先判断 ok。

用一个过滤序列做成按需读取

下面的示例模拟分页读取：生产者逐个扫描订单号，只把符合条件的值交给 yield；消费者每次调用 next，最多拿两个值就结束。

package main

import (
	"fmt"
	"iter"
)

// validOrderIDs 只推送正数订单号，false 表示消费者不再需要后续值。
func validOrderIDs(ids []int) iter.Seq[int] {
	return func(yield func(int) bool) {
		for _, id := range ids {
			if id 

这个例子里，next 只在消费者真正需要时推进生产者。读到两个有效订单后，函数直接返回，但原始序列可能还有 35 没有推送出去；此时 defer stop() 会让适配层结束这次迭代。

正常读完和提前停止要分开理解

有两条结束路径。第一条是 next 返回 ok == false，说明序列已经完成；第二条是业务已经得到答案，主动返回或跳出循环，但序列还没有完成。第二条必须调用 stop，否则生产者可能仍停在等待下一次交付的位置。

最稳妥的写法是拿到函数后立刻登记清理：

next, stop := iter.Pull(seq)
defer stop() // stop 可重复调用，适合做统一兜底

for {
	value, ok := next()
	if !ok {
		break // 正常结束，next 已经报告序列耗尽。
	}
	if shouldStop(value) {
		break // 提前结束，defer stop 仍会执行。
	}
	consume(value)
}

官方文档允许在序列结束后再次调用 next，也允许多次调用 stop，但这不代表应该用重复调用代替清晰的生命周期管理。把 stop 交给 defer，能覆盖循环退出、提前返回和异常路径。

Pull2 与并发边界不能混用

需要同时读取键和值时使用 iter.Pull2，返回的 next 形状变为 (K, V, bool)。它和 Pull 的结束规则一致：false 表示没有下一对，提前结束仍然要 stop。

next, stop := iter.Pull2(seq2)
defer stop() // 双值迭代器也要在消费者退出时收尾。

key, value, ok := next()
if !ok {
	return // key 和 value 是零值，不能当成有效记录使用。
}
fmt.Println(key, value)

next 和 stop 不能被多个 goroutine 同时调用。若要并发处理，应先由一个 goroutine 串行拉取，再把已经取得的值发送给 worker；不要让多个 worker 直接共享同一个 next。

最后用四个问题检查实现

如果消费逻辑天然是从头到尾遍历，直接使用 for range seq 更简单；只有在需要按需拉取、遇到条件立即停止，或需要把迭代器接入已有的逐项读取流程时，iter.Pull 才更合适。

相关问题

问：next 返回零值时一定代表数据为空吗？答：不一定，必须以 ok 为准；有效序列也可能产生元素类型的零值。

问：stop 要调用几次？答：通常一次并用 defer 登记；官方接口允许重复调用，但不应依赖重复调用掩盖生命周期设计。

问：Pull 适合跨 goroutine 共享吗？答：不适合。让一个 goroutine 负责调用 next/stop，再把结果发送给其他 goroutine。