在使用go语言的mgo驱动与mongodb交互时，获取新插入文档的`_id`的最佳实践并非依赖数据库自动生成后查询，而是在客户端使用`bson.newobjectid`手动创建`_id`。这种方法确保了`_id`在插入操作前即可获得，简化了后续操作，并符合mongodb驱动的常见设计模式。
引言：MongoDB _id与mgo驱动
在MongoDB中，每个文档都必须包含一个唯一的_id字段，它作为文档的主键。如果插入文档时未显式指定_id，MongoDB数据库会自动生成一个ObjectId并赋值给该字段。对于Go语言的mgo驱动用户而言，一个常见的问题是：如何在插入操作完成后，获取到这个由数据库自动生成的_id？或者，是否应该在客户端手动生成_id？
核心策略：客户端生成ObjectId
根据MongoDB的官方文档和Go语言mgo驱动的惯例，推荐的做法是在客户端（即应用程序代码中）手动生成_id值，而不是依赖数据库自动生成。这意味着在执行插入操作之前，开发者应使用bson.NewObjectId()方法创建一个新的ObjectId，并将其赋值给待插入文档的_id字段。
为何选择客户端生成？
- 即时可用性：当_id在客户端生成时，它在插入操作执行之前就已经可用。这意味着您无需执行额外的查询来检索数据库生成的_id，从而简化了代码逻辑并提高了效率。
- 符合驱动设计模式：包括mgo在内的许多MongoDB驱动，其设计哲学倾向于让客户端在插入前管理_id。虽然MongoDB服务器能够自动生成_id，但驱动通常会先检查文档是否包含_id，如果没有，它自己也会尝试生成一个。客户端主动生成_id，可以更好地与驱动的行为保持一致。
- 避免不必要的网络往返：如果依赖数据库生成_id，理论上您可能需要进行额外的查询来获取它（尽管某些驱动可能在插入操作的响应中返回）。客户端生成则完全避免了这种潜在的额外开销。
实践示例：使用bson.NewObjectId
以下是一个使用Go语言mgo驱动，并在客户端手动生成ObjectId的示例。
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("文档已清理。")
// }
}
在上述代码中：
- 我们定义了一个MyDocument结构体，其中ID字段的类型是bson.ObjectId，并带有bson:"_id,omitempty"标签。omitempty标签表示如果ID字段为空值（bson.ObjectId的零值），则在编码为BSON时会省略该字段。然而，由于我们手动生成并赋值了ID，它将始终被包含。
- 在插入文档之前，我们调用bson.NewObjectId()来生成一个新的ObjectId。
- 这个newObjectID被直接赋值给doc.ID。
- 执行c.Insert(&doc)后，doc.ID中存储的ObjectId就是该文档在MongoDB中的唯一标识。
注意事项与最佳实践
- 类型匹配：确保您的Go结构体中用于_id的字段类型是bson.ObjectId，这样mgo才能正确地进行BSON编码和解码。
- 唯一性保证：bson.NewObjectId()生成的是一个高度唯一的12字节BSON ObjectId，它包含时间戳、机器标识、进程ID和计数器，确保了全局的唯一性。
- 复合主键：如果您的应用需要复合主键，_id字段也可以是一个包含多个字段的嵌套文档。但对于大多数情况，单个ObjectId作为主键是足够且推荐的。
- 错误处理：始终对MongoDB操作进行错误检查，例如连接失败、插入失败等。
总结
在Go语言中使用mgo驱动与MongoDB交互时，获取新插入文档_id的最佳实践是在客户端使用bson.NewObjectId()手动生成ObjectId，并将其赋值给文档的_id字段，然后再执行插入操作。这种方法不仅简化了代码逻辑，提高了效率，也与MongoDB驱动的常见设计模式保持一致。通过这种方式，您可以在插入文档后立即获得其唯一的标识符，无需额外的数据库往返。
