透過範例學習 Go:泛型

從版本 1.18 開始,Go 已新增支援泛型,也就是所謂的型別參數。

package main
import "fmt"

以泛函程式的範例來說,MapKeys 會接收任何型別的地圖,並傳回其金鑰的切片。這個函式具有兩個型別參數,分別為 KV;其中 Kcomparable 約束,表示我們可以使用 ==!= 函數來比較這個型別的值。Go 中的地圖金鑰需要具備這個條件。Vany 的約束,表示沒有任何限制(anyinterface{} 的別名)。

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 時,我們不必指定 KV 的型別,因為編譯器會自動推論。

    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 中並未定義對應地圖金鑰的迭代順序,因此不同的呼叫可能會產生不同的順序。

下一個範例:錯誤