Go 1.22+ 新的 range func 迭代器怎么写业务代码？

2026-02-02 00:00:00 作者：冷漠man

Go 1.22 的 range 迭代器是 range 语句对函数返回迭代器的扩展，要求函数返回 func() (T, bool) 类型；编译器自动展开为循环，适用于懒加载、分页等场景。

Go 1.22 引入的 range 函数迭代器（即支持对函数返回的迭代器进行 range）不是新语法，而是语言对 range 语句的扩展：只要函数返回符合迭代器协议的类型（两个返回值：元素、布尔值），就能直接用 range 遍历。它让业务代码更简洁、更声明式，尤其适合懒加载、分页、流式处理等场景。

写一个符合 range 协议的迭代器函数

核心是返回一个函数，该函数每次调用返回 value, ok —— 类似 map 或 channel 的迭代行为。Go 编译器会自动识别并展开为循环。

例如，实现一个从数据库分批拉取用户 ID 的迭代器：

func UserIDsIterator(db *sql.DB, batchSize int) func() (int64, bool) {
    rows := make(chan int64, batchSize)
    go func() {
        defer close(rows)
        offset := 0
        for {
            var ids []int64
            err := db.Select(&ids, "SELECT id FROM users ORDER BY id LIMIT ? OFFSET ?", batchSize, offset)
            if err != nil || len(ids) == 0 {
                return
            }
            for _, id := range ids {
                rows <- id
            }
            offset += batchSize
        }
    }()
    return func() (int64, bool) {
        id, ok := <-rows
        return id, ok
    }
}

使用时直接 range：

for id := range UserIDsIterator(db, 100) {
    // 处理单个用户 ID
    processUser(id)
}

避免常见陷阱：状态与并发安全

迭代器函数内部需自行管理状态（如游标、缓冲、关闭逻辑）。若涉及 goroutine + channel，注意资源泄漏和 panic 传播。

不要在迭代器闭包中直接访问外部可变变量（如循环变量 i），应捕获副本
如果迭代器依赖外部资源（如 DB 连接、文件句柄），确保在迭代结束或出错时释放
不建议在迭代器函数里启动长期 goroutine；更适合“按需拉取”，比如每次调用生成下一批数据

业务常用模式：分页、过滤、转换

把通用逻辑封装成可组合的迭代器，提升复用性：

// 分页包装器
func Paginate[T any](fetcher func(offset, limit int) ([]T, error), limit int) func() (T, bool) {
    offset := 0
    var cache []T
    var cacheIdx int
    return func() (T, bool) {
        if cacheIdx >= len(cache) {
            cache, _ = fetcher(offset, limit)
            if len(cache) == 0 {
                var zero T
                return zero, false
            }
            cacheIdx = 0
            offset += limit
        }
        v := cache[cacheIdx]
        cacheIdx++
        return v, true
    }
}
// 使用：遍历所有订单（自动分页）
for order := range Paginate(db.FetchOrders, 50) {
sendNotification(order)
}