Go 中如何判断 struct 体是否包含另一个 struct 体

发布于07月20日

我有一个名为Car和Motorbike的 struct ，它们是与其他 struct 的组合.例如:

type HasSeats struct {
  ...
}

type HasDoors struct {
   ...
}

type Car {
   HasSeats
   HasDoors
}

type Motorbike {
   HasSeats
}

我现在有一个Car和Motorbike的实例.找出实例是否属于具有HasSeats和HasDoors的类型的最简单方法是什么？我找到了一种通过reflect模块的方法，但那是完全嘈杂的代码.如有任何帮助，我们将非常感激.

func CheckIfObjectHasDoors(carOrBike interface{}) bool {
    // ... ??
}

推荐答案

声明每个嵌入类型的唯一接口:

type Seats interface{ Seats() *HasSeats }
type Doors interface{ Doors() *HasDoors }

在每个类型上实现接口:

func (x *HasSeats) Seats() *HasSeats { return x }
func (x *HasDoors) Doors() *HasDoors { return x }

使用type assertions可确定合成类型是否包含以下嵌入类型之一:

var mb interface{} = &Motorbike{}
if _, ok := mb.(Seats); ok {
    fmt.Println("mb has seats")
}
if _, ok := mb.(Doors); !ok {
    fmt.Println("mb does not have doors")
}

https://go.dev/play/p/UeXgQoYu5tr个

接口方法被声明为接收方，因为应用程序似乎希望直接访问该值.示例:

type HasSeats struct {
    Color string
}

...

var mb interface{} = &Motorbike{HasSeats: HasSeats{Color: "red"}}
if s, ok := mb.(Seats); ok {
    fmt.Println("mb has seats and the seat color is", s.Seats().Color)
}

https://go.dev/play/p/DsCpoW4HsUI个

如果这不是您想要的，请删除返回值.