Go iter.Pull2 怎样把双值迭代器改成拉取式读取

如果你已经有一个 iter.Seq2[K, V]，但当前代码更适合“每次取一项”，可以直接使用 iter.Pull2。它返回 next 和 stop：前者每次给出键、值和 ok，后者负责在提前结束时让迭代器收尾。

官方资料：https://pkg.go.dev/iter

最小写法是 next, stop := iter.Pull2(seq)，紧接着 defer stop()，再循环调用 next()。读到 ok == false 时结束；如果中途因为命中条件而返回，也不要省略 stop。

先看 Seq2 和 Pull2 返回值

iter.Seq2[K, V] 是接收 yield func(K, V) bool 的函数，常见语义是键值对或索引值对。它属于“推送”形式：迭代器主动把数据交给回调。Pull2 则把同一序列转换成“拉取”形式，返回的 next 签名为 func() (K, V, bool)。

这里的 ok 不是业务字段，而是“当前这次拉取是否拿到有效键值对”的结束标志。序列耗尽后，next 返回键和值的零值以及 false。

最小配方：按需拉取一个键值对

下面用一个返回键值对的迭代器演示基本写法。代码只关心拉取接口，不需要知道迭代器内部是 map、切片还是自定义容器。

package main

import (
	"fmt"
	"iter"
)

func pairs() iter.Seq2[string, int] {
	return func(yield func(string, int) bool) {
		// yield 返回 false 时，生产方应立即停止继续提供数据。
		for _, p := range []struct {
			key string
			value int
		}{{"go", 1}, {"iter", 2}} {
			if !yield(p.key, p.value) {
				return
			}
		}
	}
}

func main() {
	next, stop := iter.Pull2(pairs())
	defer stop() // 无论正常结束还是提前 return，都负责收尾。

	for {
		key, value, ok := next()
		if !ok {
			break // ok 为 false 表示序列已经耗尽。
		}
		fmt.Println(key, value)
	}
}

这个配方有三个关键点：Pull2 的输入必须是 Seq2；每次调用 next 都要同时接收键、值和 ok；循环退出后仍然保留 stop，让调用约定清楚可见。

提前结束时为什么必须 stop

如果读到目标键后就返回，或者只需要前几项，调用方可能还没有把序列消费到 ok == false。此时应让 stop 参与清理，尤其是迭代器内部可能持有需要结束的执行上下文时。

func findValue(seq iter.Seq2[string, int], wanted string) (int, bool) {
	next, stop := iter.Pull2(seq)
	defer stop() // 提前命中或异常返回时，统一通知迭代器结束。

	for {
		key, value, ok := next()
		if !ok {
			return 0, false // 读完仍未命中，返回值零值和未找到。
		}
		if key == wanted {
			return value, true // 这里会提前返回，但 defer 仍会执行。
		}
	}
}

官方文档允许在序列耗尽后调用 stop，也允许多次调用，因此把它写成 defer stop() 通常比在多个分支手动补调用更稳。另一方面，next 和 stop 不能被多个 goroutine 同时调用；需要并发消费时，应在单独的拥有者 goroutine 内封装读取协议。

Pull2、range 与普通循环怎么选

不要把 ok == false 当成某个键值对里的业务布尔字段，也不要只在“看起来会提前返回”的分支里调用 stop。只要使用 Pull2，默认的稳妥模板就是“创建后立即 defer stop，循环中以 ok 判断结束”。

常见问题

Pull2 读完后还需要调用 stop 吗？

严格来说，next 已经返回 false 后可以不再依赖 stop 完成结束，但统一保留 defer stop() 更容易覆盖提前退出和未来修改。

Seq2 和 Pull2 的键值顺序会改变吗？

不会。next 按序返回原迭代器提供的键和值；它只改变消费方式，不替你排序，也不改变 map 迭代本身的不确定顺序。