迭代器模式

迭代器模式

被迭代对象接口

package iterator

type Aggregate interface {
createIterator() Iterator
}

具体被迭代的对象

package iterator

type array []int

func NewArray(args ...int) array {
var arr array
for _, v := range args {
arr = append(arr, v)
}
return arr
}

func (a array) createIterator() Iterator {
return &SliceIterator{
object: NewArray(a...),
index: 0,
}
}

迭代器接口

package iterator

type Iterator interface {
hasNext() bool
getNext() int
}

迭代器实现

package iterator

type SliceIterator struct {
object array
index int
}

func (s *SliceIterator) getNext() int {
res := s.object[s.index]
s.index++
return res
}

func (s *SliceIterator) hasNext() bool {
if s.index < len(s.object) {
return true
}
return false
}

测试文件

package iterator

func TestIterator(t *testing.T) {
iter := NewArray(4, 2, 3, 8).createIterator()
for iter.hasNext() {
fmt.Printf("%v\n", iter.getNext())
}
}