Go与mgo：利用bson:",inline"标签扁平化存储嵌入式结构体

package main

import (
    "fmt"
    "encoding/json" // 用于演示JSON序列化行为
)

type Square struct {
    Length int `json:"length"`
    Width  int `json:"width"`
}

type Cube struct {
    Square // 嵌入式结构体
    Depth  int `json:"depth"`
}

func main() {
    c := Cube{
        Square: Square{Length: 2, Width: 3},
        Depth:  4,
    }

    // 使用json.Marshal进行演示，其行为与期望的MongoDB扁平化存储类似
    b, err := json.Marshal(c)
    if err != nil {
        panic(err)
    }
    fmt.Println("JSON Marshal Output:", string(b))
    // 预期输出: JSON Marshal Output: {"length":2,"width":3,"depth":4}
}

// 假设 collection 是 *mgo.Collection 实例
// var c Cube = ...
// err := collection.Insert(&c)
// 或 err := collection.Upsert(bson.M{"_id": someID}, &c)

{
    "length": 2,
    "width": 3,
    "depth": 4
}

{
    "Square": { // 嵌入式结构体名称作为键
        "length": 2,
        "width": 3
    },
    "depth": 4
}

type Square struct {
    Length int `bson:"length"` // 显式指定BSON字段名
    Width  int `bson:"width"`  // 显式指定BSON字段名
}

type Cube struct {
    Square `bson:",inline"` // 添加inline标签，扁平化Square的字段
    Depth  int    `bson:"depth"` // 显式指定BSON字段名
}

package main

import (
    "fmt"
    "log"
    "time"

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

// Square 定义一个基础的二维形状
type Square struct {
    Length int `bson:"length"` // 显式指定BSON字段名
    Width  int `bson:"width"`  // 显式指定BSON字段名
}

// Cube 定义一个三维立方体，嵌入Square
type Cube struct {
    ID     bson.ObjectId `bson:"_id,omitempty"` // MongoDB文档ID，omitempty表示如果为空则不存储
    Square `bson:",inline"`                       // 使用inline标签扁平化Square的字段
    Depth  int           `bson:"depth"`         // 显式指定BSON字段名
}

func main() {
    // 1. 连接MongoDB
    session, err := mgo.Dial("mongodb://localhost:27017")
    if err != nil {
        log.Fatalf("Failed to connect to MongoDB: %v", err)
    }
    defer session.Close()

    // 可选：设置连接模式和超时
    session.SetMode(mgo.Monotonic, true)
    session.SetSyncTimeout(1 * time.Second)
    session.SetSocketTimeout(5 * time.Second)

    // 获取数据库和集合
    c := session.DB("testdb").C("cubes")

    // 2. 创建一个Cube实例
    cubeToInsert := Cube{
        ID:     bson.NewObjectId(), // 生成一个新的ObjectId
        Square: Square{Length: 2, Width: 3},
        Depth:  4,
    }

    // 3. 插入或更新文档
    // 使用Upsert可以根据_id插入或更新文档
    _, err = c.Upsert(bson.M{"_id": cubeToInsert.ID}, &cubeToInsert)
    if err != nil {
        log.Fatalf("Failed to upsert cube: %v", err)
    }
    fmt.Printf("Successfully upserted cube with ID: %s\n", cubeToInsert.ID.Hex())

    // 4. 从MongoDB中查询并验证
    var retrievedCube Cube
    err = c.Find(bson.M{"_id": cubeToInsert.ID}).One(&retrievedCube)
    if err != nil {
        log.Fatalf("Failed to retrieve cube: %v", err)
    }
    fmt.Printf("Retrieved Cube: %+v\n", retrievedCube)
    fmt.Printf("Length: %d, Width: %d, Depth: %d\n", retrievedCube.Length, retrievedCube.Width, retrievedCube.Depth)

    // 预期在MongoDB中看到的文档结构（使用mongo shell查询）：
    // 例如：db.cubes.findOne({"_id": ObjectId("...")})
    /*
    {
        "_id": ObjectId("65c8e3b3e2a7f8c1d0e5f6a7"), // 示例ID
        "length": 2,
        "width": 3,
        "depth": 4
    }
    */
}