登录
首页 >  Golang >  Go问答

如何为不同结构的指针实现相同的函数?

来源:stackoverflow

时间:2024-03-22 16:54:33 353浏览 收藏

为具有相同公共字段的结构实现相同函数时,可以通过声明一个包含所需方法的接口来解决问题。然后,让结构实现该接口,并将接口引用传递给函数。此方法允许在不同结构之间共享代码,避免重复编写相同的功能。通过使用反射机制,函数可以处理 nil 指针的情况,从而简化代码并提高可维护性。

问题内容

假设我有很多不同的结构,但它们都共享一个公共字段,例如“名称”。例如:

type foo struct {
    name string
    someotherstring string
    // other fields
}

type bar struct {
    name string
    somenumber int
    // other fields
}

在程序中,我反复遇到这样的情况:我获得了指向这些结构体的指针(例如 *foo、*bar 等),并且需要根据指针是否为 nil 来执行操作,基本上就像这样:

func workonname(f *foo) interface{} {
    if (f == nil) {
        // do lots of stuff
    } else {
        // do lots of other stuff
    }
    // do even more stuff
    return something
}

此函数仅使用 name,在所有结构中都是相同的。如果这些不是指针,我知道我可以为每个返回名称的结构编写一个公共接口并将其用作类型。但对于指针来说,这一切都不起作用。 go 要么在编译时抱怨,要么 nil 检查不起作用并且 go 发生恐慌。我还没有找到比为我拥有的每个结构复制/粘贴完全相同的代码更聪明的方法,所以基本上实现所有功能:

func (f *foo) workonname() interface{}
func (b *bar) workonname() interface{}
func (h *ham) workonname() interface{}
// and so on...

有没有办法做得更好,即只为我的所有结构实现一个简单的函数(甚至更好,根本没有函数),并为所有结构简单地编写一次复杂的东西?

编辑:感谢您到目前为止的回答,但只需使用以下类型的界面:

func (f foo) Name() string {
    return f.name
}

对于某些提供 name() 的接口, 不起作用,因为指针不被识别为 nil。请参阅此演示:https://play.golang.org/p/_d1qizwnme_f


解决方案


您可以声明一个接口,该接口声明一个返回名称的函数:

type withname interface {
    name() string
}

为了实现该接口,您的类型(foobar 等)需要具有该方法 - 不仅仅是字段、方法。

func (f foo) name() string {
    return f.name
}

然后,workonname 需要接收该接口的引用:

func workonname(n withname) interface{} {
    if (n == nil || reflect.valueof(n).isnil()) {
        // do lots of stuff
    } else {
        // do lots of other stuff
    }
    // do even more stuff
    return something
}

请记住,参数 n withname 始终被视为指针,而不是对象值。

我认为 reflect 就是这种情况。

类似于:

package main

import (
    "fmt"
    "reflect"
)

func SetFieldX(obj, value interface{}) {
    v := reflect.ValueOf(obj).Elem()
    if !v.IsValid() {
        fmt.Println("obj is nil")
        return
    }
    f := v.FieldByName("X")
    if f.IsValid() && f.CanSet() {
        f.Set(reflect.ValueOf(value))
    }

}

func main() {
    st1 := &struct{ X int }{10}
    var stNil *struct{}

    SetFieldX(st1, 555)
    SetFieldX(stNil, "SSS")

    fmt.Printf("%v\n%v\n", st1, stNil)
}

https://play.golang.org/p/OddSWT4JkSG

请注意,isvalid 不仅仅检查 obj==nil,但如果您确实想区分 nil 指针 和非结构对象的情况 - 您可以自由地实现它。

今天关于《如何为不同结构的指针实现相同的函数?》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

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