理解 mgo 模型中的接口类型

在 MongoDB 和 Go 的上下文中，由于动态特性，使用接口建模数据可能会带来挑战接口。以下是您遇到的问题的简要说明以及建议的解决方案。

接口类型的问题

MongoDB 基于文档的数据模型不提供类型嵌入文档的信息。当使用 mgo 将包含接口类型的 MongoDB 文档解组为 Go 结构体时，mgo 无法确定每个嵌入文档的具体类型。这会导致错误“bson.M 类型的值无法分配给 Node 类型。”

解决方案：包装接口类型

要解决此限制，有一种方法是将接口类型包装在提供类型信息的自定义结构中。这允许 mgo 在解组期间识别嵌入文档的特定类型。

考虑以下示例：

type NodeWithType struct {
   Node Node `bson:"-"`
   Type string
}

type Workflow struct {
   CreatedAt time.Time
   StartedAt time.Time
   CreatedBy string
   Nodes []NodeWithType
}

实现 SetBSON 函数

要完成此解决方案，您需要为 NodeWithType 类型实现 SetBSON 函数。该函数将解码类型字符串，创建相应类型的实例，并对其进行解组。

func (nt *NodeWithType) SetBSON(r bson.Raw) error {
   // Decode the "Type" field and determine the Node type
   var typeStr string
   if err := r.Unmarshal(&typeStr); err != nil {
      return err
   }

   // Create a new instance of the Node type based on the type string
   node, ok := reflect.New(reflect.TypeOf(Node).Elem()).Interface().(Node)
   if !ok {
      return errors.New("invalid Node type")
   }

   // Unmarshal the remaining data into the Node instance
   if err := r.Unmarshal(node); err != nil {
      return err
   }

   // Assign the Node instance to the NodeWithType struct
   nt.Node = node
   return nil
}

结论

利用此模式使您能够有效地利用接口，同时保持解组不同类型的嵌入文档的能力。通过提供明确的类型信息，mgo 可以将这些文档无缝解码为所需的 Go 结构。