使用 mgo 存储嵌套结构体

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

如果我们直接将 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"`
}

代码示例：

网奇.NET网络商城系统

系统优势： 1、 使用全新ASP.Net＋c#和三层结构开发. 2、 可生成各类静态页面（html,htm,shtm,shtml和.aspx) 3、 管理后台风格模板自由选择，界面精美 4、 风格模板每月更新多套，还可按需定制 5、 独具的缓存技术加快网页浏览速度 6、 智能销售统计，图表分析 7、 集成国内各大统计系统 8、 多国语言支持，内置简体繁体和英语 9、 UTF-8编码，可使用于全球

下载

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 版本兼容性等问题。