Go的泛型来了

大家好，我是peachesTao，本周三Go语言史上改动最大的版本Go 1.18发布了，为什么说改动最大？因为Gopher们期待已久的泛型包含在这里版本中，之前Go 1.18 Beta 1中有实验性的泛型。

除了泛型，Go 1.18 还包含其他的新功能，如：模糊测试、工作区，详细的更新内容请参考Go 1.18 Release Notes。

如果你想知道go为什么需要泛型可以参考文章底部的连接why-generics，里面详细介绍了泛型的需求及引入泛型后要解决的一序列问题。

今天我们就介绍一下go中泛型的基本使用

泛型方法

假如我要分别求[]int和[]float64切片元素之和

没有泛型时

要写两个方法，分别实现int和float64两种类型

func SumSliceInt(slice []int) int {
 sum := 0
 for _, v := range slice {
  sum += v
 }
 return sum
}

func SumSliceFloat64(slice []float64) float64 {
 sum := float64(0)
 for _, v := range slice {
  sum += v
 }
 return sum
}

有了泛型后

一个标准的泛型函数模板如下：

func GenericFunc[T any](args T) {
   // logic code
}

其中T是参数类型，any是参数类型的约束

现在我们将上面两个方法改造成泛型方法，参数类型T的约束为int或float64,中间用“|”隔开，返回值类型为T

func sumSlice[T int|float64](slice []T) (T) {
 sum := T(0)
 for _, v := range slice {
  sum += v
 }
 return sum
}

调用

func main() {
 fmt.Println(sumSlice[int]([]int{1, 2, 3}))
 fmt.Println(sumSlice[float64]([]float64{1.1, 2.2, 3.3}))
}

有了泛型后，两个方法二合一，简单明了

调用方法时数据类型int和float64可以不写，编译器会自动推导

 sumSlice([]int{1, 2, 3})
 sumSlice([]float64{1.1, 2.2, 3.3})

此时如果我还有一个泛型方法：SortSlice，将切片中的元素排序，

func SortSlice[T int|float64](slice []T) (T) {
 // ...
}

数据类型也是int和float64，和上面的sumSlice一样，那么可不可复用，减少维护成本？答案是肯定的，可以通过定义一个新的inteface实现，如下：

type Number interface {
 int | float64
}

将sumSlice和SortSlice形参中的int | float64改成Number即可：

func sumSlice[T Number](slice []T) (T) {
 sum := T(0)
 for _, v := range slice {
  sum += v
 }
 return sum
}

func SortSlice[T Number](slice []T) (T) {
 // ...
}

泛型数据结构

复杂数据结构支持泛型跟函数泛型用法差不多，下面定义一个支持泛型的结构体

type SumStruct[T int|float64] struct {
 First T
 Second T
}

func(s SumStruct[T]) Sum() T  {
 return s.First + s.Second
}

Go的官方也出了一个泛型的教程，链接在底部，也可以去看下。

目前Go的IDE goland支持泛型语法还不是很好，估计要等一段时间了。

今天泛型的用法就介绍到这里了，我们下期见。

如果这篇文章对你有帮助，欢迎转发和点赞。

参考链接

Go1.18泛型编程体验
[Go 1.18 Release Notes]
https://go.dev/doc/go1.18
[why-generics]
https://go.dev/blog/why-generics

[Go泛型官方教程] 

https://go.dev/doc/tutorial/generics