当前位置:首页 > 文章列表 > 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推荐
  • ljg-skills -
    ljg-skills
    ljg-skills 是李继刚开源的 AI 技能与提示词集合,面向大模型使用者整理了一批可复用的 prompt、角色设定和任务技能模板,适合用于学习提示词设计、搭建个人 AI 工作流和沉淀团队常用智能体能力。
    1576次使用
  • MELO音乐 - AI 音乐生成平台,支持多模态创作能力
    MELO音乐
    MELO音乐是一站式AI视频与音乐制作助手,对标suno, udio的高品质体验。提供伴奏生成、原创写词、无损导出、哼唱识曲、混音变声等全套音频与短视频编辑工具。无论是流行Kpop、电音说唱、民谣古风、摇滚儿歌还是商用轻音乐,MELO为你免费谱曲,轻松做同款!
    1513次使用
  • UniScribe - AI 免费在线音视频转文字平台
    UniScribe
    UniScribe 是一款 AI 音视频转文字与内容整理工具,支持上传音频、视频文件或粘贴 YouTube 链接,自动生成转写文本、摘要、思维导图和关键问题,并支持多格式导出,适合会议记录、课程学习、访谈整理和内容创作复盘。
    1453次使用
  • 剧云 - 免费 AI 智能中文剧本创作平台
    剧云
    剧云是专业中文剧本创作平台,安全稳定运行十余年,集成AI编剧、剧本医生审核、人物小传、剧情关系图、大纲编写、多人协作、Word导入导出、版权管控功能,数据安全防护,轻松高效创作剧本。
    1653次使用
  • 万象有声 - AI 一站式有声内容创作平台
    万象有声
    万象有声,一个专为有声创作者打造的新一代智能有声内容创作平台。平台提供专业的智能拆章、智能画本编辑、AI配音、AI生成音效、后期制作、智能对轨、智能审听等有声创作全流程工具,可以帮助创作者高效、低成本创作出引人入胜的有声作品。立即体验,让有声书制作更简单!
    1639次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码