当前位置:首页 > 文章列表 > Golang > Go教程 > Go 迭代器实战:range over func 别写成炫技,先把 yield 用明白

Go 迭代器实战:range over func 别写成炫技,先把 yield 用明白

来源:Go Blog 2026-06-01 21:51:46 0浏览 收藏

Go 1.23 支持 range over func 之后,我见过两种极端:一种是完全不用,觉得它像别的语言硬塞进来的语法;另一种是到处改成迭代器,连一个普通 slice 遍历都包装一层。我的看法比较朴素:迭代器是给库 API 和惰性遍历用的,不是给业务代码增加仪式感的。

这篇不讲概念堆叠,直接按我做代码 review 的方式讲:iter.Seq 怎么写、yield 返回值为什么重要、什么时候该用 Seq2,以及哪些场景别硬上。

Go iter.Seq 思维导图
思维导图:range 函数、yield 返回、Seq2 键值、惰性遍历、提前停止和兼容边界。

先看最小写法

iter.Seq[T] 的本质是一个函数:它接收 yield func(T) bool。生产者每产生一个值,就调用一次 yield;如果 yield 返回 false,说明消费者停止了,生产者也必须停。

func Numbers(n int) iter.Seq[int] {
    return func(yield func(int) bool) {
        for i := 0; i < n; i++ {
            if !yield(i) {
                return
            }
        }
    }
}

for v := range Numbers(3) {
    fmt.Println(v)
}

这里最容易被忽略的是 if !yield(i) { return }。如果调用方在 for rangebreakyield 会返回 false。你不处理它,就可能继续做没必要的扫描、计算或者 I/O。

它解决的是“遍历协议”问题

以前我们给库暴露遍历能力,常见选择有三种:返回 slice、传 callback、或者开 channel。返回 slice 简单,但可能一次性分配很多内存;callback 能惰性,但写起来不像 Go 的遍历;channel 可以流式,但多了 goroutine、关闭和阻塞问题。

range over func 给了第四种选择:调用方仍然用熟悉的 for range,库内部仍然可以惰性地产生数据。对标准库、集合库、解析器、扫描器这类 API 来说,这是很舒服的补位。

Go range over func 迭代器流程图
流程图:定义 Seq,调用 yield,for range 消费,break 时停止,必要时再 Collect。

什么时候用 Seq2

如果你的遍历天然有两个值,比如 map 的 key/value、树节点的路径和值、分页游标和记录,就可以用 iter.Seq2[K, V]。它的 yield 形态是 func(K, V) bool

func Pairs(m map[string]int) iter.Seq2[string, int] {
    return func(yield func(string, int) bool) {
        for k, v := range m {
            if !yield(k, v) {
                return
            }
        }
    }
}

for k, v := range Pairs(scores) {
    fmt.Println(k, v)
}

但我不会为了“看起来统一”把所有东西都变成 Seq2。业务代码里如果一个 map 直接 range 就够清楚,那就直接 range。抽象不是越多越专业,读代码的人少绕一圈也很重要。

别用 channel 冒充迭代器

不少老代码会用 channel 暴露遍历结果,看起来也能 range,但代价并不小。你要考虑 goroutine 谁退出、channel 谁关闭、消费者提前 break 后生产者会不会卡住。

如果只是同步地产生一串值,iter.Seq 更直接。只有当数据天然来自异步事件、跨 goroutine 生产,或者需要并发管道时,channel 才更合适。

Go iter.Seq 代码案例图
代码案例:yield=false 要退出,别用 channel 冒充普通迭代器,库 API 更适合这个抽象。

Collect 是工具,不是默认动作

Go 1.23 之后,slicesmaps 包也围绕迭代器补了不少工具,比如把序列收集成 slice。它很方便,但别一上来就 Collect。

如果你最终还是一次性收集所有元素,再马上遍历一遍,那要问一句:直接返回 slice 会不会更简单?迭代器的价值通常在于惰性、组合、提前停止和隐藏数据源,而不是把所有代码都包装得更抽象。

我在项目里的使用边界

  • 库 API 可以优先考虑 iter.Seq,业务内部普通循环不用硬改。
  • 生产者必须尊重 yield 返回值,false 就停止。
  • 有 key/value 语义时用 Seq2,不要滥用二元值。
  • 同步遍历用迭代器,异步生产和并发管道再考虑 channel。
  • 需要兼容旧 Go 版本的公共库,要明确最低 Go 版本。

一个真实落地点:分页扫描

我比较喜欢把迭代器用在分页扫描这种场景。调用方不关心你背后是一页一页查数据库、读文件,还是扫远端 API;它只想按记录遍历,并且能在找到目标后提前停。

func ScanUsers(ctx context.Context, repo *Repo) iter.Seq[*User] {
    return func(yield func(*User) bool) {
        cursor := ""
        for {
            users, next, err := repo.Page(ctx, cursor)
            if err != nil || len(users) == 0 {
                return
            }
            for _, u := range users {
                if !yield(u) {
                    return
                }
            }
            cursor = next
        }
    }
}

这里有一个工程取舍:iter.Seq 自身不携带 error。需要错误语义时,可以把 error 放到元素里、使用额外回调、或者保留显式方法返回。不要为了套迭代器,把错误处理写得像谜语。

我的建议

range over func 是 Go 近几年比较有意思的语法补齐。它让“自定义集合也能被自然 range”这件事变得顺手。但它越顺手,越要克制。

我会把它放在库边界、集合封装、解析扫描、分页读取这些位置,而不是把业务代码里的每个 for 循环都改掉。好抽象的标准不是新,而是让调用方更少知道内部细节,同时还能保持 Go 代码那种一眼能读懂的味道。

参考资料:Go Blog《Range Over Function Types》、Go 1.23 Release Notes、标准库 iter/slices/maps 文档。

版本声明
本文转载于:Go Blog 如有侵犯,请联系study_golang@163.com删除
Go weak.Pointer 实战:缓存别越跑越胖,先搞懂弱引用和 AddCleanupGo weak.Pointer 实战:缓存别越跑越胖,先搞懂弱引用和 AddCleanup
上一篇
Go weak.Pointer 实战:缓存别越跑越胖,先搞懂弱引用和 AddCleanup
Go map 新实现实战:Swiss Tables 变快了,但别急着改业务代码
下一篇
Go map 新实现实战:Swiss Tables 变快了,但别急着改业务代码
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    516次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    500次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    485次学习
查看更多
AI推荐
  • ChatExcel酷表:告别Excel难题,北大团队AI助手助您轻松处理数据
    ChatExcel酷表
    ChatExcel酷表是由北京大学团队打造的Excel聊天机器人,用自然语言操控表格,简化数据处理,告别繁琐操作,提升工作效率!适用于学生、上班族及政府人员。
    5893次使用
  • Any绘本:开源免费AI绘本创作工具深度解析
    Any绘本
    探索Any绘本(anypicturebook.com/zh),一款开源免费的AI绘本创作工具,基于Google Gemini与Flux AI模型,让您轻松创作个性化绘本。适用于家庭、教育、创作等多种场景,零门槛,高自由度,技术透明,本地可控。
    6327次使用
  • 可赞AI:AI驱动办公可视化智能工具,一键高效生成文档图表脑图
    可赞AI
    可赞AI,AI驱动的办公可视化智能工具,助您轻松实现文本与可视化元素高效转化。无论是智能文档生成、多格式文本解析,还是一键生成专业图表、脑图、知识卡片,可赞AI都能让信息处理更清晰高效。覆盖数据汇报、会议纪要、内容营销等全场景,大幅提升办公效率,降低专业门槛,是您提升工作效率的得力助手。
    6138次使用
  • 星月写作:AI网文创作神器,助力爆款小说速成
    星月写作
    星月写作是国内首款聚焦中文网络小说创作的AI辅助工具,解决网文作者从构思到变现的全流程痛点。AI扫榜、专属模板、全链路适配,助力新人快速上手,资深作者效率倍增。
    8108次使用
  • MagicLight.ai:叙事驱动AI动画视频创作平台 | 高效生成专业级故事动画
    MagicLight
    MagicLight.ai是全球首款叙事驱动型AI动画视频创作平台,专注于解决从故事想法到完整动画的全流程痛点。它通过自研AI模型,保障角色、风格、场景高度一致性,让零动画经验者也能高效产出专业级叙事内容。广泛适用于独立创作者、动画工作室、教育机构及企业营销,助您轻松实现创意落地与商业化。
    6594次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码