<p>本文介绍了如何使用 mgo (MongoDB 的 Go 驱动) 将嵌套的 Go 结构体以扁平化的形式存储到 MongoDB 数据库中。通过 <code>bson:",inline"</code> 标签，可以轻松地将嵌套结构体的字段提升到父级结构体中，从而实现期望的数据结构。</p> 在使用 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"`
}

代码示例：

Pub Session Logs

使用 jq 搜索和分析您自己的会话日志。还有 50+ 模型用于图像生成、视频生成、语音合成、语音识别、音乐、聊天等

下载

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 }
}

注意事项：

• bson:",inline" 标签只能用于结构体或 map 类型的字段。
• 如果嵌套结构体中的字段与父级结构体中的字段名称冲突，会导致错误。 为避免冲突，建议显式指定bson标签。
• 确保使用的 mgo 版本支持 inline 标签。 mgo/v1 和 mgo/v2 都支持该标签。
• 如果需要更复杂的自定义序列化/反序列化逻辑，可以考虑实现 bson.Setter 和 bson.Getter 接口。

总结：

通过使用 bson:",inline" 标签，我们可以方便地将嵌套结构体的字段提升到父级结构体中，从而简化 MongoDB 中的数据结构。这在处理复杂的数据模型时非常有用，可以提高代码的可读性和可维护性。 在使用该标签时，请注意字段名称冲突和 mgo 版本兼容性等问题。