体验 Go 1.18（二）：一文掌握泛型用法

泛型，可以说是 Go 这几年来最具争议的功能，应该没人有意见吧？

其实 Go 在早前的 Beta 版本中，就提供了对泛型的支持，但还不够成熟，直到 Go 1.18 才是支持泛型的正式版本。

下面我学习了官方关于泛型的文档之后，将学习的心得总结分享给大家

# 1. 非泛型的写法

现有一个 map ，我们需要实现一个函数，来遍历该 map 然后将 value 的值全部相加并返回。

而由于这个 map 的 value 可以是任意类型的数值，比如 int64, float64

于是为了接收不同类型的 map，我们就得定义多个函数，这些函数 除了入参类型及返回值类型不同外，没有任何不同

func SumInts(m map[string]int64) int64 {
    var s int64
    for _, v := range m {
        s += v
    }
    return s
}

func SumFloats(m map[string]float64) float64 {
    var s float64
    for _, v := range m {
        s += v
    }
    return s
}

# 2. 用泛型的写法

若是以代码行数来定义工作量，我可不希望泛型的出现，但从另一方面来讲，这种代码横看竖看都让人非常不舒服。

同样的需求，在有了泛型之后，写法就变得简洁许多，两个函数可以简化成一个函数

func SumIntsOrFloats[K comparable, V int64 | float64](m map[K]V) V {
    var s V
    for _, v := range m {
        s += v
    }
    return s
}

在这个函数中，它比常规的函数多了 [K comparable, V int64 | float64] 这么一段代码，这便是 泛型新增的语法，位于函数名与形参之间。

我来解释下这段 “代码”：

• K 和 V 你可以理解为类型别名，在中括号之间进行定义，作用域也只在此函数内，可以在形参、函数主体、返回值类型 里使用
• comparable 是 Go 语言预声明的类型，是那些可以比较（可哈希）的类型的集合，通常用于定义 map 里的 key 类型
• int64 | float64 意思是 V 可以是 int64 或 float64 中的任意一个
• map[K]V 就是使用了 K 和 V 这两个别名类型的 map

有了泛型函数的定义，那如何调用该函数？

调用方式还是跟普通函数一样，只是在函数名和实参之间，可以再次使用中括号来指明上面的 K 和 V 分别是什么类型？

func main() {
    // Initialize a map for the integer values
    ints := map[string]int64{
        "first": 34,
        "second": 12,
    }

    // Initialize a map for the float values
    floats := map[string]float64{
        "first": 35.98,
        "second": 26.99,
    }

    fmt.Printf("Generic Sums: %v and %v\n",
        SumIntsOrFloats[string, int64](ints),
        SumIntsOrFloats[string, float64](floats),
    )
}

最后使用 go run 去跑一下，结果正常输出

# 3. 简化泛型写法

3.1 类型自动推导

在调用大部分的泛型函数时，中括号里的内容，是可以省略不写的，而这个不写的前提是，编译器有办法根据你的实参及形参来自动推导出泛型函数中 别名类型对应的类型（在上例中就是 K 和 V）。

而在上面的例子中，刚好是满足的，于是泛型函数的调用就可以简化成这样

    fmt.Printf("Generic Sums: %v and %v\n",
        SumIntsOrFloats(ints),
        SumIntsOrFloats(floats),
    )

3.2 使用类型别名

上面的 V 使用 int64 | float64 这样的写法来表示 V 可以是其中的任意一种类型。

若这个 V 用得比较多呢？可以考虑用 type 来事先定义别名，一来方便复用，二来在泛型函数使用更加简短的别名，可以提高代码可读性

type Number interface {
    int64 | float64
}

然后泛型函数的定义就可以简化成下面这样

func SumNumbers[K comparable, V Number](m map[K]V) V {
    var s V
    for _, v := range m {
        s += v
    }
    return s
}

# 4. 写在最后

在去年，其实就通过其他人的文章中事先了解到了 Go 泛型的写法，给我的第一印象是，函数的定义变得更复杂，可读性也越来越差，一时间我也有点难以接受。

不过经过自己试用后，情况倒没有我想象的那么糟糕！新版没有改变原有函数的定义与调用，若你没有使用泛型，那么有没有泛型对你来说没有区别。

但即使你有想法需要用到泛型，我也相信这种的不适感会在时间的流逝中慢慢淡化。

本文由哈喽比特于2年以前收录，如有侵权请联系我们。
文章来源：https://mp.weixin.qq.com/s/xc3b9cB9WsDoK92wIg6RNw