Go iter.Pull2 怎么安全消费双值迭代器：停止时机、资源释放与错误传递

把一个 iter.Seq2[K, V] 接到需要“每次只取一条”的处理逻辑时，iter.Pull2 会返回 next 和 stop 两个函数。真正容易出错的地方不在泛型，而在消费提前结束后有没有停掉生产端，以及把结束标志和零值混在了一起。

先看一个键值迭代场景：为什么需要 Pull2

假设配置加载器按顺序产出键和值，普通的 range 很顺手；但当调用方要在命中某个键后立刻停止，或者要把每一条记录交给一个有状态的校验器，逐项拉取会更直观。Pull2 正是把“生产者主动推送”转换成“消费者主动取下一项”。

package main

import (
	"fmt"
	"iter"
)

func settings() iter.Seq2[string, string] {
	return func(yield func(string, string) bool) {
		for _, item := range [][2]string{{"region", "cn"}, {"timeout", "3s"}, {"retries", "2"}} {
			if !yield(item[0], item[1]) {
				return
			}
		}
	}
}

func main() {
	next, stop := iter.Pull2(settings())
	defer stop()

	for {
		key, value, ok := next()
		if !ok {
			break
		}
		fmt.Println(key, value)
	}
}

这段代码的关键不是打印结果，而是 defer stop() 的位置：拿到 next 后立刻注册收尾动作，后面的校验失败、提前返回或异常路径都不会漏掉停止调用。

双值生产端经过 Pull2 后，由消费者逐项读取键和值；最后用 ok=false 识别自然结束。

next 的三个返回值应该怎样判断

next 的签名是 func() (K, V, bool)。第三个返回值表示这一次是否真的取到了一对数据。迭代结束后，键和值会回到各自类型的零值，所以只看键或值是不可靠的：空字符串、0 或 nil 都可能是合法数据。

for {
	key, value, ok := next()
	if !ok {
		break
	}
	if key == "timeout" {
		fmt.Println("found:", value)
		break
	}
}

这里的 break 是“提前停止”，并不等于生产端已经自然跑完。只要此前没有拿到过 ok=false，就应让函数返回时执行 stop。

提前命中时，stop 负责把生产端收回来

一个常见任务是找到第一条符合条件的配置，然后停止读取后面的项目。安全写法是把停止动作放在同一函数的资源边界上，而不是只在某个 break 后手写一次。

func findTimeout(seq iter.Seq2[string, string]) (string, bool) {
	next, stop := iter.Pull2(seq)
	defer stop()

	for {
		key, value, ok := next()
		if !ok {
			return "", false
		}
		if key == "timeout" {
			return value, true
		}
	}
}

stop 会让尚未完成的迭代停止；官方文档建议通常使用 defer stop()。它可重复调用，也可以在 next 已经返回 false 后调用，因此自然结束和提前返回可以共用一条清理路径。

提前命中后，消费者不再取新数据；stop 把仍未完成的生产端收束到结束状态。

不要把 Pull2 当成并发安全队列

同一组 next 和 stop 不能被多个协程同时调用。若多个协程都想消费，应在外层明确分工：要么让一个协程独占拉取，再把结果发送给其他协程；要么为每个消费者构造独立的序列。直接给 next 加一个看似方便的并发调用，很容易把停止时机和生产端状态搅在一起。

另外，迭代器函数如果在调用 next 或 stop 的过程中 panic，这个 panic 会传回对应的调用点。不要把它误判成 ok=false；结束标志只表示序列正常结束。

用一个小测试核对三种边界

建议至少验证自然结束、提前结束和空序列。测试重点不是断言零值，而是断言是否读到数据、生产端是否看到了停止信号。

func pairs(yield func(string, int) bool) {
	defer func() { fmt.Println("producer stopped") }()
	for _, n := range []int{1, 2, 3} {
		if !yield(fmt.Sprintf("n%d", n), n) {
			return
		}
	}
}

func consumeFirst() {
	next, stop := iter.Pull2(pairs)
	defer stop()
	key, value, ok := next()
	fmt.Println(key, value, ok)
}

如果测试只覆盖完整循环，往往看不出漏掉 stop 的问题。把“读第一项后直接返回”也加进去，才能确认提前路径仍然收尾。

常见问题

next 返回 false 后还需要调用 stop 吗？

可以调用。文档说明在自然结束后调用 stop 是有效的，通常保留 defer stop() 能让所有返回路径保持一致。

为什么不能用零值判断迭代结束？

因为合法数据本身可能就是零值。ok 才是本次是否取到键值对的判断依据。

能否把 next 分给多个 goroutine？

不能直接这样做。同一组 next/stop 不支持多个协程同时调用，需要由一个协程独占拉取并转发结果，或为不同消费者创建独立迭代器。

Pull2 适合所有 Seq2 场景吗？

不一定。如果顺序遍历已经清楚，直接使用 range 更简单；只有在逐项控制、提前停止或需要把拉取动作嵌入现有流程时，Pull2 才更有价值。

把判断顺序固定下来

使用 iter.Pull2 时，可以把流程固定成三步：拿到 next/stop 后立即 defer stop()，每次先判断 ok 再读键和值，提前返回时不额外复制一套清理逻辑。这样既能保留双值迭代器的惰性，也能把资源释放和并发边界留在代码最容易检查的位置。