如何为指向不同结构的指针实现相同的功能?

时间:2019-09-09 12:34:18

标签: go struct interface

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

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,在所有结构中均相同。如果这些不是指针,我知道我可以为返回名称的每个结构编写一个通用接口并将其用作类型。但是使用指针,这些都不起作用。要么在编译时抱怨,要么nil检查不起作用,从而导致恐慌。我发现没有比复制/粘贴我拥有的每个结构完全相同的代码更聪明的方法了,因此基本上可以实现所有功能:

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

2 个答案:

答案 0 :(得分:2)

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

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始终被视为指针,而不是对象值。

答案 1 :(得分:0)

我认为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 pointers和非结构对象的情况-您可以自由地实现它。