泛型在 go 中的应用场景:集合操作:创建适用于任何类型的集合操作,例如过滤。数据结构:编写通用的数据结构,如队列,栈和映射,可存储和操作各种类型的数据。算法:编写通用的算法,如排序,搜索和归约,可处理不同类型的数据。
Go 中泛型的具体应用场景
泛型是 Go 1.18 中引入的一项重大特性,它允许我们在类型系统中使用类型变量。这为代码重用和减少冗余提供了强大的功能,特别是在实现通用的数据结构或算法时。
1. 集合操作
使用泛型,我们可以轻松创建适用于任何类型 T 的集合操作。例如:
func Filter[T any](slice []T, pred func(T) bool) []T {
var filtered []T
for _, v := range slice {
if pred(v) {
filtered = append(filtered, v)
}
}
return filtered
}
2. 数据结构
泛型可以帮助我们编写通用的数据结构,例如队列、栈或映射。这些数据结构可以存储和操作各种类型的数据,提供更大的灵活性。
type Queue[T any] struct {
items []T
}
func (q *Queue[T]) Enqueue(item T) {
q.items = append(q.items, item)
}
func (q *Queue[T]) Dequeue() T {
item := q.items[0]
q.items = q.items[1:]
return item
}
3. 算法
泛型还允许我们编写通用的算法,例如排序、搜索或归约操作。这些算法可以处理不同类型的数据,确保代码重用性和可维护性。
func Sort[T any](slice []T) []T {
sort.Slice(slice, func(i, j int) bool { return slice[i] < slice[j] })
return slice
}
实战案例:自定义排序类型
考虑一个包含不同类型数据的 Customer
结构:
type Customer struct {
ID int
Name string
Age int
Score float64
}
我们可以使用泛型编写一个通用的 SortCustomers
函数,根据任意字段对客户进行排序:
func SortCustomers[T comparable](customers []Customer, field T) []Customer {
sort.Slice(customers, func(i, j int) bool {
return customers[i][field] < customers[j][field]
})
return customers
}
该函数可以根据客户 ID、姓名、年龄或分数进行排序,如下所示:
sortedByID := SortCustomers(customers, customers[0].ID)
sortedByName := SortCustomers(customers, customers[0].Name)