在Golang开发中，集合类型的遍历是高频操作，传统的遍历方式往往将遍历逻辑与集合内部结构绑定，当集合结构发生变化时，遍历逻辑也需要同步修改。迭代器模式通过抽象遍历逻辑，让集合和遍历操作解耦，既方便扩展新的遍历方式，也能让集合的实现更独立。
迭代器模式的核心概念
迭代器模式主要包含四个核心角色：
- 迭代器接口：定义遍历集合的统一方法，比如获取下一个元素、判断是否还有剩余元素等。
- 具体迭代器：实现迭代器接口，维护当前遍历的位置，完成具体的遍历逻辑。
- 集合接口：定义获取迭代器的方法，让外部可以通过统一方式获取对应集合的迭代器。
- 具体集合：实现集合接口，存储实际的元素，并且返回对应的具体迭代器实例。
Golang实现迭代器模式遍历集合的步骤
1. 定义迭代器接口
首先定义迭代器的通用接口，包含遍历需要的基础方法：
// Iterator 迭代器接口
type Iterator interface {
// Next 返回下一个元素，如果没有更多元素则返回nil和false
Next() (interface{}, bool)
// HasNext 判断是否还有下一个元素
HasNext() bool
}
2. 定义集合接口
集合接口只需要提供一个获取迭代器的方法即可：
// Collection 集合接口
type Collection interface {
// Iterator 返回当前集合对应的迭代器
Iterator() Iterator
}
3. 实现具体集合和具体迭代器
以字符串切片作为底层存储的集合为例，实现对应的迭代器和集合：
// StringSliceCollection 基于字符串切片的集合实现
type StringSliceCollection struct {
data []string
}
// NewStringSliceCollection 创建字符串切片集合
func NewStringSliceCollection(data []string) *StringSliceCollection {
return &StringSliceCollection{
data: data,
}
}
// Iterator 返回对应的迭代器实例
func (c *StringSliceCollection) Iterator() Iterator {
return &StringSliceIterator{
collection: c,
index: 0,
}
}
// StringSliceIterator 字符串切片集合的迭代器实现
type StringSliceIterator struct {
collection *StringSliceCollection
index int
}
// HasNext 判断是否还有下一个元素
func (it *StringSliceIterator) HasNext() bool {
return it.index < len(it.collection.data)
}
// Next 返回下一个元素
func (it *StringSliceIterator) Next() (interface{}, bool) {
if !it.HasNext() {
return nil, false
}
val := it.collection.data[it.index]
it.index++
return val, true
}
4. 使用迭代器遍历集合
通过迭代器完成集合的遍历，不需要关心集合内部的存储结构：
package main
import (
"fmt"
)
func main() {
// 初始化集合数据
data := []string{"golang", "迭代器", "模式", "集合", "遍历"}
collection := NewStringSliceCollection(data)
// 获取迭代器
iterator := collection.Iterator()
// 遍历集合
for iterator.HasNext() {
val, ok := iterator.Next()
if ok {
fmt.Println(val)
}
}
}
迭代器模式的扩展场景
如果需要实现逆序遍历，只需要新增一个逆序迭代器实现即可，不需要修改原有集合的代码：
// ReverseStringSliceIterator 逆序遍历的迭代器
type ReverseStringSliceIterator struct {
collection *StringSliceCollection
index int
}
// NewReverseStringSliceIterator 创建逆序迭代器
func NewReverseStringSliceIterator(c *StringSliceCollection) Iterator {
return &ReverseStringSliceIterator{
collection: c,
index: len(c.data) - 1,
}
}
// HasNext 判断是否还有上一个元素
func (it *ReverseStringSliceIterator) HasNext() bool {
return it.index >= 0
}
// Next 返回上一个元素
func (it *ReverseStringSliceIterator) Next() (interface{}, bool) {
if !it.HasNext() {
return nil, false
}
val := it.collection.data[it.index]
it.index--
return val, true
}
使用时只需要将集合获取迭代器的方法调整为返回逆序迭代器，或者在集合接口中新增获取逆序迭代器的方法即可，原有遍历逻辑不需要修改。
迭代器模式的优势
- 解耦集合和遍历逻辑，集合内部结构变化时，只要迭代器接口不变，遍历代码不需要修改。
- 支持多种遍历方式，比如正序、逆序、过滤遍历等，只需要新增对应的迭代器实现。
- 符合单一职责原则，集合只负责存储元素，遍历逻辑由迭代器单独负责。
在Golang中实现迭代器模式不需要复杂的语法支持，通过接口和结构体就能完成，适合需要灵活扩展遍历逻辑的集合场景使用。