如何在Golang中实现迭代器模式遍历数据集合

答案:通过定义Iterator接口和实现其方法的结构体,可在Go中实现迭代器模式。具体步骤为:先定义含HasNext和Next方法的接口;再为整数切片集合IntSlice创建迭代器IntSliceIterator,维护索引状态;HasNext判断是否越界,Next返回当前元素并移动指针,从而实现安全遍历。

在Golang中实现迭代器模式,可以通过接口和结构体组合的方式,将遍历逻辑与数据集合解耦。Go没有内置的迭代器关键字(如Java的Iterator),但凭借其简洁的接口设计和方法集机制,可以轻松实现灵活、可复用的迭代器。

定义迭代器接口

一个典型的迭代器包含两个核心方法:判断是否还有下一个元素,以及获取当前元素并移动指针。

注意: 接口定义应尽量简单通用。

以下是一个基础的迭代器接口:

type Iterator interface {
    HasNext() bool
    Next() interface{}
}

HasNext用于安全检查,避免越界;Next返回当前值并推进内部状态。

为具体集合实现迭代器

假设有一个简单的整数切片集合,我们可以为其定制迭代器。

先定义集合类型:

type IntSlice struct {
    data []int
}

再定义对应的迭代器结构体:

type IntSliceIterator struct {
    data  []int
    index int
}

func (it *IntSliceIterator) HasNext() bool {
    return it.index < len(it.data)
}

func (it *IntSliceIterator) Next() interface{} {
    if !it.HasNext() {
        return nil
    }
    val := it.data[it.index]
    it.index++
    return val
}

这样就将遍历过程封装在迭代器内部,外部无需关心索引管理。

在集合中提供创建迭代器的方法

让集合暴露一个方法来生成对应的迭代器实例。

func (s *IntSlice) Iterator() Iterator {
    return &IntSliceIterator{
        data:  s.data,
        index: 0,
    }
}

使用方式如下:

collection := &IntSlice{data: []int{1, 2, 3}}
it := collection.Iterator()

for it.HasNext() {
    fmt.Println(it.Next())
}

输出结果为1、2、3,实现了安全可控的遍历。

支持泛型提升复用性(Go 1.18+)

使用泛型可以让迭代器更通用,避免重复定义。

例如,定义一个泛型版本的迭代器接口:

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

然后实现一个适用于任意切片的泛型迭代器:

type SliceIterator[T any] struct {
    data  []T
    index int
}

func (it *SliceIterator[T]) HasNext() bool {
    return it.index < len(it.data)
}

func (it *SliceIterator[T]) Next() T {
    var zero T
    if !it.HasNext() {
        return zero
    }
    val := it.data[it.index]
    it.index++
    return val
}

这样就能用于字符串、结构体等不同类型的切片,提高代码复用率。

基本上就这些。通过接口抽象加结构体实现,Golang能干净地表达迭代器模式,尤其结合泛型后更加灵活。关键是把“遍历”这个行为独立出来,让集合专注数据存储,迭代器负责访问逻辑,符合单一职责原则。