go - 修改任何具有特定字段的结构的函数

Question

我有几个结构，它们继承了一些基本结构。像这样:

type s1 struct {
    a string `json:"a"`
    b string `json:"b"`
}

type s2 struct {
    s1
    c string `json:"c"`
    d string `json:"d"`
}

type s3 struct {
    s1
    c string `json:"c"`
    d string `json:"d"`
    e string `json:"d"`
    f string `json:"d"`
}

现在我需要定义一个函数，它对任何具有字段a、b 的结构进行操作。有点像

func modifyStruct(s *s1) {
    s.a, s.b = s.b, s.a
}

但必须在 s2、s3 和任何其他继承 s1 的结构上工作。我试图通过一个界面来实现这一点，但到目前为止还没有成功。有什么办法可以做到这一点？模板在go-playground .

Answer 1

一般的解决方案是使用反射，如 Cerise Limón 的回答所示。使用反射的缺点是您必须导出字段，而且它比应该或可能的要慢。

在您的示例中，尽管函数采用 *s1 类型的值完全没问题且足够，因为嵌入 s1 的所有类型的值都是 s1(明确地)。非限定类型名称(没有包名称)充当嵌入字段的字段名称:

s2 := s2{s1: s1{"A", "B"}}
fmt.Println(s2)
modifyStruct(&s2.s1)
fmt.Println(s2)

输出(在 Go Playground 上尝试):

{{A B}  }
{{B A}  }

如果您仍然希望它接受“任何”(某些)类型的值(这样您就不必引用嵌入的 s1 字段)，但不想导出字段，那么您可以为此使用接口(interface)。使用接口(interface)，您可以保留两种解决方案的优点(保持快速、灵活，并且您不必导出字段):

type S1 interface {
    AB() (string, string)
    SetAB(a, b string)
}

type s1 struct {
    a string `json:"a"`
    b string `json:"b"`
}

func (s s1) AB() (string, string) { return s.a, s.b }
func (s *s1) SetAB(a, b string)   { s.a, s.b = a, b }

func modifyStruct(s S1) {
    a, b := s.AB()
    s.SetAB(b, a)
}

测试它:

s2 := s2{s1: s1{"A", "B"}}
fmt.Println(s2)
modifyStruct(&s2)
fmt.Println(s2)

输出相同(在 Go Playground 上尝试):

{{A B}  }
{{B A}  }

请注意(除了 *s1 类型本身)任何嵌入 *s1 的结构类型(以及任何指向结构类型的指针)自动(隐式)实现 S1 接口(interface)，类似地，任何指向嵌入 s1 的结构类型的指针也实现了 S1(因此 *s2 和 *s3 在你的例子中)。