go语言泛型使用技巧与示例

作者：远客网络发布时间：2024-12-08 16:57:59

go语言泛型怎么用

Go语言在其1.18版本中引入了对泛型的支持，这使得我们可以编写更通用和可重用的代码。1、使用类型参数定义泛型函数，2、使用类型参数定义泛型类型，3、使用类型约束来限制类型参数。以下将详细介绍如何在Go语言中使用这些泛型功能。

一、使用类型参数定义泛型函数

定义泛型函数时，我们可以使用类型参数来表示函数可以接受任何类型的参数。类型参数在函数名后面的方括号中指定。例如，编写一个通用的求和函数：

package main
import "fmt"
func Sum[T int | float64](a, b T) T {
    return a + b
}
func main() {
    fmt.Println(Sum(1, 2))       // 输出: 3
    fmt.Println(Sum(1.5, 2.3))   // 输出: 3.8
}

在这个例子中，T 是类型参数，它可以是 int 或 float64。这使得 Sum 函数可以接受整数或浮点数参数。

二、使用类型参数定义泛型类型

同样，我们可以使用类型参数来定义泛型类型。这使得我们可以创建更通用的数据结构。例如，定义一个通用的栈：

package main
import "fmt"
type Stack[T any] struct {
    elements []T
}
func (s *Stack[T]) Push(element T) {
    s.elements = append(s.elements, element)
}
func (s *Stack[T]) Pop() T {
    if len(s.elements) == 0 {
        var zero T
        return zero
    }
    element := s.elements[len(s.elements)-1]
    s.elements = s.elements[:len(s.elements)-1]
    return element
}
func main() {
    intStack := Stack[int]{}
    intStack.Push(1)
    intStack.Push(2)
    fmt.Println(intStack.Pop())  // 输出: 2
    stringStack := Stack[string]{}
    stringStack.Push("hello")
    stringStack.Push("world")
    fmt.Println(stringStack.Pop())  // 输出: world
}

在这个例子中，Stack 是一个泛型类型，它可以接受任何类型的元素。我们定义了 Push 和 Pop 方法来操作栈中的元素。

三、使用类型约束来限制类型参数

有时，我们希望限制类型参数的类型。这可以通过类型约束来实现。例如，编写一个函数来求数组的最大值，我们希望数组中的元素可以比较大小：

package main
import "fmt"
type Ordered interface {
    ~int | ~float64 | ~string
}
func Max[T Ordered](array []T) T {
    max := array[0]
    for _, v := range array {
        if v > max {
            max = v
        }
    }
    return max
}
func main() {
    intArray := []int{1, 2, 3, 4}
    fmt.Println(Max(intArray))  // 输出: 4
    floatArray := []float64{1.1, 2.2, 3.3, 4.4}
    fmt.Println(Max(floatArray))  // 输出: 4.4
    stringArray := []string{"apple", "banana", "cherry"}
    fmt.Println(Max(stringArray))  // 输出: cherry
}