目前 golang 无法支持对泛型参数空检查,例如下面代码就会报错:
func IsNil[T any](t T) bool {
// invalid operation: t == nil (mismatched types T and untyped nil)
return t == nil
}
解决方式只能把参数 T 改为指针:
// Works now
func IsNil[T any](t *T) bool {
return t == nil
}
但是如果我们需要写一个函数可以把任意参数转化为指针,比如 aws sdk go v2中几乎所有参数都是指针类型,所以提供了一系列的函数,例如取 string 指针:
// String returns a pointer value for the string value passed in.
func String(v string) *string {
return ptr.String(v)
}
如果写一个泛型的取指针的函数就会遇到上面 IsNil 的问题。
如果不使用 *T 参数的话,这里有一个取巧的方式,使用 refelct 包:
func TakePoint[T any](t T) *T {
if reflect.TypeOf(t) == nil {
return nil
}
return &t
}
reflect.TypeOf(i) 如果参数 i 是 nil 那么也会返回 nil,所以可以利用这个特性解决这个问题。
附上一个简单的测试用例
func TestTakePoint(t *testing.T) {
if res := TakePoint[string]("string"); *res != "string" {
t.Fatal("string")
}
if res := TakePoint[int](100); *res != 100 {
t.Fatal("int")
}
if res := TakePoint[any](nil); res != nil {
t.Fatal("fail nil")
}
}
可以关注这个 issue 22729 和 issue 61489 进行跟进。