Go range over func 怎样让函数迭代器支持提前停止处理空输入并保留错误边界

批量读取记录时，调用方经常只需要前几项，或者遇到一条坏数据就要停止。直接把全部结果装进切片虽然容易写，却把“什么时候停”和“读取错误怎么交付”混在了一起。Go 1.23 起，for range 可以遍历函数迭代器；只要迭代器检查 yield 的返回值，就能把提前停止传回生产端。

官方地址：https://go.dev/doc/go1.23

这类问题的稳妥写法是：用 iter.Seq2[Record, error] 同时承载记录和终止错误，空输入直接返回，消费者通过 break 表达“不再需要”，生产端收到 yield=false 后立即收口。这样不会伪造一条空记录，也不会把读取错误藏在迭代器内部。

先把提前停止的协议说清楚

函数迭代器的形状不是任意函数，而是 func(yield func(V) bool) 或 func(yield func(K, V) bool)。迭代器每交付一项就调用一次 yield；调用方继续消费时，返回值为 true。一旦循环因为 break、返回或其他原因提前结束，合成的 yield 会返回 false，迭代器必须停止后续调用。

因此，问题不在于能不能写出 for range，而在于生产端是否把这个布尔值当成协议处理。忽略它会继续读取无用数据；在它已经返回 false 后再次调用 yield，则违反 range over func 的约定。

用 Seq2 同时表达记录和读取错误

函数迭代器本身没有返回值，读取过程中又可能出现错误。一个实用的边界是把正常记录放在第一个值，把错误放在第二个值；发生错误时交付一次“零值记录 + 错误”，随后结束。空切片不调用 yield，所以不会被误判成一条空记录。

package main

import (
	"errors"
	"fmt"
	"iter"
)

type Record struct {
	ID   int
	Name string
}

var errBadRecord = errors.New("bad record")

// Records 把记录和读取错误放在同一个 Seq2 中，failAt 为 -1 表示不注入错误。
func Records(rows []Record, failAt int) iter.Seq2[Record, error] {
	return func(yield func(Record, error) bool) {
		if len(rows) == 0 {
			return // 空输入没有记录，也没有错误，不调用 yield。
		}

		for i, row := range rows {
			if i == failAt {
				_ = yield(Record{}, fmt.Errorf("读取第 %d 条记录失败: %w", i, errBadRecord))
				return // 错误只交付一次，随后结束迭代器。
			}
			if !yield(row, nil) {
				return // 消费者 break 后，停止读取剩余记录。
			}
		}
	}
}

这里的错误项是一个明确的终止边界，而不是把错误打印后继续产出数据。调用方可以决定是立即返回、记录上下文，还是把错误转换成上层业务错误；迭代器只负责交付一次并停止。

让消费者决定何时结束

消费者要先判断错误，再处理正常记录。下面的例子在达到 limit 后退出；如果处理函数失败，也可以用同样的 break 路径停止。若在错误项到达前就主动停止，那么后续错误不会被读取到，这正是提前停止的语义，调用方应根据业务需要选择停止位置。

// consume 只消费需要的记录，并把迭代器交付的错误原样带回调用方。
func consume(rows []Record, failAt, limit int) error {
	for row, err := range Records(rows, failAt) {
		if err != nil {
			return fmt.Errorf("消费记录失败: %w", err) // 保留原始错误，补充调用上下文。
		}

		fmt.Println(row.ID, row.Name) // 这里代表实际的单项处理逻辑。
		if row.ID >= limit {
			break // range 会让底层 yield 返回 false，生产端随即返回。
		}
	}
	return nil // 空输入和自然遍历结束都会走到这里。
}

这段代码有三个容易忽略的结果：空输入不会进入循环；自然遍历完所有记录时返回 nil；达到上限后退出时，迭代器不会再读取后面的元素。如果 failAt 指向尚未消费的位置，调用方提前 break 就不会看到那条错误，这不是错误丢失，而是调用方明确选择了更早的终点。

空输入、错误和并发修改分别怎么判断

不要把 iter.Seq2 当成并发安全层。示例捕获的是 rows 的切片描述符；遍历期间如果另一个 goroutine 修改底层数据，仍然需要快照、锁或其他同步方案。若迭代器要打开文件、连接或启动后台生产者，也要在 yield=false 分支之前设计好资源释放。

还要注意接口选择：顺序只读、可能提前停止时，函数迭代器很合适；需要排序、随机索引、重复遍历或把结果交给多个消费者时，返回切片或显式快照通常更直白。它消除的是中间容器和无用读取，不自动保证零分配或更高吞吐。

相关问题

为什么空输入不应该 yield 一个零值？零值记录可能被误认为真实数据；零次调用才能保持“没有元素”和“元素值为空”的区别。

提前 break 后还能拿到后面的错误吗？不能。break 表示调用方不再需要后续数据，若错误在后续位置，就应由调用方承担未继续读取的语义。

可以只返回 iter.Seq 而不返回错误吗？如果读取过程绝不会失败可以；只要存在可观察的读取错误，就应使用 Seq2、外部错误状态或其他明确的错误通道，不要静默吞掉。