来源:公众号【peachesTao】
作者:peachesTao
原文地址: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的官方也出了一个泛型的教程,也可以去看下.
今天泛型的用法就介绍到这里了,我们下期见。
Go1.18泛型编程体验
如果这篇文章对你有帮助,可以关注我的公众号,第一时间获取最新的原创文章