当前位置：首页 > news >正文

GO基础进阶篇 (十三)、泛型

news 来源：原创 2024/5/21 4:37:43

泛型

先看一下这段代码。

package mainimport "fmt"func main() {strs := []string{"a", "b"}printArray(strs)
}func printArray(arr []interface{}) {for _, a := range arr {fmt.Println(a)}
}

上面的代码中，我们想要打印参数arr的信息。运行报错

 cannot use strs (variable of type []string) as []interface{} value in argument to printArray

想要解决的话，按照之前我们的学习，可以将函数改编如下（使用断言）

package mainimport "fmt"func main() {strs := []string{"a", "b"}printStringArray(strs)
}func printStringArray(arr interface{}) {for _, a := range arr.([]string) {fmt.Println(a)}
}

但这样会有一个坏处，当我们想要打印另一个非string的数组时，就不得不再写一个方法

package mainimport "fmt"func main() {ints := []int{1, 2}printIntArray(ints)
}func printIntArray(arr interface{}) {for _, a := range arr.([]int) {fmt.Println(a)}
}

这样处理，就会导致有无限多相似的代码产生，这样的代码时不合格的。

此时，泛型就出现了。它的意义时不在方法定义时决定变量的类型，而让使用者使用时决定。

package mainimport "fmt"func main() {ints := []int{1, 2}strs := []string{"a", "b"}printArray(ints)printArray(strs)
}func printArray[T string | int](arr []T) {for _, a := range arr {fmt.Println(a)}
}

T代表了用户传入的类型，并对T进行了约束。上面的代码中，我们再定一个float的数组，是无法通过程序校验的，因为我们约束了T的可用类型为stirng与int。

使用泛型，你可能会产生一个疑惑，通过我们刚刚学习的反射，再加上接口。也可以类似泛型这样的函数。这样是可行的，但反射的机制存在一些问题

1.用起来麻烦
2.失去了编译时的类型检查，容易出错
3.性能不理想

结论：当需要因为不同类型写完全相同的逻辑代码时，使用泛型时最合适的选择。

泛型类型

泛型切片

package mainimport "fmt"func main() {type Slice[T int | string | float32] []Tvar a Slice[int] = []int{1, 2, 3}var b Slice[string] = []string{"a", "b", "c"}var c Slice[float32] = []float32{1, 2, 3}fmt.Printf("%T", a)fmt.Println(a)fmt.Printf("%T", b)fmt.Println(b)fmt.Printf("%T", c)fmt.Println(c)
}//main.Slice[int][1 2 3]
//main.Slice[string][a b c]
//main.Slice[float32][1 2 3]

泛型map

package mainimport "fmt"func main() {type MyMap[KEY int | string, VALUE float32] map[KEY]VALUEvar m1 MyMap[string, float32] = map[string]float32{"a": 1.1,"b": 1.2,}fmt.Println(m1)
}

其他

	//泛型结构体type MyStruct[T int | string] struct {id   TName stirng}//泛型接口type IPrintData[T int | float32 | string] interface {}//泛型通道type MyChan[T string | int] chan T

泛型函数与方法

package mainimport "fmt"func main() {//给泛型添加方法var s MySlice[int] = []int{1, 2, 3, 4}fmt.Println(s.Sum())var s1 MySlice[float64] = []float64{1.1, 2.1, 3.1, 4.1}fmt.Println(s1.Sum())//泛型函数fmt.Println(Add[int](1, 2))fmt.Println(Add[string]("1", "2"))//如果类型能被自动推断，函数调用时的T可以省略fmt.Println(Add(1, 2))fmt.Println(Add("1", "2"))
}type MySlice[T int | string | float64] []T
func (s MySlice[T]) Sum() T {var sum Tfor _, v := range s {sum += v}return sum
}func Add[T int | float32 | string](a T, b T) T {return a + b
}

实际开发中，泛型使用较多的场景就是泛型的函数与方法。

自定义泛型约束

package mainimport "fmt"func main() {fmt.Println(GetMaxNum(1, 2))fmt.Println(GetMaxNum(1.5, 2.6))
}type MyInt interface {int | int8 | int16 | int32 | float64
}func GetMaxNum[T MyInt](a, b T) T {if a > b {return a}return b
}