今天我们要实现的函数功能是:对 map 里所有的 value 求和。
如果没有泛型该如何实现呢?因为 value 可能会有不同的类型,比如:int64、float64 等,没有泛型的话,针对不同类型需要通过不同的函数来实现,所以我们一般都会像下面这样实现:
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 }
上面的代码,我们定义了两个函数:
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 作为参数。
完整代码如下:
func Sum[K comparable, V int64 | float64](m map[K]V "K comparable, V int64 | float64") V { var s V for _, v := range m { s += v } return s } func main() { ints := map[string]int64{ "first": 1, "second": 2, } fmt.Println(Sum[string, int64](ints "string, int64")) floats := map[string]float64{ "first": 35.98, "second": 26.99, } fmt.Println(Sum[string, float64](floats "string, float64")) }
上面的代码,我们定义了 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。
通过泛型编程,我们就可以实现一个函数处理多种数据类型。
执行上面的代码输出:
3 62.97