Go程序例子(25)：遍历迭代器

Go程序例子(25)：遍历迭代器

从 1.2 版本开始，Go 语言增加了对迭代器的支持，这让我们几乎可以对任何东西进行 range 遍历！

package main

import (
    "fmt"
    "iter"
    "slices"
)
// 让我们再次看看前面示例中的 List 类型。在那个示例中，我们有一个 AllElements 方法，它返回列表中所有元素的切片。使用 Go 的迭代器，我们可以做得更好——如下所示。
type List[T any] struct {
    head, tail *element[T]
}

type element[T any] struct {
    next *element[T]
    val  T
}

func (lst *List[T]) Push(v T) {
    if  == nil {
        lst.head = &element[T]{val: v}
         = lst.head
    } else {
        .next = &element[T]{val: v}
         = .next
    }
}
// ll 返回一个迭代器，在 Go 中，它是一个具有特殊签名的函数。
func (lst *List[T]) All() iter.Seq[T] {
    return func(yield func(T) bool) {
		// 迭代器函数接受另一个函数作为参数，通常称为 yield（但名称可以是任意的）。它会在每个我们想要迭代的元素上调用 yield，并注意 yield 的返回值，以便在需要时提前终止迭代。
        for e := lst.head; e != nil; e =  {
            if !yield(e.val) {
                return
            }
        }
    }
}
// 迭代不需要底层数据结构，甚至不必是有限的！这里有一个返回斐波那契数列迭代器的函数：只要 yield 一直返回 true，它就会继续运行。
func genFib() iter.Seq[int] {
    return func(yield func(int) bool) {
        a, b := 1, 1

        for {
            if !yield(a) {
                return
            }
            a, b = b, a+b
        }
    }
}

func main() {
    lst := List[int]{}
    lst.Push(10)
    lst.Push(1)
    lst.Push(2)
	// 由于 List.All 返回一个迭代器，我们可以在常规的 range 循环中使用它。
    for e := range lst.All() {
        fmt.Println(e)
    }
	// 像 slices 这样的包提供了许多有用的函数来处理迭代器。例如，Collect 接受任何迭代器，并将其所有值收集到一个切片中。
    all := slices.Collect(lst.All())
    fmt.Println("all:", all)

    for n := range genFib() {
		// 一旦循环遇到 break 或提前返回，传递给迭代器的 yield 函数将返回 false。
        if n >= 10 {
            break
        }
        fmt.Println(n)
    }
}

运行结果：

$ go run range-over-iterators.go
10
1
2
all: [10 1 2]
1
1
2

5
8