Go 怎样让函数支持 range 遍历

Go 1.23 起，for range 可以消费特定签名的函数。要让函数支持 range 遍历，核心不是给函数加一个特殊关键字，而是让它接收一个 yield 回调：每交付一个值就调用一次，回调返回 false 时立即停止。

最常用的类型是 iter.Seq[V]，它等价于 func(yield func(V) bool)；键值对则使用 iter.Seq2[K, V]。官方资料：https://go.dev/blog/range-functions，规范入口：https://go.dev/ref/spec#For_range。本文只讨论函数迭代器的签名、终止条件和资源边界。

用 iter.Seq 写出最小可用签名

package main

import (
    "fmt"
    "iter"
)

// Numbers 返回一个可被 range 消费的整数序列。
// 上限只决定后续产出范围，不会在构造时预先生成切片。
func Numbers(limit int) iter.Seq[int] {
    return func(yield func(int) bool) {
        for i := 0; i 

这里的关键是返回函数的形状，而不是函数名。Numbers(100) 先得到一个序列函数，range 再把内部循环体转换成符合约定的 yield 回调。预期只处理 0、1、2；这段代码用于说明语义，文中的输出是推导结果，不是运行截图。

如果要遍历键值对，使用 iter.Seq2[K, V]，其签名是 func(yield func(K, V) bool)。不要把任意返回值的普通函数直接放在 range 右侧；只有规范列出的迭代器函数形态才受到支持。

拆分构造函数、序列函数与消费者职责

Numbers 是构造函数，负责接收上限并把它捕获进闭包；返回的匿名函数是生产者，负责按照规则调用 yield；for 循环体是消费者，负责决定收到一个值后做什么。三者之间的唯一交接点就是 yield 的参数和布尔返回值。

这种写法适合把容器的内部结构藏起来。例如一个集合可以提供 All() iter.Seq[E]，调用方只看到“取出所有元素”的接口，不需要知道底层是切片、树还是哈希表。接口名称要说明遍历内容，参数要表达范围；不要为了看起来像迭代器而把已经完整构造好的大切片再包进闭包。

用 yield 的 false 实现提前停止

消费者正常完成一次循环体时，yield 可以继续返回 true；遇到 break 等结束当前 range 的控制转移时，生产者收到的继续许可会变成 false。生产者必须检查它，不能忽略返回值后继续调用 yield，也不应把 yield 保存到异步任务里稍后使用。

// FirstTwo 直接调用迭代器，只收集最前面的两个值。
func FirstTwo(seq iter.Seq[int]) []int {
    result := make([]int, 0, 2)
    seq(func(v int) bool {
        result = append(result, v)
        // 第二个值已经收到，返回 false 让生产者停止。
        return len(result) 

直接调用和使用 range 遵守的是同一个协议：回调返回 true 才允许继续交付。停止意味着“本次遍历不再需要后续值”，不代表已经交付的值会被撤销。若生产者在下一次 yield 前还有昂贵计算，应该把计算放在检查点之后，避免消费者已经停止却继续准备数据。

把清理放在真正拥有资源的函数里

简单的整数序列没有资源需要关闭。若迭代器内部打开文件、持有锁或创建临时对象，应在真正取得资源的那一层安排 defer；收到 false 后通过 return 离开该函数，清理才会沿正常返回路径执行。不要在构造函数中提前打开资源，却期待一个从未被 range 消费的序列自动释放它。

还要区分停止许可和全系统取消。yield 的 false 只能约束当前迭代器继续交付；如果生产者正阻塞在网络或磁盘操作上，range 的 break 不会替你打断那项操作。需要取消阻塞任务时，应另行设计 context 或底层接口，不能把 yield 的布尔值当成通用取消信号。

重复遍历、缓存和性能需要单独约定

本例的游标 i 定义在返回函数内部，因此同一个序列再次被遍历时会重新从 0 开始；它没有缓存上次结果，也不是只能消费一次。若把游标移到外层闭包，重复遍历就可能变成续读。iter.Seq 本身无法替你表达这些业务语义，接口文档需要明确说明。

惰性只表示工作可以随着消费发生，不自动意味着更快、可并发或零内存。如果调用方最终要收集全部结果，内存成本仍然存在；只有经常只取前缀，或剩余值生成成本明显时，函数迭代器才更有收益。小而固定的结果集继续返回切片，通常更容易理解。

相关问题

Go 版本低于 1.23 能直接 range 函数吗？

不能按本文的函数迭代器语法使用。需要升级工具链，或改用切片、channel 等旧接口。

yield 返回 false 后还能再调用吗？

不应再调用。false 就是当前消费者撤回继续接收的许可，生产者应结束本次序列函数。

iter.Seq 是否自带错误返回值？

不自带。若遍历过程可能出错，应把错误状态放入明确的接口设计中，例如返回结果对象或另设错误通道，不能把错误悄悄塞进 yield。

实现函数迭代器时，先固定 iter.Seq 或 iter.Seq2 签名，再把“交付一个值”和“检查是否继续”写在同一处；最后补清楚资源归属、重复遍历和取消边界，range 才会既方便又可控。