Go 泛型实战:写一个类型安全的通用工具集,告别到处复制 for 循环

Go 泛型实战:写一个类型安全的通用工具集,告别到处复制 for 循环Go 1.18 之前,想写一个「过滤切片」的函数是件憋屈的事:要么给每种类型([]int、[]string、[]User)各写一遍,要么用[]interface{} 反射,牺牲类型安全和性能。Filter、Map、Contains这种在别的语言里一行的东西,Go 里到处是复制粘贴的 for 循环。Go 1.18 的泛型终于解决了这个问题。这篇不堆语法概念,而是带你写一个真正能用的类型安全工具集,顺便把constraints、comparable这些容易含糊的点讲清。先看泛型解决的痛点假设要判断切片里有没有某个元素。没有泛型时你得这么写:// 每种类型都要来一遍,纯体力活funcContainsInt(s[]int,targetint)bool{for_,v:ranges{ifvtarget{returntrue}}returnfalse}funcContainsString(s[]string,targetstring)bool{/* 一模一样,只是类型换了 */}有了泛型,一份代码搞定所有可比较的类型:// T 是类型参数,comparable 约束它必须支持 比较funcContains[T comparable](s[]T,target T)bool{for_,v:ranges{ifvtarget{returntrue}}returnfalse}funcmain(){fmt.Println(Contains([]int{1,2,3},2))// truefmt.Println(Contains([]string{a,b},c))// false// 类型自动推导,不用写 Contains[int](...)}[T comparable]是关键:T是类型参数,comparable是「约束」——它告诉编译器T必须是能用比较的类型。这样函数体里的v target才合法。约束(constraint)到底是什么约束本质就是一个 interface,规定了类型参数「能做什么」。Go 内置了两个最常用的:any:等价于interface{},任何类型都行,但你几乎不能对它做任何操作(不能比较、不能相加);comparable:支持和!,用于 map key、去重、查找这类场景。需要「能比大小 / 能相加」时,用官方扩展包golang.org/x/exp/constraints:importgolang.org/x/exp/constraints// constraints.Ordered 涵盖所有支持 的类型(整数、浮点、字符串)funcMax[T constraints.Ordered](a,b T)T{ifab{returna}returnb}fmt.Println(Max(3,7))// 7fmt.Println(Max(apple,banana))// banana(字符串按字典序)也可以自己写约束。比如「只接受整数」,用|列出允许的类型,~表示「底层类型是这个的也算」:// ~int 意味着 type MyInt int 这种自定义类型也满足约束typeIntegerinterface{~int|~int8|~int16|~int32|~int64}funcSum[T Integer](nums[]T)T{vartotal T// 零值,int 就是 0for_,n:rangenums{totaln}returntotal}~这个符号很容易被忽略但很重要:不加~时,type UserID int定义的UserID不满足int约束;加了~int才把「底层是 int 的所有类型」都纳入。动手:写一个 Map / Filter / Reduce 工具集这三个是函数式处理切片的核心。注意Map有两个类型参数——输入元素类型T和输出元素类型U,因为转换后类型可能变:// Map:把 []T 的每个元素经 f 转成 U,得到 []UfuncMap[T,U any](s[]T,ffunc(T)U)[]U{result:make([]U,len(s))// 预分配,避免 append 反复扩容fori,v:ranges{result[i]f(v)}returnresult}// Filter:保留 predicate 返回 true 的元素funcFilter[T any](s[]T,predicatefunc(T)bool)[]T{result:make([]T,0,len(s))// 容量给满,长度给 0for_,v:ranges{ifpredicate(v){resultappend(result,v)}}returnresult}// Reduce:把切片折叠成单个值,acc 是累加器funcReduce[T,U any](s[]T,init U,ffunc(U,T)U)U{acc:initfor_,v:ranges{accf(acc,v)}returnacc}用起来就很舒服了:nums:[]int{1,2,3,4,5}// 转成字符串切片strs:Map(nums,func(nint)string{returnfmt.Sprintf(#%d,n)})// [#1 #2 #3 #4 #5]// 只留偶数evens:Filter(nums,func(nint)bool{returnn%20})// [2 4]// 求和total:Reduce(nums,0,func(acc,nint)int{returnaccn})// 15一个细节:Map里用make([]U, len(s))直接按长度分配再按下标赋值,比append快,因为省掉了扩容和边界检查。Filter因为不知道最终有几个元素,用make([]T, 0, len(s))预留容量但长度为 0,再append。泛型不是银弹:什么时候别用泛型很香,但有三个地方要克制:1. 方法不能有自己的类型参数。这是 Go 泛型的硬限制。下面这样是编译不过的:typeContainer[T any]struct{items[]T}// 编译错误:方法不能引入新的类型参数 U// func (c Container[T]) Map[U any](f func(T) U) []U { ... }要做类型转换的Map,只能写成上面那样的包级函数,不能挂成方法。这是很多人从别的语言过来最不适应的一点。2. 只有一种类型时别硬上泛型。如果一个函数实际上只处理[]string,写成泛型只是增加认知负担。泛型的价值在「同一逻辑服务多种类型」,没有复用就没有意义。3. 需要运行时类型判断时,泛型帮不上忙。泛型是编译期的,如果你的逻辑要「根据实际类型走不同分支」,那还是得用 interface type switch,泛型做不到。小结泛型用[T 约束]声明类型参数,把「每种类型抄一遍 for 循环」变成一份通用代码,且保持编译期类型安全。约束就是 interface:any(啥都行但啥都不能做)、comparable(能)、constraints.Ordered(能比大小);自定义约束用|列类型、~涵盖底层类型。工具集里Map需要两个类型参数(输入T、输出U);预分配切片(make给长度或容量)比反复append更高效。三条克制原则:方法不能有独立类型参数(转换逻辑写成包级函数)、单一类型别硬用泛型、要运行时分派仍用 interface type switch。一句话记忆:泛型解决的是「同一份逻辑服务多种类型」,约束决定类型参数能做什么,方法里不能再引入新类型参数。