Go Mgo驱动：MongoDB文档_id管理与获取最佳实践

package main

import (
    "fmt"
    "log"
    "time"

    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson" // 导入bson包
)

// 定义一个结构体来映射MongoDB文档
type MyDocument struct {
    ID        bson.ObjectId `bson:"_id,omitempty"` // _id字段，使用bson.ObjectId类型
    Name      string        `bson:"name"`
    Value     int           `bson:"value"`
    CreatedAt time.Time     `bson:"created_at"`
}

func main() {
    // 1. 连接到MongoDB
    session, err := mgo.Dial("mongodb://localhost:27017") // 根据您的MongoDB地址修改
    if err != nil {
        log.Fatalf("无法连接到MongoDB: %v", err)
    }
    defer session.Close() // 确保会话在程序结束时关闭

    // 设置会话模式，例如Monotonic确保读写一致性
    session.SetMode(mgo.Monotonic, true)

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

    // 2. 在客户端手动生成ObjectId
    newObjectID := bson.NewObjectId()

    // 3. 创建文档实例并赋值_id
    doc := MyDocument{
        ID:        newObjectID, // 将生成的ObjectId赋值给_id字段
        Name:      "示例文档",
        Value:     100,
        CreatedAt: time.Now(),
    }

    // 4. 插入文档
    err = c.Insert(&doc)
    if err != nil {
        log.Fatalf("插入文档失败: %v", err)
    }

    // 5. 此时，_id已经可用，可以直接访问
    fmt.Printf("文档已成功插入，其ObjectId为: %s\n", doc.ID.Hex())
    fmt.Printf("生成的ObjectId (与doc.ID相同): %s\n", newObjectID.Hex())

    // 可选：验证文档是否已成功插入并能通过_id检索
    var retrievedDoc MyDocument
    err = c.FindId(newObjectID).One(&retrievedDoc)
    if err != nil {
        log.Fatalf("通过ObjectId检索文档失败: %v", err)
    }
    fmt.Printf("成功检索到的文档: %+v\n", retrievedDoc)

    // 清理（可选）：删除刚刚插入的文档
    // err = c.RemoveId(newObjectID)
    // if err != nil {
    //  log.Printf("删除文档失败: %v", err)
    // } else {
    //  fmt.Println("文档已清理。")
    // }
}