通八洲科技

如何使用Golang实现迭代器模式_Golang迭代器模式集合遍历方法

日期:2025-12-30 00:00 / 作者:P粉602998670
Go没有原生迭代器接口,因其哲学强调显式优于隐式、简单优于抽象;range已满足多数遍历需求,而函数值、接口和泛型三种方式可按场景选择实现轻量、可组合或类型安全的迭代器。

为什么 Go 语言没有原生迭代器接口

Go 不像 Java 或 C# 那样内置 Iterator 接口或 for...of 语法糖,核心原因是 Go 哲学强调显式优于隐式、简单优于抽象。标准库中 range 已覆盖绝大多数集合遍历需求,强行套用传统迭代器模式反而增加冗余类型和内存分配。

但当你需要:延迟计算(如读取大文件行)、状态保持(如树的中序遍历)、封装内部结构(如跳表、B+ 树)、或统一多数据源访问(DB cursor + slice + channel)时,手写迭代器就变得必要。

用函数值实现轻量级迭代器(最常用)

Go 中最自然、零开销的方式是返回一个闭包函数,每次调用返回下一个元素和是否结束的标志。它避免了接口定义、结构体字段和指针解引用开销。

func IntSliceIter(nums []int) func() (int, bool) {
    i := 0
    return func() (int, bool) {
        if i >= len(nums) {
            return 0, false
        }
        val := nums[i]
        i++
        return val, true
    }
}

// 使用 next := IntSliceIter([]int{1, 2, 3}) for { if val, ok := next(); ok { fmt.Println(val) } else { break } }

用接口定义可组合的迭代器(适合复杂场景)

当需要链式操作(如 FilterMapTake)或多态统一处理时,定义接口更清晰。但要注意:接口值包含动态分发开销,且每次调用都涉及一次间接跳转。

type Iterator[T any] interface {
    Next() (T, bool)
}

func Filter[T any](it Iterator[T], f func(T) bool) Iterator[T] { return &filterIter[T]{it: it, f: f} }

type filterIter[T any] struct { it Iterator[T] f func(T) bool cur T ok bool }

func (f *filterIter[T]) Next() (T, bool) { for { if val, ok := f.it.Next(); ok { if f.f(val) { return val, true } } else { return f.cur, false } } }

泛型 + 迭代器工厂函数(Go 1.18+ 推荐方式)

Go 泛型让迭代器能复用逻辑又保持类型安全。比起为每种类型写一遍函数,用泛型一次定义,编译期生成特化版本,既无接口开销,又避免重复代码。

func MapIter[T, U any](it func() (T, bool), f func(T) U) func() (U, bool) {
    return func() (U, bool) {
        if val, ok := it(); ok {
            return f(val), true
        }
        var zero U
        return zero, false
    }
}

// 使用 nums := []int{1, 2, 3} iter := MapIter(IntSliceIter(nums), func(x int) string { return fmt.Sprintf("v%d", x) }) for { if s, ok := iter(); ok { fmt.Println(s) // "v1", "v2", "v3" } else { break } }

真正难的不是写出一个能跑的迭代器,而是判断该不该用、用哪种形式。函数值适用于 90% 的场景;接口适合需要动态组合或统一调度的地方;泛型则是在两者之间找平衡——它要求你对类型关系有足够清晰的建模,否则容易陷入过度设计。