当前位置 : 主页 > 网络编程 > 其它编程 >

Go1.18 泛型初探

来源:互联网 收集:自由互联 发布时间:2023-07-31
今天我们要实现的函数功能是:对 map 里所有的 value 求和。 非泛型函数 如果没有泛型该如何实现呢?因为 value 可能会有不同的类型,比如:int64、float64 等,没有泛型的话,针对不同类

今天我们要实现的函数功能是:对 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

网友评论