登录
首页 >  Golang >  Go问答

如何编写一个允许指向多个基元类型的通用 Go 函数?

来源:stackoverflow

时间:2024-02-21 12:09:21 138浏览 收藏

一分耕耘,一分收获!既然都打开这篇《如何编写一个允许指向多个基元类型的通用 Go 函数?》,就坚持看下去,学下去吧!本文主要会给大家讲到等等知识点,如果大家对本文有好的建议或者看到有不足之处,非常欢迎大家积极提出!在后续文章我会继续更新Golang相关的内容,希望对大家都有所帮助!

问题内容

我正在尝试使用 go 泛型编写一个函数来减少代码中的一些样板 if/else 块。我想出了一些适用于单个类型参数的方法,如下所示:

func valueornil[t *int](value t) any {
    if value == nil {
        return nil
    }
    return *value
}

虽然这工作正常,但它并不是真正有用,因为它只允许 *int,并且我希望此代码可以与任何原始类型一起使用。我尝试扩展它以支持第二种类型,如下所示:

func valueornil[t *int | *uint](value t) any {
    if value == nil {
        return nil
    }
    return *value
}

但是,此变体失败并出现编译器错误:

invalid operation: pointers of value (variable of type T constrained by *int|*uint) must have identical base types

任何人都可以发现我在这里做错了什么,或者类似的事情只是“不支持”?


正确答案


问题似乎在于您试图通过指向类型的指针而不是类型本身实现通用。如果我们将指针移动到参数本身而不是类型参数上,它就可以工作。

解释如下,但这是工作代码:

func valueornil[t ~int | ~uint](value *t) t {
    if value == nil {
        var zero t
        return zero
    }
    return *value
}

所以代替这个(这不起作用):

func valueornil[t *int | *uint](value t) any

你可以这样做:

func valueornil[t int | uint](value *t) any

但是,您可能想更进一步并处理底层类型:

func valueornil[t ~int | ~uint](value *t) any

这将允许自定义类型与函数一起使用:

type thing int

var thing thing
println(valueornil(thing))

您可能需要考虑的另一个方面是返回类型的通用性。您可以使用相同的 t 参数来执行此操作。

例如:

func valueornil([t ~int | ~uint](value *t) t

但这意味着您需要更改部分实现。而不是这个:

if value == nil {
    return nil
}

你可以这样做:

if value == nil {
    var zero t
    return zero
}

来自 Miquellaanswer I accepted (感谢您的见解;我也了解了 ~ 的用法)让我意识到,一旦我将 * 移动到参数类型,因此 this is what I ended up with

func valueOrNil[T any](value *T) any {
    if value == nil {
        return nil
    }
    return *value
}

以上就是《如何编写一个允许指向多个基元类型的通用 Go 函数?》的详细内容,更多关于的资料请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>