Go by Example: Generics泛型

英文源地址
从Go1.18版本开始,Go增加了对泛型(也成为类型参数)的支持.

package main

import "fmt"

// MapKeys 作为泛型函数的一个示例, MapKeys接受任何类型的映射,并返回其键的一个切片
// 该函数有两个类型参数K和V
// K具有可比较的约束, 这意味着我们可以用==和!=操作符比较这种类型的值
// 这是Go语言中作为map的键的必需条件
// 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 作为泛型类型的一个示例.List是具有任意类型值的单链表
type List[T any] struct {
	head, tail *element[T]
}

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

// Push 我们在泛型类型上定义方法, 就像在常规类型上定义方法一样
// 但必须保留类型参数. 类型为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]

下一讲将介绍:Errors错误