使用 mgo 存储嵌套结构体_技术教程

本文介绍了如何使用 mgo (mongodb 的 go 驱动) 将嵌套的 go 结构体以扁平化的形式存储到 mongodb 数据库中。通过 bson:",inline" 标签，可以轻松地将嵌套结构体的字段提升到父级结构体中，从而实现期望的数据结构。

在使用 mgo 操作 MongoDB 数据库时，经常会遇到需要存储嵌套结构体的情况。默认情况下，mgo 会将嵌套结构体以嵌套文档的形式存储在 MongoDB 中，但有时我们希望将嵌套结构体的字段直接提升到父级文档中，以简化数据结构和查询。例如，我们有以下结构体： ```go type Square struct { Length int Width int } type Cube struct { Square Depth int }

如果我们直接将 Cube 类型的对象存储到 MongoDB 中，会得到类似如下的文档结构：

{
    "Square": {
        "Length": 2,
        "Width": 3
    },
    "Depth": 4
}

但我们希望得到如下的扁平化结构：

{
    "Length": 2,
    "Width": 3,
    "Depth": 4
}

为了实现这个目标，我们可以使用 bson:",inline" 标签。这个标签告诉 mgo，将嵌套结构体的字段直接嵌入到父级结构体中。

修改后的结构体定义如下：

type Square struct {
    Length int `bson:"Length"`
    Width  int `bson:"Width"`
}

type Cube struct {
    Square `bson:",inline"`
    Depth  int    `bson:"Depth"`
}

代码示例：

package main

import (
    "fmt"
    "log"

    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
)

type Square struct {
    Length int `bson:"Length"`
    Width  int `bson:"Width"`
}

type Cube struct {
    Square `bson:",inline"`
    Depth  int    `bson:"Depth"`
}

func main() {
    session, err := mgo.Dial("localhost")
    if err != nil {
        panic(err)
    }
    defer session.Close()

    // Optional. Switch the session to a monotonic behavior.
    session.SetMode(mgo.Monotonic, true)

    c := session.DB("test").C("cubes")

    err = c.DropCollection()
    if err != nil && err.Error() != "ns not found" {
        log.Fatal("Error dropping collection:", err)
    }

    cube := Cube{
        Square: Square{
            Length: 2,
            Width:  3,
        },
        Depth: 4,
    }

    err = c.Insert(&cube)
    if err != nil {
        log.Fatal(err)
    }

    result := Cube{}
    err = c.Find(bson.M{"Depth": 4}).One(&result)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println(result) // Output: {{2 3} 4}

    // Check the document in MongoDB
    // You should see: { "_id" : ObjectId("..."), "Length" : 2, "Width" : 3, "Depth" : 4 }
}