首页 > 解决方案 > 如何为指向不同结构的指针实现相同的功能?

问题描述

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

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

标签: gostructinterface

解决方案


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

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


推荐阅读