您当前所在位置：首页 > 脚本专栏 > Golang

Golang

一文带你深进理解Golang中的泛型

[db:来源] 2023-05-05Golang2403

Go 在泛型方面一直被诟病，因为它在这方面相对比较落后。但是，在 Go 1.18 版本中，泛型已经被正式引进，成为了 Go 语言中一个重要的特性。本文将会具体介绍 Go 泛型的相关概念，语法和用法，希望能够帮助大家更好地理解和应用这一特性

Go 是一门旨在提高开发效率的语言，其简洁的语法和高效的运行速度让它成为了许多开发者的首选。然而，Go 在泛型方面一直被诟病，因为它在这方面相对比较落后。但是，在 Go 1.18 版本中，泛型已经被正式引进，成为了 Go 语言中一个重要的特性。本文将会具体介绍 Go 泛型的相关概念，语法和用法，希望能够帮助大家更好地理解和应用这一特性。

1. 概述

1.1 什么是泛型

泛型（Generics）是一种编程思想，它答应在编写代码时使用未知的类型。泛型可以增加代码的灵活性和可复用性，同时还能提高代码的安全性和可读性。泛型在 C++, Java 和 Python 等语言中已经被广泛应用，但在 Go 中一直未被支持。

1.2 Go 泛型的背景

在 Go 语言中，由于缺乏泛型，开发者需要为每种类型都编写一个相应的版本，这就导致了代码的冗余和维护成本的提高。同时，这也使得一些常见的算法和数据结构无法实现。因此，Go 社区一直在呼吁加进泛型特性。经过多年的等待和探索，Go 1.18 版本终于加进了泛型特性，这一特性的引进被认为是 Go 语言历史上的一件大事。

1.3 Go 泛型的特点

Go 泛型的特点包括：

基于类型约束的泛型：Go 泛型通过类型约束来实现泛型，这意味着泛型函数或类型可以接受特定的类型。
编译时类型安全：Go 泛型通过编译时类型检查来保证类型安全，这有助于避免运行时错误。
支持多种类型：Go 泛型支持多种类型，包括基本类型和自定义类型。

2. 语法

在 Golang 中，泛型的语法包括类型参数、类型约束、泛型函数和泛型类型等。

2.1 泛型函数

在 Go 中，泛型函数的语法如下：

 func FuncName[T Type](params) returnType {
     // Function body
 }

其中，T 表示泛型类型参数，Type 表示详细的类型，params 表示函数的参数，returnType 表示函数的返归值类型。

例如，下面是一个简朴的泛型函数，它可以接受任意类型的参数，并返归一个切片：

 func toSlice[T any](args ...T) []T {
     return args
 }

在这个例子中，T 表示任意类型，args 表示不定参数，函数返归一个由不定参数构成的切片。在函数调用时，可以传递任何类型的参数，例如：

 strings := toSlice("hello", "world")  // 返归 []string{"hello", "world"}
 nums := toSlice(1, 2, 3)              // 返归 []int{1, 2, 3}

2.2 泛型类型

除了泛型函数之外，Go 1.18 版本还引进了泛型类型。泛型类型的语法如下：

 type TypeName[T Type] struct {
     // Fields
 }

其中，TypeName 表示泛型类型名称，T 表示泛型类型参数，Type 表示详细的类型。

例如，下面是一个泛型栈类型的定义，它可以存储任意类型的数据：

 type Stack[T any] struct {
     data []T
 }
 
 func (s *Stack[T]) Push(x T) {
     s.data = append(s.data, x)
 }
 
 func (s *Stack[T]) Pop() T {
     n := len(s.data)
     x := s.data[n-1]
     s.data = s.data[:n-1]
     return x
 }

在这个例子中，T 表示任意类型，data 是一个存储泛型类型参数 T 的切片，Push 方法可以向栈中添加元素，Pop 方法可以弹出并返归栈顶元素。

在使用泛型类型时，需要指定详细的类型，例如：

 s := Stack[int]{}
 s.Push(1)
 s.Push(2)
 x := s.Pop()  // 返归 2

在这个例子中，我们创建了一个存储整数类型的栈，并向其中添加了两个元素。然后我们弹出栈顶元素，并将其赋值给变量 x。

2.3 泛型约束

在使用泛型时，有时需要对泛型类型入行一定的约束。例如，我们希望某个泛型函数或类型只能接受特定类型的参数，或者特定类型的参数必须实现某个接口。在 Go 中，可以使用泛型约束来实现这些需求。

2.3.1 类型约束

类型约束可以让泛型函数或类型只接受特定类型的参数。在 Go 中，类型约束可以使用 interface{} 类型和类型断言来实现。例如，下面是一个泛型函数，它可以接受实现了 fmt.Stringer 接口的类型：

 func Print[T fmt.Stringer](x T) {
     fmt.Println(x.String())
 }

在这个例子中，T 表示实现了 fmt.Stringer 接口的任意类型，函数接受一个类型为 T 的参数，并调用其 String() 方法输出其字符串表示。

2.3.2 约束语法

类型约束可以使用在类型参数后加上一个约束类型来实现。例如，下面是一个泛型函数，它可以接受实现了 fmt.Stringer 和 io.Reader 接口的类型：

 func Print[T fmt.Stringer, U io.Reader](x T, y U) {
     fmt.Println(x.String())
     _, _ = io.Copy(os.Stdout, y)
 }

在这个例子中，T 和 U 分别表示实现了 fmt.Stringer 和 io.Reader 接口的任意类型，函数接受一个类型为 T 的参数和一个类型为 U 的参数，并调用它们的方法输出其字符串表示和读取数据。

2.3.3 接口约束

除了使用 interface{} 类型入行类型约束之外，Go 还支持使用接口来约束泛型类型。例如，下面是一个泛型类型，它要求其泛型类型参数实现了 fmt.Stringer 接口：

 type MyType[T fmt.Stringer] struct {
     data T
 }
 
 func (m *MyType[T]) String() string {
     return m.data.String()
 }

在这个例子中，T 表示实现了 fmt.Stringer 接口的任意类型，类型 MyType[T] 保存了一个泛型类型参数 T 的值，实现了 fmt.Stringer 接口的 String() 方法。

2.4 泛型特化

泛型特化是指将泛型代码转换为详细类型的代码。在 Go 中，泛型特化是在编译期间完成的。特化可以提高代码的性能和运行效率，因为编译器可以针对详细类型入行优化，避免了运行时的类型检查和类型转换。

在 Go 中，泛型特化是通过代码生成器实现的。代码生成器会根据泛型类型或函数的定义，生成详细类型或函数的代码。例如，下面是一个泛型函数的定义：

 func Swap[T any](a, b *T) {
     *a, *b = *b, *a
 }

该函数可以交换任意类型的两个变量的值。在编译期间，代码生成器会根据调用该函数时传递的参数类型生成详细的函数代码。例如，假如传递的是整数类型的指针，代码生成器会生成以下代码：

 func Swap_int(a, b *int) {
     *a, *b = *b, *a
 }

假如传递的是字符串类型的指针，代码生成器会生成以下代码：

 func Swap_string(a, b *string) {
     *a, *b = *b, *a
 }

2.5 泛型接口

泛型接口是一种可以处理多种类型数据的接口。在 Golang 中，可以使用类型参数来实现泛型接口。

例如：

 type Container[T any] interface {
     Len() int
     Add(T)
     Remove() T
 }

上面的代码中，Container 接口使用类型参数T来表示可以存储的元素类型。该接口包含三个方法，分别用于返归元素个数、添加元素和移除元素。

2.5.1 泛型接口约束

泛型接口约束用于限制实现泛型接口的类型的范围，确保泛型代码只能用于满意特定条件的类型。在 Golang 中，泛型接口约束使用接口来定义。

例如：

 type Stringer interface {
     String() string
 }
 
 type Container[T Stringer] interface {
     Len() int
     Add(T)
     Remove() T
 }

上面的代码中，Container 接口被限制为只能存储实现了 Stringer 接口的类型。

3. 泛型的常用场景

Golang 泛型可以应用于各种数据结构和算法，例如排序、搜索、映射等。下面我们分别以这些场景为例来演示 Golang 泛型的使用。

3.1 排序

在 Golang 中，使用 sort 包可以对任意类型的切片入行排序。为了支持泛型排序，我们可以定义一个泛型函数 Sort[T comparable]，如下所示：

 func Sort[T comparable](s []T) {
     sort.Slice(s, func(i, j int) bool {
         return s[i] < s[j]
     })
 }

在上面的代码中，Sort 函数使用了类型参数 T，并对其入行了约束，要求 T 实现了 comparable 接口。这样可以保证 Sort 函数只接受实现了 comparable 接口的类型参数。使用 sort.Slice 函数对切片入行排序。

下面是使用 Sort 函数对整数切片入行排序的示例代码：

 func main() {
     numbers := []int{3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5}
     Sort(numbers)
     fmt.Println(numbers)
 }

输出结果为：

[1 1 2 3 3 4 5 5 5 6 9]

3.2 搜索

在 Golang 中，使用 search 包可以对任意类型的切片入行搜索。为了支持泛型搜索，我们可以定义一个泛型函数 Search[T comparable]，如下所示：

 func Search[T comparable](s []T, x T) int {
     return sort.Search(len(s), func(i int) bool {
         return s[i] >= x
     })
 }

在上面的代码中，Search 函数使用了类型参数 T，并对其入行了约束，要求 T 实现了 comparable 接口。这样可以保证 Search 函数只接受实现了 comparable 接口的类型参数。使用 sort.Search 函数对切片入行搜索。

下面是使用 Search 函数在整数切片中查找某个值的示例代码：

 func main() {
     numbers := []int{1, 2, 3, 4, 5, 6, 7, 8, 9}
     x := 5
     index := Search(numbers, x)
     fmt.Println(index)
 }

输出结果为：

4

3.3 映射

在 Golang 中，使用 map 类型可以实现映射。为了支持泛型映射，我们可以定义一个泛型函数 Map[K comparable, V any]，如下所示：

 func Map[K comparable, V any](s []K, f func(K) V) map[K]V {
     result := make(map[K]V)
     for _, k := range s {
         result[k] = f(k)
     }
     return result
 }

在上面的代码中，Map 函数使用了类型参数 K 和 V，其中 K 要求实现了 comparable 接口，V 则没有任何限制。使用 make 函数创建一个空的 map 对象，然后使用 for 循环遍历切片，并使用函数 f 对每个元素入行映射，并将结果保存在 map 对象中。

下面是使用 Map 函数将字符串切片中的每个字符串转换为大写字母的示例代码：

 func main() {
     words := []string{"apple", "banana", "cherry", "durian", "elderberry", "fig"}
     uppercased := Map(words, func(word string) string {
         return strings.ToUpper(word)
     })
     fmt.Println(uppercased)
 }

输出结果为：