本文详解如何使用 mongodb 聚合管道（aggregation pipeline）替代简单投影（projection），精准统计 building 文档中符合条件的 floor 数量，避免误返回整个父文档，并提供可直接运行的 go + mgo v2 实战代码。
本文详解如何使用 mongodb 聚合管道（aggregation pipeline）替代简单投影（projection），精准统计 building 文档中符合条件的 floor 数量，避免误返回整个父文档，并提供可直接运行的 go + mgo v2 实战代码。
在使用 Mgo v2 操作 MongoDB 时，一个常见误区是：试图通过 .Select() 投影（如 {"p": {"$elemMatch": ...}}）让查询“只返回子文档”，但实际上 MongoDB 的 find 操作始终以完整文档为单位返回——投影仅控制字段显隐，不会改变返回结构层级。因此，你调用 One(&result) 并传入 *Floor 类型变量时，Go 会尝试将整个 Building JSON 解析进 Floor 结构体，导致字段错位、数据丢失甚至 panic。
要真正实现「统计某 Building 中 floorNumber 等于 fnum 的 Floor 数量」，必须切换到 Aggregation Framework，利用 $size 与 $filter（或 $elemMatch 配合 $project）组合完成精确计数。
✅ 正确方案：使用聚合管道统计匹配 Floor 数量
以下是一个健壮、可复用的 Go 函数示例（兼容 Mgo v2）：
func CountFloorsByNumber(database *mgo.Database, bldgUUID string, fnum int) (int, error) {
// 构建聚合阶段
matchStage := bson.M{
"$match": bson.M{
"b": bldgUUID, // 匹配 Building
"p.l": fnum, // 确保至少有一个 Floor 的 l == fnum（优化索引友好）
},
}
// 使用 $filter 提取所有匹配的 Floor，再用 $size 统计数量
projectStage := bson.M{
"$project": bson.M{
"count": bson.M{
"$size": bson.M{
"$filter": bson.M{
"input": "$p",
"cond": bson.M{"$eq": []interface{}{"$$this.l", fnum}},
},
},
},
},
}
pipe := database.C("buildings").Pipe([]bson.M{matchStage, projectStage})
var result struct {
Count int `bson:"count"`
}
err := pipe.One(&result)
if err != nil {
if err == mgo.ErrNotFound {
return 0, nil // Building 不存在或无匹配 Floor
}
return 0, fmt.Errorf("aggregation failed: %w", err)
}
return result.Count, nil
}
? 关键说明与最佳实践
为什么不用 $elemMatch 在 $match 中？
$elemMatch 用于筛选文档（即只返回满足条件的 Building），但它本身不提供子文档数量信息；而 $filter + $size 才能在文档内精确计数。-
性能提示：
建议为 p.l 字段建立复合索引以加速匹配：database.C("buildings").EnsureIndex(mgo.Index{Key: []string{"b", "p.l"}}) -
替代方案（更简洁）：若只需判断是否存在，推荐 Count() + $elemMatch
count, err := database.C("buildings"). Find(bson.M{"b": bldgUUID, "p": bson.M{"$elemMatch": bson.M{"l": fnum}}}). Count() // count > 0 即表示存在该 Floor 插入前校验逻辑建议：
不必先查后插（易受竞态影响），可直接尝试插入并捕获唯一键冲突（如为 FloorUUID 设置唯一索引），或使用 Upsert + 条件更新。
✅ 总结
- ❌ .Select() + .One(&Floor{}) 无法返回子文档，只会导致解析错误；
- ✅ 使用 Pipe() 调用聚合管道，配合 $filter 和 $size 精准统计嵌套数组中匹配元素数量；
- ✅ 对高频查询字段建立索引，显著提升性能；
- ✅ 根据业务场景权衡：存在性检查优先用 Count()，精确数量需求才用聚合。
此方法稳定、语义清晰，是处理 MongoDB 嵌套数组统计问题的标准实践。
