本文介绍如何利用 mgo 驱动的 MongoDB 聚合管道（$unwind + $match + $project）高效提取嵌套在 Requests 数组中的特定子文档，避免全量加载并实现字段精简投影。
本文介绍如何利用 mgo 驱动的 mongodb 聚合管道（`$unwind` + `$match` + `$project`）高效提取嵌套在 `requests` 数组中的特定子文档，避免全量加载并实现字段精简投影。
在使用 mgo（现已归档，但仍在维护项目中广泛使用）操作 MongoDB 时，若需从嵌套数组（如 Group.Requests）中精确筛选若干子文档（例如按 _id 列表匹配），必须借助聚合管道——因为常规 Find() 无法直接返回“扁平化”的子文档集合，而只能返回完整父文档。
核心思路是三步聚合：
- $unwind：将 Requests 数组展开为独立文档流，使每个 Request 成为一条可单独过滤的记录；
- $match：在展开后的流中，用 {$in: [...]} 精准匹配目标 _id；
- $project：仅保留所需字段（如 _id 和 name），剔除冗余层级与字段，输出纯净结果。
以下是完整的 Go 实现示例（适配 mgo v2）：
import (
"gopkg.in/mgo.v2"
"gopkg.in/mgo.v2/bson"
)
type Request struct {
ID bson.ObjectId `bson:"_id"`
Name string `bson:"name"`
}
func GetRequests(requestIDs []int) ([]Request, error) {
session := // your mgo.Session
defer session.Close()
// 将 int 切片转为 bson.ObjectId（若 _id 是 ObjectId 类型）
// 注意：示例数据中 _id 为整数，实际需根据 schema 调整类型
var ids []interface{}
for _, id := range requestIDs {
ids = append(ids, id) // 若 _id 是 int，直接传入；若是 ObjectId，用 bson.ObjectIdHex()
}
pipe := []bson.M{
{"$unwind": "$Group.Requests"},
{"$match": bson.M{"Group.Requests._id": bson.M{"$in": ids}}},
{"$project": bson.M{
"_id": "$Group.Requests._id",
"name": "$Group.Requests.name",
"_id": 0, // 可选：显式排除原始 _id 字段（避免冲突）
}},
}
collection := session.DB("your_db").C("Groups")
iter := collection.Pipe(pipe).Iter()
var results []Request
var item Request
for iter.Next(&item) {
results = append(results, item)
}
if err := iter.Close(); err != nil {
return nil, err
}
return results, nil
}
⚠️ 关键注意事项：
- 字段路径需准确：因数据结构为 {"Group": {"Requests": [...]}}，$unwind 必须写为 "$Group.Requests"，而非 "$Requests"；
- ID 类型一致性：确保 requestIDs 类型与数据库中 Requests._id 类型严格匹配（整数 vs ObjectId）；
- 性能考量：$unwind 在大数据集上可能产生大量中间文档，建议在 Group.Requests._id 上建立索引（如 { "Group.Requests._id": 1 }）；
- mgo 替代建议：新项目推荐迁移到官方 mongo-go-driver，其聚合 API 更现代、类型安全且持续维护。
通过该方案，输入 [1,2,4] 即可精准返回三条扁平化 Request 记录，完全符合预期输出格式，兼顾效率与可读性。
