Go1.18都出泛型了速来围观

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

go泛型使用的官方说明：https://go.dev/doc/tutorial/generics

在使用之前先把go更新到1.18或者以上的版本：https://go.dev/doc/install

如果用过c++或者Java的话，那么对泛型这个概念应该是不陌生的。（下面这段定义摘抄自百度百科）
泛型程序设计（generic programming）是程序设计语言的一种风格或范式。泛型允许程序员在强类型程序设计语言中编写代码时使用一些以后才指定的类型，在实例化时作为参数指明这些类型。

我们可以用个简单的例子来看看如何使用泛型。

在没有泛型之前，如果我们要写一个比较大小的函数的话，因为go里面没有函数重载的概念，所以我们必须对需要比较的类型都创建一个比较函数，不过这样的话，就会有大量逻辑重复的代码，他们的区别只是变量的类型不同罢了。

package main

import "fmt"
func MaxInt(a, b int) int {
	if a > b {
		return a
	} else {
		return b
	}
}
func MaxFloat32(a, b float32) float32 {
func main() {
	var a, b int
	var c, d float32
	a = 1
	b = 2
	c = 1.1
	d = 1.4
	fmt.Println(MaxInt(a, b))
	fmt.Println(MaxFloat32(c, d))

那么泛型该如何使用呢？
go预留了一个泛型，名为comparable，这是官方定义的一个可比较的类型约束

// comparable is an interface that is implemented by all comparable types
// (booleans, numbers, strings, pointers, channels, arrays of comparable types,
// structs whose fields are all comparable types).
// The comparable interface may only be used as a type parameter constraint,
// not as the type of a variable.
//翻译如下
//comparable是由所有可比较类型实现的接口
//（布尔、数字、字符串、指针、通道、类似类型的数组、，
//字段均为可比较类型的结构）。
//可比接口只能用作类型参数约束，
//不是作为变量的类型。
type comparable interface{ comparable }

我们可以来自己声明一个类型约束。

type Number interface {
	int | int32 | int64 | float32
}

如上定义表示如果参数类型要为int、int32、int64、float32这四个类型其中一个的适合，可以用Number替代。
所以我们可以把最初比较大小的代码进行修改、简化一下。

package main

import "fmt"
type Number interface {
	int | int32 | int64 | float32
}
func MaxNumber[K Number](a K, b K) K {
	if a > b {
		return a
	} else {
		return b
	}
func main() {
	var a, b int
	var c, d float32
	a = 1
	b = 2
	c = 1.1
	d = 1.4
	fmt.Println(MaxNumber(a, b))
	fmt.Println(MaxNumber(c, d))

在函数名后面用在中括号中使用类型约束，这时候在函数里面K就表示Number这个类型约束，然后可以用在函数的参数列表、函数体、返回值中。

加入需要用到以int或者float32等基本类型为基础的类型时，例如type MyInt int，需要在类型名前面加上~