体验 Go 1.18（二）：一文掌握泛型用法 - 哈喽比特

402次阅读 | 发布于2年以前

泛型，可以说是 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 泛型的写法，给我的第一印象是，函数的定义变得更复杂，可读性也越来越差，一时间我也有点难以接受。

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

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