在线文档  >   Golang练习   >   泛型

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

package main

import "fmt"

// 作为一个泛型函数的例子,`MapKeys` 接受任意类型的 map 并返回其键的切片。
// 此函数有两个类型参数 - `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]