Go 泛型的這 3 個核心設計,你都知道嗎?

煎魚發表於2022-01-05

大家好,我是煎魚。

Go1.18 的泛型是鬧得沸沸揚揚,雖然之前寫過很多篇針對泛型的一些設計和思考。但因為泛型的提案之前一直還沒定型,所以就沒有寫完整介紹。

如今已經基本成型,就由煎魚帶大家一起摸透 Go 泛型。本文內容主要涉及泛型的 3 大概念,非常值得大家深入瞭解。

如下:

  • 型別引數。
  • 型別約束。
  • 型別推導。

型別引數

型別引數,這個名詞。不熟悉的小夥伴咋一看就懵逼了。

泛型程式碼是使用抽象的資料型別編寫的,我們將其稱之為型別引數。當程式執行通用程式碼時,型別引數就會被型別引數所取代。也就是型別引數是泛型的抽象資料型別

簡單的泛型例子:


func Print(s []T) {
    for _, v := range s {
        fmt.Println(v)
    }
}

程式碼有一個 Print 函式,它列印出一個片斷的每個元素,其中片斷的元素型別,這裡稱為 T,是未知的。

這裡引出了一個要做泛型語法設計的點,那就是:T 的泛型型別引數,應該如何定義

在現有的設計中,分為兩個部分:

  • 型別引數列表:型別引數列表將會出現在常規引數的前面。為了區分型別引數列表和常規引數列表,型別引數列表使用方括號而不是小括號。
  • 型別引數約束:如同常規引數有型別一樣,型別引數也有元型別,被稱為約束(後面會進一步介紹)。

結合完整的例子如下:

// Print 可以列印任何片斷的元素。
// Print 有一個型別引數 T,並有一個單一的(非型別)的 s,它是該型別引數的一個片斷。
func Print[T any](s []T) {
    // do something...
}

在上述程式碼中,我們宣告瞭一個函式 Print,其有一個型別引數 T,型別約束為 any,表示為任意的型別,作用與 interface{} 一樣。他的入參變數 s 是型別 T 的切片。

函式宣告完了,在函式呼叫時,我們需要指定型別引數的型別。如下:

    Print[int]([]int{1, 2, 3})

在上述程式碼中,我們指定了傳入的型別引數為 int,並傳入了 []int{1, 2, 3} 作為引數。

其他型別,例如 float64:

    Print[float64]([]float64{0.1, 0.2, 0.3})

也是類似的宣告方式,照著套就好了。

型別約束

說完型別引數,我們再說說 “約束”。在所有的型別引數中都要指定型別約束,才能叫做完整的泛型。

以下分為兩個部分來具體展開講解:

  • 定義函式約束。
  • 定義運算子約束。

為什麼要有型別約束

為了確保呼叫方能夠滿足接受方的程式訴求,保證程式中所應用的函式、運算子等特效能夠正常執行。

泛型的型別引數,型別約束,相輔相成。

定義函式約束

問題點

我們看看 Go 官方所提供的例子:

func Stringify[T any](s []T) (ret []string) {
    for _, v := range s {
        ret = append(ret, v.String()) // INVALID
    }
    return ret
}

該方法的實現目的是:任何型別的切片都能轉換成對應的字串切片。但程式邏輯裡有一個問題,那就是他的入參 T 是 any 型別,是任意型別都可以傳入。

其內部又呼叫了 String 方法,自然也就會報錯,因為只像是 int、float64 等型別,就可能沒有實現該方法。

你說要定義有效的型別約束,那像是上面的例子,在泛型中如何實現呢?

要求傳入方要有內建方法,就得定義一個 interface 來約束他。

單個型別

例子如下:

type Stringer interface {
    String() string
}

在泛型方法中應用:

func Stringify[T Stringer](s []T) (ret []string) {
    for _, v := range s {
        ret = append(ret, v.String())
    }
    return ret
}

再將 Stringer 型別放到原有的 any 型別處,就可以實現程式所需的訴求了。

多個型別

如果是多個型別約束。例子如下:

type Stringer interface {
    String() string
}

type Plusser interface {
    Plus(string) string
}

func ConcatTo[S Stringer, P Plusser](s []S, p []P) []string {
    r := make([]string, len(s))
    for i, v := range s {
        r[i] = p[i].Plus(v.String())
    }
    return r
}

與常規的入參、出參型別宣告一樣的規則。

定義運算子約束

完成了函式約束的定義後,剩下一個要啃的大骨頭就是 “運算子” 的約束了。

問題點

我們看看 Go 官方的例子:

func Smallest[T any](s []T) T {
    r := s[0] // panic if slice is empty
    for _, v := range s[1:] {
        if v < r { // INVALID
            r = v
        }
    }
    return r
}

經過上面的函式例子,我們很快能意識到這個程式根本無法執行成功。

其入參是 any 型別,程式內部是按 slice 型別來獲取值,且在內部又進行運算子比較,那如果真是 slice,內部就可能每個值型別都不一樣。

如果一個是 slice,一個是 int 型別,又如何進行運算子的值對比?

近似元素

可能有的同學想到了過載運算子,但...想太多了,Go 語言沒有支援的計劃。為此做了一個新的設計,那就是允許限制型別引數的型別範圍。

語法如下:

InterfaceType  = "interface" "{" {(MethodSpec | InterfaceTypeName | ConstraintElem) ";" } "}" .
ConstraintElem = ConstraintTerm { "|" ConstraintTerm } .
ConstraintTerm = ["~"] Type .

例子如下:

type AnyInt interface{ ~int }

上述宣告的型別集是 ~int,也就是所有型別為 int 的型別(如:int、int8、int16、int32、int64)都能夠滿足這個型別約束的條件。

包括底層型別是 int8 型別的,例如:

type AnyInt8 int8

也就是在該匹配範圍內的。

聯合元素

如果希望進一步縮小限定型別,可以結合分隔符來使用,用法為:

type AnyInt interface{
 ~int8 | ~int64
}

就可以將型別集限定在 int8 和 int64 之中。

實現運算子約束

基於新的語法,結合新的概念聯合和近似元素,可以把程式改造一下,實現在泛型中的運算子的匹配。

型別約束的宣告,如下:

type Ordered interface {
    ~int | ~int8 | ~int16 | ~int32 | ~int64 |
        ~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64 | ~uintptr |
        ~float32 | ~float64 |
        ~string
}

應用的程式如下:

func Smallest[T Ordered](s []T) T {
    r := s[0] // panics if slice is empty
    for _, v := range s[1:] {
        if v < r {
            r = v
        }
    }
    return r
}

確保了值均為基礎資料型別後,程式就可以正常執行了。

型別推導

程式設計師寫程式碼,一定程度的偷懶是必然的。

在一定的場景下,可以通過型別推導來避免明確地寫出一些或所有的型別引數,編譯器會進行自動識別。

建議複雜函式和引數能明確是最好的,否則讀程式碼的同學會比較麻煩,可讀性和可維護性的保證也是工作中重要的一點。

引數推導

函式例子。如下:

func Map[F, T any](s []F, f func(F) T) []T { ... }

公共程式碼片段。如下:

var s []int
f := func(i int) int64 { return int64(i) }
var r []int64

明確指定兩個型別引數。如下:

r = Map[int, int64](s, f)

只指定第一個型別引數,變數 f 被推斷出來。如下:

r = Map[int](s, f)

不指定任何型別引數,讓兩者都被推斷出來。如下:

r = Map(s, f)

約束推導

神奇的在於,型別推導不僅限與此,連約束都可以推導。

函式例子,如下:

func Double[E constraints.Number](s []E) []E {
    r := make([]E, len(s))
    for i, v := range s {
        r[i] = v + v
    }
    return r
}

基於此的推導案例,如下:

type MySlice []int

var V1 = Double(MySlice{1})

MySlice 是一個 int 的切片型別別名。變數 V1 的型別編譯器推導後 []int 型別,並不是 MySlice。

原因在於編譯器在比較兩者的型別時,會將 MySlice 型別識別為 []int,也就是 int 型別。

要實現 “正確” 的推導,需要如下定義:

type SC[E any] interface {
    []E 
}

func DoubleDefined[S SC[E], E constraints.Number](s S) S {
    r := make(S, len(s))
    for i, v := range s {
        r[i] = v + v
    }
    return r
}

基於此的推導案例。如下:

var V2 = DoubleDefined[MySlice, int](MySlice{1})

只要定義顯式型別引數,就可以獲得正確的型別,變數 V2 的型別會是 MySlice。

那如果不宣告約束呢?如下:

var V3 = DoubleDefined(MySlice{1})

編譯器通過函式引數進行推導,也可以明確變數 V3 型別是 MySlice。

總結

今天我們在文章中給大家介紹了泛型的三個重要概念,分別是:

  • 型別引數:泛型的抽象資料型別。
  • 型別約束:確保呼叫方能夠滿足接受方的程式訴求。
  • 型別推導:避免明確地寫出一些或所有的型別引數。

在內容中也涉及到了聯合元素、近似元素、函式約束、運算子約束等新概念。本質上都是基於三個大概念延伸出來的新解決方法,一環扣一環。

你學會 Go 泛型了嗎,設計的如何,歡迎一起討論:)

若有任何疑問歡迎評論區反饋和交流,最好的關係是互相成就,各位的點贊就是煎魚創作的最大動力,感謝支援。

文章持續更新,可以微信搜【腦子進煎魚了】閱讀,本文 GitHub github.com/eddycjy/blog 已收錄,學習 Go 語言可以看 Go 學習地圖和路線,歡迎 Star 催更。

參考

相關文章