瞭解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 結構。