非泛型函数如果没有泛型该如何实现呢?因为 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。
通过泛型编程,我们就可以实现一个函数处理多种数据类型。
执行上面的代码输出:
362.97
以上就是go1.18 泛型初探的详细内容。