注：本文已发布超过一年，请注意您所使用工具的相关版本是否适用
序
笔记
代码实现
下面是一个简单的自定义数组类型的例子
Code
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
| package iterator// Iterator 迭代器接口
type Iterator interface {
HasNext() bool
Next()
// 获取当前元素，由于 Go 1.15 中还没有泛型，所以我们直接返回 interface{}
CurrentItem() interface{}
} // ArrayInt 数组
type ArrayInt []int // Iterator 返回迭代器
func (a ArrayInt) Iterator() Iterator {
return &ArrayIntIterator{
arrayInt: a,
index: 0,
}
} // ArrayIntIterator 数组迭代
type ArrayIntIterator struct {
arrayInt ArrayInt
index int
} // HasNext 是否有下一个
func (iter *ArrayIntIterator) HasNext() bool {
return iter.index < len(iter.arrayInt)-1
} // Next 游标加一
func (iter *ArrayIntIterator) Next() {
iter.index++
} // CurrentItem 获取当前元素
func (iter *ArrayIntIterator) CurrentItem() interface{} {
return iter.arrayInt[iter.index]
}
|
单元测试
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| package iteratorimport (
"testing" "github.com/stretchr/testify/assert"
) func TestArrayInt_Iterator(t *testing.T) {
data := ArrayInt{1, 3, 5, 7, 8}
iterator := data.Iterator()
// i 用于测试
i := 0
for iterator.HasNext() {
assert.Equal(t, data[i], iterator.CurrentItem())
iterator.Next()
i++
}
}
|
关注我获取更新
猜你喜欢
