泛型是随着Go 1.18版本发布的。它基本上意味着参数化的类型,也就是说,它允许程序员在写代码时,类型可以稍后指定,因为类型在当时并不相关。换句话说,在编写一些代码时,你不提供数值的类型。这些类型的值会在以后传递。
其语法为:
func funcName[type_parameter type_constraint](… type_parameter) type_parameter { … } func funcName[T any](… T) T { … } func funcName[T interface{}](… T) T { … }
这里T是类型参数,any是类型约束,可以是任何接口,意味着无限的值,这里any代表一个空接口。
在下面的例子中,有两个函数,第一个returnFirst
接受 any
,意味着我们可以传递任何非特定的int或float,并且它返回第一个参数。第二个函数即returnFloatFirst只接受float64类型的参数,如果我们传递任何其他类型的参数,它就会抛出一个错误。在main函数中,你可以看到我们可以在不指定约束类型的情况下调用returnFirst
函数,并且它是有效的,这是因为类型推理。
func returnFirst[T any](a T, b T) T { return a } func returnFloatFirst[T float64](a T, b T) T { return a } func main() { fmt.Println(returnFirst[int](1,3)) fmt.Println(returnFirst(1,3)) fmt.Println(returnFirst[float64](1.8,3.9)) fmt.Println(returnFirst(1.8,3.9)) fmt.Println(returnFirst("a","b")) fmt.Println(returnFloatFirst(1.2,3.4)) }
Go 1.18带有类型推理功能,可以帮助我们编写无需显式类型就能调用通用函数的代码。
让我们再举一个例子,比如说你要计算数组中所有元素的总和,现在如果我说每次改变数组存储的数据类型,那么函数的实现将保持不变,但你必须写一个单独的函数来适应不同的类型,所以让我们写一个通用函数,将数组中的元素相加并返回总和。
func sumAll[T any](arr []T) T { var s T for _, ele := range arr { s += ele } return s } func main() { fmt.Println("sum: ", sumAll([]int{1, 2, 3, 5, 6})) }
当你运行上述代码时,会出现一个错误:
$ go run main.go ./main.go:6:9: invalid operation: operator + not defined on a (variable of type T constrained by any)
这是因为任何约束都可以持有任何值,在上面的例子中,它是int,但它可以是任何东西,而且有可能运算符对那个特定的类型不起作用,所以它抛出一个错误。为了解决这个问题,我们使用类型集,在接口的帮助下定义一个自定义约束,并在类型约束的地方使用它。我们为该约束声明一组类型,我们必须只使用这些类型。
定义自定义约束的语法是:
type customConstraint interface { type1 | type2 | type3 … } type cusConstraint interface { float64 | int | string }
我们还可以使用约束包,它定义了一组有用的约束,可以与参数一起使用。
首先,我们必须安装约束包。
$ go get golang.org/x/exp/constraints
来自包的一些限制。
type Signed interface { ~int | ~int8 | ~int16 | ~int32 | ~int64 } type Unsigned interface { ~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64 | ~uintptr } type Integer interface { Signed | Unsigned } type Float interface { ~float32 | ~float64 } type Complex interface { ~complex64 | ~complex128 } type Ordered interface { Integer | Float | ~string }
所以现在让我们创建一个自定义约束,它可以支持float64和int类型的数据,并在任何约束的位置使用它。
type constraint interface { ~float64 | int } func sumAll[T constraint](arr []T) T { var s T for _, ele := range arr { s += ele } return s } func main() { fmt.Println("sum: ", sumAll([]int{1, 2, 3, 5, 6})) fmt.Println("sum: ", sumAll([]float64{1.2, 2.1, 3.8, 5.4})) }
上述代码是有效的。