
通过引入 类型形参 和 类型实参 这两个概念,我们让一个函数获得了处理多种不同类型数据的能力,这种编程方式被称为 泛型编程。
基本格式:
type Slice[T int|float32|float64 ] []T
泛型类型不能直接拿来使用,必须传入类型实参(Type argument) 将其确定为具体的类型之后才可使用。而传入类型实参确定具体类型的操作被称为 实例化(Instantiations)
- // 声明一个泛型
- type slice[T int | float32 | float64] []T
-
- func main() {
- // 这里传入了类型实参int,泛型类型Slice[T]被实例化为具体的类型 Slice[int]
- var a slice[int] = []int{1, 2, 3}
- fmt.Println(a) //[1 2 3]
- // 传入类型实参float32, 将泛型类型Slice[T]实例化为具体的类型 Slice[float32]
- var b slice[float32] = []float32{1.2, 123.123, 2123.1}
- fmt.Println(b) //[1.2 123.123 2123.1]
- }
其他类型的泛型
-
- //泛型map
- type myMap[KEP int | string, VAL int | float32] map[KEP]VAL
-
- func main() {
- table := myMap[string, int]{
- "xiaoming": 190,
- "xiaohong": 150,
- }
-
- fmt.Printf("%+v", table) //map[xiaohong:150 xiaoming:190]
- }
- //泛型结构体
- type myStruct[T int|string] struct {
- name string
- data T
- }
- // 一个泛型接口(关于泛型接口在后半部分会详细讲解)
- type IPrintData[T int | float32 | string] interface {
- Print(data T)
- }
- // 一个泛型通道,可用类型实参 int 或 string 实例化
- type MyChan[T int | string] chan T
- // 泛型嵌套
- type woStruct[T int | string, s []T] struct {
- Data s
- maxval T
- minval T
- }
-
- func main() {
- var test woStruct[int, []int] = woStruct[int, []int]{
- []int{1, 2, 3},
- 12,
- 31,
- }
- fmt.Printf("%+v", test) //{Data:[1 2 3] maxval:12 minval:31}
- }
任何泛型类型都必须传入类型实参实例化才可以使用。上面的代码中,我们为T传入了实参 int,然后因为 S 的定义是 []T ,所以 S 的实参自然是 []int
因为 S 的定义是 []T ,所以 T 一定决定了的话 S 的实参就不能随便乱传了
- go
- 复制代码// 错误,类型形参不能单独使用
- type CommonType[T int|string|float32] T
- go
- 复制代码//✗ 错误。T *int会被编译器误认为是表达式 T乘以int,而不是int指针
- type NewType[T *int] []T
- // 上面代码再编译器眼中:它认为你要定义一个存放切片的数组,数组长度由 T 乘以 int 计算得到
- type NewType [T * int][]T
-
- //✗ 错误。和上面一样,这里不光*被会认为是乘号,| 还会被认为是按位或操作
- type NewType2[T *int|*float64] []T
-
- //✗ 错误
- type NewType2 [T (int)] []T
为了避免这种误解,解决办法就是给类型约束包上 interface{} 或加上逗号消除歧义(关于接口具体的用法会在后半篇提及)
- go
- 复制代码type NewType[T interface{*int}] []T
- type NewType2[T interface{*int|*float64}] []T
-
- // 如果类型约束中只有一个类型,可以添加个逗号消除歧义
- type NewType3[T *int,] []T
-
- //✗ 错误。如果类型约束不止一个类型,加逗号是不行的
- type NewType4[T *int|*float32,] []T
因为上面逗号的用法限制比较大,这里推荐统一用 interface{} 解决问题
-
- type mySlice[T int | string | float32] []T
-
- //泛型方法
- func (m mySlice[T]) sum() T {
- var sum T
- for _, v := range m {
- sum += v
- }
- return sum
- }
-
- func main() {
- var t mySlice[int] = []int{1, 2, 3, 4, 5}
- fmt.Println(t.sum()) //15
-
- var f mySlice[float32] = []float32{1.2, 3.4, 5.6}
- fmt.Println(f.sum()) //10.200001
-
- }
泛型不像接口一样可以通过类型断言来判断其类型
但可以通过反射来实现动态判断其类型
- func (receiver Queue[T]) Put(value T) {
- // Printf() 可输出变量value的类型(底层就是通过反射实现的)
- fmt.Printf("%T", value)
-
- // 通过反射可以动态获得变量value的类型从而分情况处理
- v := reflect.ValueOf(value)
-
- switch v.Kind() {
- case reflect.Int:
- // do something
- case reflect.String:
- // do something
- }
-
- // ...
- }
- func main() {
- //在调用函数的时候声明类型
- fmt.Println(add[float64](float64(6.123), float64(8.12312)))
- //自动类型推断
- fmt.Println(add(19, 123)) //142
- }
- //定义一个函数泛型
- func add[T int | float32 | float64](a T, b T) T {
- return a + b
- }
匿名函数不能自己定义类型形参:
但是匿名函数可以使用别处定义好的类型实参,如:
- go
- 复制代码func MyFunc[T int | float32 | float64](a, b T) {
-
- // 匿名函数可使用已经定义好的类型形参
- fn2 := func(i T, j T) T {
- return i*2 - j*2
- }
-
- fn2(a, b)
- }
既然函数都支持泛型了,那你应该自然会想到,方法支不支持泛型?很不幸,目前Go的方法并不支持泛型
- type IntUintFloat interface {
- int | int8 | int16 | int32 | int64 | uint | uint8 | uint16 | uint32 | uint64 | float32 | float64
- }
-
- type Slice[T IntUintFloat] []T
这段代码把类型约束给单独拿出来,写入了接口类型 IntUintFloat 当中。需要指定类型约束的时候直接使用接口 IntUintFloat 即可。
不过这样的代码依旧不好维
上面定义的 Slie[T] 虽然可以达到目的,但是有一个缺点:
- go
- 复制代码var s1 Slice[int] // 正确
-
- type MyInt int
- var s2 Slice[MyInt] // ✗ 错误。MyInt类型底层类型是int但并不是int类型,不符合 Slice[T] 的类型约束
这里发生错误的原因是,泛型类型 Slice[T] 允许的是 int 作为类型实参,而不是 MyInt (虽然 MyInt 类型底层类型是 int ,但它依旧不是 int 类型)。
为了从根本上解决这个问题,Go新增了一个符号 ~ ,在类型约束中使用类似 ~int 这种写法的话,就代表着不光是 int ,所有以 int 为底层类型的类型也都可用于实例化。
限制:使用 ~ 时有一定的限制:
- type MyInt int
-
- type _ interface {
- ~[]byte // 正确
- ~MyInt // 错误,~后的类型必须为基本类型
- ~error // 错误,~后的类型不能为接口
- }
当满足以下条件时,我们可以说 类型 T 实现了接口 I ( type T implements interface I):
并集我们已经很熟悉了,之前一直使用的 | 符号就是求类型的并集( union )
接口可以不止书写一行,如果一个接口有多行类型定义,那么取它们之间的 交集
- go
- 复制代码type AllInt interface {
- ~int | ~int8 | ~int16 | ~int32 | ~int64 | ~uint | ~uint8 | ~uint16 | ~uint32 | ~uint32
- }
-
- type Uint interface {
- ~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64
- }
-
- type A interface { // 接口A代表的类型集是 AllInt 和 Uint 的交集
- AllInt
- Uint
- }
-
- type B interface { // 接口B代表的类型集是 AllInt 和 ~int 的交集
- AllInt
- ~int
- }
上面这个例子中
除了上面的交集,下面也是一种交集:
- go
- 复制代码type C interface {
- ~int
- int
- }
很显然,~int 和 int 的交集只有int一种类型,所以接口C代表的类型集中只有int一种类型
当多个类型的交集如下面 Bad 这样为空的时候, Bad 这个接口代表的类型集为一个空集:
- go
- 复制代码type Bad interface {
- int
- float32
- } // 类型 int 和 float32 没有相交的类型,所以接口 Bad 代表的类型集为空
上面说了空集,接下来说一个特殊的类型集——空接口 interface{} 。因为,Go1.18开始接口的定义发生了改变,所以 interface{} 的定义也发生了一些变更:
空接口代表了所有类型的集合
- // 空接口代表所有类型的集合。写入类型约束意味着所有类型都可拿来做类型实参
- type Slice[T interface{}] []T
因为空接口是一个包含了所有类型的类型集,所以我们经常会用到它。于是,Go1.18开始提供了一个和空接口 interface{} 等价的新关键词 any ,用来使代码更简单:
type Slice[T any] []T // 代码等价于 type Slice[T interface{}] []T
Go直接内置了一个叫 comparable 的接口,它代表了所有可用 != 以及 == 对比的类型
接口定义中如果只有方法的话,那么这种接口被称为基本接口(Basic interface)。这种接口就是Go1.18之前的接口,用法也基本和Go1.18之前保持一致
如果接口内不光只有方法,还有类型的话,这种接口被称为 一般接口(General interface) ,如下例子都是一般接口:
- go
- 复制代码type Uint interface { // 接口 Uint 中有类型,所以是一般接口
- ~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64
- }
-
- type ReadWriter interface { // ReadWriter 接口既有方法也有类型,所以是一般接口
- ~string | ~[]rune
-
- Read(p []byte) (n int, err error)
- Write(p []byte) (n int, err error)
- }
一般接口类型不能用来定义变量,只能用于泛型的类型约束中。所以以下的用法是错误的:
- go
- 复制代码type Uint interface {
- ~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64
- }
-
- var uintInf Uint // 错误。Uint是一般接口,只能用于类型约束,不得用于变量定义
这一限制保证了一般接口的使用被限定在了泛型之中,不会影响到Go1.18之前的代码,同时也极大减少了书写代码时的心智负担