使用Go函数创建泛型类型的新指针或实例

在 Go 中，泛型函数无法直接返回指针或实例化泛型类型。因此，当函数接受泛型类型并需要返回一个始终返回指针的函数时，需要特殊处理。一种方法是使用反射来检查输入类型是否为指针，并根据需要返回指向该类型或基本类型的指针。但是，这种方法会影响性能，因此需要仔细权衡。另一种方法是使用条件类型来区分指针和非指针类型，并根据需要使用不同的工厂函数来创建指针或实例化该类型。不过，这种方法需要在工厂函数中使用类型断言，这可能会影响代码的可读性和维护性。

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

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

这就是我正在尝试做的事情：

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())
}

正确答案

这种条件类型不能用泛型很好地解决，因为当您使用 *foo 实例化 t any 时，您会丢失有关基本类型的信息。事实上，您的代码仍然使用反射和 any (= interface{})，并且 maken 函数的返回类型必须类型断言为 *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
    }
}

然后两个生成器函数都会返回一个 any ，它包含一个非零 *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 的实例化发生在编译时。在编译时，t 仅受 any 约束，并且 any (= interface{}) 不实现 ptr[u]
• 仅使用参数 v 无法捕获有关指针类型和基类型的信息。例如，当使用 makenew(foo{}) 和 makenew[t ptr[u], u any](v u) 调用时，makenew[t ptr[u], u any](v t) 将不会编译。当使用 *foo 调用时，将 t 推断为 **foo

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。