Go语言mgo驱动MongoDB：嵌套文档操作与字段映射指南

package main

import (
    "fmt"
    "log"
    "time"

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

// Timer 结构体代表嵌套文档
type Timer struct {
    Start time.Time `bson:"start"`
    End   time.Time `bson:"end"`
}

// Document 结构体代表主文档
type Document struct {
    ID    bson.ObjectId `bson:"_id,omitempty"`
    Name  string        `bson:"name"`
    Timer Timer         `bson:"timer"`
}

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

    session.SetMode(mgo.Monotonic, true)
    c := session.DB("testdb").C("documents")

    // 清理旧数据并插入一个初始文档
    c.RemoveAll(nil)
    docID := bson.NewObjectId()
    initialDoc := Document{
        ID:   docID,
        Name: "示例文档",
        Timer: Timer{
            Start: time.Now().Add(-1 * time.Hour),
            End:   time.Now().Add(-30 * time.Minute),
        },
    }
    err = c.Insert(&initialDoc)
    if err != nil {
        log.Fatalf("插入初始文档失败: %v", err)
    }
    fmt.Printf("插入初始文档，ID: %s\n", docID.Hex())

    // 更新嵌套字段：timer.end
    newEndTime := time.Now()
    err = c.UpdateId(docID, bson.M{"$set": bson.M{"timer.end": newEndTime}})
    if err != nil {
        log.Fatalf("更新嵌套字段失败: %v", err)
    }
    fmt.Printf("成功更新 'timer.end' 为: %v\n", newEndTime)

    // 验证更新结果
    var updatedDoc Document
    err = c.FindId(docID).One(&updatedDoc)
    if err != nil {
        log.Fatalf("查询更新后的文档失败: %v", err)
    }
    fmt.Printf("更新后的文档 Timer 结束时间: %v\n", updatedDoc.Timer.End)
}

    // 假设我们想删除 timer.start 字段
    err = c.UpdateId(docID, bson.M{"$unset": bson.M{"timer.start": ""}})
    if err != nil {
        log.Fatalf("删除嵌套字段失败: %v", err)
    }
    fmt.Println("成功删除 'timer.start' 字段。")

    // 验证删除结果 (查询文档，会发现 Timer.Start 字段变为零值)
    var docAfterUnset Document
    err = c.FindId(docID).One(&docAfterUnset)
    if err != nil {
        log.Fatalf("查询删除字段后的文档失败: %v", err)
    }
    fmt.Printf("删除 'timer.start' 后的文档 Timer: %+v\n", docAfterUnset.Timer)
}

type Timer struct {
    Start time.Time `bson:"start"` // Go的Start字段映射到MongoDB的start字段
    End   time.Time `bson:"end"`   // Go的End字段映射到MongoDB的end字段
}

type Document struct {
    ID    bson.ObjectId `bson:"_id,omitempty"` // _id是MongoDB的默认ID字段，omitempty表示如果为空则不插入
    Name  string        `bson:"name"`
    Timer Timer         `bson:"timer"` // Go的Timer结构体字段映射到MongoDB的timer嵌套文档
}

    // 插入一个带有额外字段的文档，以更好地演示非结构化获取
    extraDocID := bson.NewObjectId()
    extraDoc := bson.M{
        "_id":  extraDocID,
        "name": "动态文档",
        "data": bson.M{
            "value": 123,
            "tags":  []string{"go", "mgo"},
        },
        "timestamp": time.Now(),
    }
    err = c.Insert(&extraDoc)
    if err != nil {
        log.Fatalf("插入额外文档失败: %v", err)
    }
    fmt.Printf("插入额外文档，ID: %s\n", extraDocID.Hex())

    // 查询并获取为 map[string]interface{} 类型
    var rawDoc map[string]interface{}
    err = c.FindId(extraDocID).One(&rawDoc)
    if err != nil {
        log.Fatalf("查询非结构化文档失败: %v", err)
    }
    fmt.Printf("获取的非结构化文档: %+v\n", rawDoc)

    // 访问非结构化文档中的字段
    if name, ok := rawDoc["name"].(string); ok {
        fmt.Printf("文档名称: %s\n", name)
    }

    // 访问嵌套字段
    if dataMap, ok := rawDoc["data"].(bson.M); ok { // 注意：mgo解码嵌套文档为bson.M
        if value, ok := dataMap["value"].(int); ok {
            fmt.Printf("嵌套数据值: %d\n", value)
        }
        if tags, ok := dataMap["tags"].([]interface{}); ok { // 数组元素类型是interface{}
            fmt.Printf("嵌套数据标签: %v\n", tags)
        }
    }
}