Go 示例: 泛型

从 1.18 版本开始，Go 添加了对泛型的支持，也称为类型参数。

 
package main
 
import "fmt"

例如，泛型函数 MapKeys 接受任何类型的映射并返回其键的切片。此函数有两个类型参数 - K 和 V；K 具有 comparable 的约束，这意味着我们可以使用 == 和 != 运算符比较此类型的值。这是 Go 中映射键所必需的。V 具有 any 约束，这意味着它不受任何限制（any 是 interface{} 的别名）。

 
func MapKeys[K comparable, V any](m map[K]V) []K {
    r := make([]K, 0, len(m))
    for k := range m {
        r = append(r, k)
    }
    return r
}

例如，泛型类型 List 是一个单链表，其值可以是任何类型。

 
type List[T any] struct {
    head, tail *element[T]
}
 
type element[T any] struct {
    next *element[T]
    val  T
}

我们可以像在普通类型上一样在泛型类型上定义方法，但我们必须保留类型参数。类型是 List[T]，而不是 List。

 
func (lst *List[T]) Push(v T) {
    if lst.tail == nil {
        lst.head = &element[T]{val: v}
        lst.tail = lst.head
    } else {
        lst.tail.next = &element[T]{val: v}
        lst.tail = lst.tail.next
    }
}
 
func (lst *List[T]) GetAll() []T {
    var elems []T
    for e := lst.head; e != nil; e = e.next {
        elems = append(elems, e.val)
    }
    return elems
}
 
func main() {
    var m = map[int]string{1: "2", 2: "4", 4: "8"}

在调用泛型函数时，我们通常可以依赖类型推断。请注意，在调用 MapKeys 时，我们不必指定 K 和 V 的类型 - 编译器会自动推断它们。

 
    fmt.Println("keys:", MapKeys(m))

… 尽管我们也可以显式地指定它们。

 
    _ = MapKeys[int, string](m)
 
    lst := List[int]{}
    lst.Push(10)
    lst.Push(13)
    lst.Push(23)
    fmt.Println("list:", lst.GetAll())
}
 
$ go run generics.go
keys: [4 1 2]
list: [10 13 23]

注意：Go 中未定义遍历映射键的顺序，因此不同的调用可能会导致不同的顺序。

下一个示例：错误.

作者：Mark McGranaghan 和 Eli Bendersky | 源代码 | 许可证