你好,我是四哥。
3 月 15 号,Go1.18 发布,这么多年来讨论最多的泛型终于正式 realse,可以说是近几年最大的特性变更。
既然已经正式发布,肯定是要来尝鲜的,宠粉的四哥当然也是要给大家安排上的,安排!
今天我们要实现的函数功能是:对 map 里所有的 value 求和。
非泛型函数
如果没有泛型该如何实现呢?因为 value 可能会有不同的类型,比如:int64、float64 等,没有泛型的话,针对不同类型需要通过不同的函数来实现,所以我们一般都会像下面这样实现:
上面的代码,我们定义了两个函数:
1、SumInts() 计算 value 为 int64 类型的总和;2、SumFloats() 计算 value 为 float64 类型的总和;
上面函数只适用于 int64、float64 类型,那如果我们还想计算 int、float32 等,就得把上面的函数重新 copy 一份修改下类型。
这样做岂不是稍显麻烦,而且代码很臃肿,你还别说,之前标准包有很多类似的代码,比如标准库的 Sort 包(sort.go) 为了实现不同类型切片的排序,定义了 IntSlice、Float64Slice、StringSlice。
因为之前 Go 版本不支持泛型,只能这样实现,这也是最简单的一种方法。
有了泛型之后,针对不同类型 int64、float64,该如何实现一个函数就能计算出 value 的总和呢?
泛型函数
这一小节我们通过一个泛型函数实现既可以接收 value 为 int64 类型的 map 作为参数,也可以接收 value 为 float64 类型的 map 作为参数。
完整代码如下:
上面的代码,我们定义了 Sum() 函数,它是一个泛型函数,可以看到与普通函数不同的是,在函数名与函数参数列表之间有一组方括号 [],方括号里有两个参数 K 和 V,Go 语言里称为类型参数,紧跟在 K、V 后面是类型限制,其中 comparable 是 Go 语言预声明的,表示任何能做 == 和 != 操作的类型。V 的类型限制是 int64、float64 其中的一种。函数参数是 m,类型是 map[K]V,返回类型是 V。
调用函数也与调用普通函数有点不同,在函数名与实参之间有一组方括号 [],指明了类型名称。比如第一次调用 Sum() 函数时,类型名称分别是 string、int64,那么在执行函数时,对应 K、V 的类型分别是 string 和 int64。
第二次调用该 Sum() 函数时,K、V 的类型分别是 string 和 float64。
通过泛型编程,我们就可以实现一个函数处理多种数据类型。
执行上面的代码输出:
这篇文章我们通过一个简单示例入手泛型编程,关于泛型的语法、相关概念(比如类型参数、类型限制、类型推到等),四哥会在后续的文章里给大家讲解,抓紧学起来,没准还能在同事面前装 X 呢!
如果您的朋友也在学习 Go 语言,相信这篇文章对 TA 有帮助,欢迎转发分享给 TA,非常感谢!