在使用golang的mgo驱动操作mongodb时，获取插入文档的_id的最佳实践是主动使用bson.newobjectid()在插入前生成并赋值。这种方法不仅符合mongodb的设计哲学，还能确保开发者在操作前就能获得并管理文档的唯一标识符，从而简化后续的业务逻辑处理。
理解MongoDB的ObjectId与mgo驱动
MongoDB中的_id字段是每个文档的唯一标识符。如果文档在插入时没有明确指定_id，MongoDB会自动生成一个ObjectId并赋值。然而，官方文档指出：“大多数驱动程序会创建一个ObjectId并插入_id字段，但如果驱动程序或应用程序没有提供，mongod也会创建并填充_id。” 这意味着，对于开发者而言，主动生成_id是一个被鼓励且常见的做法。
对于Golang的mgo驱动，其Collection.Insert()方法在成功插入文档后，并不会直接返回新生成的_id。因此，如果依赖MongoDB自动生成_id，之后再想获取这个_id，就需要执行额外的查询操作，这无疑增加了复杂性和潜在的性能开销。
手动生成ObjectId的实践方法
鉴于上述原因，最佳实践是在应用程序层面，使用bson.NewObjectId()函数预先生成_id，并将其赋值给待插入的文档结构体。这样，在执行插入操作之前，您就已经掌握了文档的唯一标识符。
示例代码
以下是一个如何在Golang中使用mgo驱动手动生成并插入带有ObjectId的文档的示例：
立即学习“go语言免费学习笔记（深入）”；
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())
}
注意事项与总结
- 提前获取ID： 手动生成ObjectId让您在执行插入操作前就拥有了文档的唯一标识符。这对于需要立即引用该ID的后续操作（例如，创建关联文档、日志记录或响应给客户端）至关重要。
- 简化业务逻辑： 避免了插入后再次查询以获取_id的复杂性，使代码更简洁、逻辑更清晰。
- 符合设计哲学： 这种做法与MongoDB驱动程序通常处理_id的方式保持一致。
- omitempty标签： 在结构体字段中使用bson:"_id,omitempty"标签非常重要。omitempty确保如果ID字段在赋值前为空（即bson.ObjectId{}的零值），它将不会被序列化到MongoDB文档中，从而允许MongoDB在必要时自动生成_id。然而，当您手动赋值后，它就会被序列化。
综上所述，当您使用Golang的mgo驱动与MongoDB交互时，主动使用bson.NewObjectId()来生成并管理文档的_id字段是推荐的最佳实践。这不仅提高了开发效率，也确保了数据操作的清晰性和可控性。
