Go函数创建泛型的新指针或新值

Question

我有一个采用泛型类型的函数，应该 return 一个始终 return 指针的函数。 IE。如果你传递给它一个非指针类型它应该 return 一个指向那个类型的指针，如果你传递给它一个指针类型它应该 return 相同的类型。我不想使用 reflect.New，因为它是一个性能关键型应用程序。

我不介意在 return 作为工厂函数的函数中使用反射，但理想情况下甚至不在那里。

这就是我想要做的：

package main

import (
    "fmt"
    "reflect"
)

type Ptr[T any] interface {
    *T
}

func makeNewA[T Ptr[U], U any]() any {
    return new(U)
}

func makeNewB[T any]() any {
    return new(T)
}

func makeNew[T any](v T) func() any {
    if reflect.TypeOf(v).Kind() == reflect.Ptr {
        return makeNewA[T] // <-- error: T does not match *U
    } else {
        return makeNewB[T]
    }
}

type Foo struct{}

func main() {
    make1 := makeNew(Foo{})
    make2 := makeNew(&Foo{})

    // should both return &Foo{}
    fmt.Println(make1())
    fmt.Println(make2())
}

Answer 1

这种条件类型不能用泛型很好地解决，因为当您用 *Foo 实例化 T any 时，您会丢失有关基类型的信息。事实上，您的代码仍然使用反射和 any (= interface{})，并且 makeN 函数的 return 类型必须是 type-asserted 才能*Foo.

您当前的代码最接近的是：

func makeNew[T any](v T) func() any {
    if typ := reflect.TypeOf(v); typ.Kind() == reflect.Ptr {
        elem := typ.Elem()
        return func() any {
            return reflect.New(elem).Interface() // must use reflect
        }
    } else {
        return func() any { return new(T) } // v is not ptr, alloc with new
    }
}

然后两个 maker 函数将 return 一个 any 包装一个 non-nil *Foo 值：

fmt.Printf("%T, %v\n", make1(), make1()) // *main.Foo, &{}
fmt.Printf("%T, %v\n", make2(), make2()) // *main.Foo, &{}

游乐场：https://gotipplay.golang.org/p/kVUM-qVLLHG

进一步考虑：

return makeNewA[T] 在您的第一次尝试中不起作用，因为条件 reflect.TypeOf(v).Kind() == reflect.Ptr 是在运行时计算的，而 makeNewA 的实例化发生在 compile-time。 compile-time T 仅受 any 约束，而 any (= interface{}) 未实现 Ptr[U]
您无法仅通过参数 v 捕获有关指针类型和基类型的信息。例如 makeNew[T Ptr[U], U any](v T) 在用 makeNew(Foo{}) 调用时不会编译，而 makeNew[T Ptr[U], U any](v U) 在用 *Foo[=49= 调用时会将 T 推断为 **Foo ]

Go函数创建泛型的新指针或新值

Go function to create either new pointer or new value of a generic type

generics

pointers

go