Golang mgo驱动与MongoDB：管理文档ObjectId的最佳实践

package main

import (
	"fmt"
	"log"
	"time"

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

// User 定义用户结构体，_id字段使用bson.ObjectId类型
type User struct {
	ID        bson.ObjectId `bson:"_id,omitempty"` // _id字段，omitempty表示如果为空则不序列化
	Name      string        `bson:"name"`
	Email     string        `bson:"email"`
	CreatedAt time.Time     `bson:"created_at"`
}

func main() {
	// 连接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)

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

	// 1. 创建一个新的用户实例
	newUser := User{
		Name:      "Alice Smith",
		Email:     "alice@example.com",
		CreatedAt: time.Now(),
	}

	// 2. 手动生成ObjectId并赋值给_id字段
	newUser.ID = bson.NewObjectId()
	fmt.Printf("Generated ObjectId before insert: %s\n", newUser.ID.Hex())

	// 3. 插入文档
	err = c.Insert(&newUser)
	if err != nil {
		log.Fatalf("Failed to insert user: %v", err)
	}

	fmt.Printf("User inserted successfully with ID: %s\n", newUser.ID.Hex())

	// 验证：尝试根据ID查找刚刚插入的文档
	var retrievedUser User
	err = c.FindId(newUser.ID).One(&retrievedUser)
	if err != nil {
		log.Fatalf("Failed to retrieve user by ID: %v", err)
	}
	fmt.Printf("Retrieved user: %+v\n", retrievedUser)

	// 示例：插入一个不手动生成_id的文档 (不推荐)
	// var anotherUser = struct {
	// 	Name string `bson:"name"`
	// 	Age  int    `bson:"age"`
	// }{
	// 	Name: "Bob Johnson",
	// 	Age:  30,
	// }
	// err = c.Insert(&anotherUser)
	// if err != nil {
	// 	log.Fatalf("Failed to insert another user: %v", err)
	// }
	// // 此时，anotherUser中并没有_id字段，无法直接获取MongoDB生成的ID
	// // 需要通过查询来获取，例如：
	// // var tempUser struct {
	// // 	ID bson.ObjectId `bson:"_id"`
	// // 	Name string `bson:"name"`
	// // 	Age  int    `bson:"age"`
	// // }
	// // c.Find(bson.M{"name": "Bob Johnson"}).One(&tempUser)
	// // fmt.Printf("Another user inserted with ID (retrieved via query): %s\n", tempUser.ID.Hex())
}