本文介绍如何借助 mgo 驱动的 mongodb 聚合管道（aggregation pipeline），从嵌套数组字段中高效提取满足 id 列表条件的子文档，并仅返回所需字段，避免冗余数据传输。
本文介绍如何借助 mgo 驱动的 mongodb 聚合管道（aggregation pipeline），从嵌套数组字段中高效提取满足 id 列表条件的子文档，并仅返回所需字段，避免冗余数据传输。
在使用 mgo（现已归档，但仍在许多遗留 Go 项目中广泛使用）操作 MongoDB 时，若需从嵌套结构（如 Group.Requests 数组）中精确筛选并投影特定子文档（例如按 _id 列表匹配多个请求），直接使用 .Find() 无法跨数组元素做高效过滤——此时聚合管道是标准且高效的解决方案。
核心思路是三步聚合操作：
- $unwind：将 Requests 数组展开为独立文档流，使每个请求成为一条可单独匹配的记录；
- $match：在展开后的文档中，用 $in 匹配目标 requestIDs（注意类型一致性：MongoDB 中 _id 若为 ObjectId，需转换；若为数值型整数，则直接传入 int）；
- $project：重塑输出结构，仅保留 _id 和 name 字段，去除外层 Group 包裹，实现“扁平化请求列表”结果。
以下是完整 Go 函数实现示例（基于 mgo v2）：
func GetRequests(requestIDs []int) ([]map[string]interface{}, error) {
session := getSession() // 假设已封装 session 获取逻辑
defer session.Close()
collection := session.DB("your_db").C("Groups")
// 构建聚合管道
pipeline := []bson.M{
{"$unwind": "$Group.Requests"},
{"$match": bson.M{"Group.Requests._id": bson.M{"$in": requestIDs}}},
{"$project": bson.M{
"_id": "$Group.Requests._id",
"name": "$Group.Requests.name",
"_id": 1, // 显式保留 _id（等价于 true）
}},
}
iter := collection.Pipe(pipeline).Iter()
var results []map[string]interface{}
var doc bson.M
for iter.Next(&doc) {
results = append(results, doc)
}
if err := iter.Err(); err != nil {
return nil, err
}
return results, nil
}
⚠️ 关键注意事项：
- 确保 requestIDs 类型与数据库中 Requests._id 字段类型严格一致（如示例数据中为整数，不可混用字符串 "1"）；
- 若 _id 实际为 ObjectId，需将 []int 替换为 []bson.ObjectId，并用 bson.ObjectIdHex() 转换；
- $unwind 对空数组或 null 字段默认跳过，如需保留，可加 preserveNullAndEmptyArrays: true 选项；
- 生产环境建议添加上下文超时控制（mgo 不原生支持 context，可结合 session.SetSafe(&mgo.Safe{}) 或升级至官方 mongo-go-driver）；
- 返回类型建议定义为结构体（如 type Request struct { ID intbson:"_id"; Name stringbson:"name"}），而非 map[string]interface{}，以提升类型安全与可维护性。
通过该方案，你将获得完全符合预期的扁平化请求列表——无外层容器、无冗余字段、无性能浪费，真正实现“所查即所得”。
