泛型(generics)
动手尝试
升级go 1.18
前往官网下载最新版本
需要卸载旧版本再安装新版本
修改VSCODE配置
VSCODE 可能会提示泛型语法只有 go 为 1.18 才可以使用
悬停红色波浪线,选择快速修复,执行提供的命令。重启VSCODE。
由于VSCODE go staticheck 会对go的泛型语法报黄色警告,我们需要修改 VSCODE 拓展设置,搜索 Lint Tool,将默认修改为 golint。
学习泛型
我们可以先参考 Go by Example 给出的例子
package main
import "fmt"
func MapKeys[K comparable, V any](m map[K]V) []K {
r := make([]K, 0, len(m))
for k := range m {
r = append(r, k)
}
return r
}
K,V 两种类型,要求K特征是能比较,V特征是全部,相当于 interface{}
type List[T any] struct {
head, tail *element[T]
}
type element[T any] struct {
next *element[T]
val T
}
List
func (lst *List[T]) Push(v T) {
if lst.tail == nil {
lst.head = &element[T]{val: v}
lst.tail = lst.head
} else {
lst.tail.next = &element[T]{val: v}
lst.tail = lst.tail.next
}
}
MapKeysKV
fmt.Println("keys m:", MapKeys(m))
尝试泛型
我们经常需要程序对数据集合执行操作,例如选择满足给定条件的全部item,或通过自定义函数将全部item映射到一个新的集合。
在其它语言中,通常会使用泛型数据结构和算法。
- Index、Include 方法参数为 comparable 类型,因为comparable特征的类型才可以进行比较
- Any、All、Filter、Map 方法参数为 any 类型,使用了组合函数,可以传入一个方法。
package main
import (
"fmt"
"strings"
)
func Index[T comparable](vs []T, item T) int {
for i, v := range vs {
if v == item {
return i
}
}
return -1
}
func Include[T comparable](vs []T, item T) bool {
return Index(vs, item) > 0
}
func Any[T any](vs []T, f func(T) bool) bool {
for _, v := range vs {
if f(v) {
return true
}
}
return false
}
func All[T any](vs []T, f func(T) bool) bool {
for _, v := range vs {
if !f(v) {
return false
}
}
return true
}
func Filter[T any](vs []T, f func(T) bool) []T {
res := make([]T, 0)
for _, v := range vs {
if f(v) {
res = append(res, v)
}
}
return res
}
func Map[T any, K any](vs []T, f func(T) K) []K {
res := make([]K, len(vs))
for i, v := range vs {
res[i] = f(v)
}
return res
}
func main() {
fmt.Println("Golang 泛型的初步学习")
ints := []int{10, 20, 30, 40, 50}
intCompare := func(t int) bool {
if t > 30 {
return true
}
return false
}
fmt.Println(ints)
fmt.Println("查找20下标", Index(ints, 20))
fmt.Println("是否存在100", Include(ints, 100))
fmt.Println("至少有一个大于30", Any(ints, intCompare))
fmt.Println("全部大于30", All(ints, intCompare))
fmt.Println("筛选大于30", Filter(ints, intCompare))
fmt.Println("放大一倍", Map(ints, func(t int) int { return t * 2 }))
textCompare := func(t string) bool {
if len(t) > 5 {
return true
}
return false
}
fruits := []string{"Apple", "Banana", "Orange"}
fmt.Println("查找Orange下标", Index(fruits, "Orange"))
fmt.Println("是否存在Apple", Include(fruits, "Apple"))
fmt.Println("至少有一个元素长度大于5", Any(fruits, textCompare))
fmt.Println("全部元素长度大于5", All(fruits, textCompare))
fmt.Println("筛选长度大于5", Filter(fruits, textCompare))
fmt.Println("每个元素重复一遍", Map(fruits, func(t string) string { return strings.Repeat(t, 2) }))
newS := Map(fruits, func(t string) string { return strings.Repeat(t, 2) })
fmt.Printf("%p %p", fruits, newS)
}