本文介绍如何避免冗长的 if-else 分支，通过动态拼接 bson.m 查询映射来灵活、简洁、可维护地实现 mongodb 的条件匹配。核心是利用 map[string]interface{} 的动态特性，在运行时按需添加过滤字段。
本文介绍如何避免冗长的 if-else 分支，通过动态拼接 bson.m 查询映射来灵活、简洁、可维护地实现 mongodb 的条件匹配。核心是利用 map[string]interface{} 的动态特性，在运行时按需添加过滤字段。
在实际开发中，尤其是构建聚合视图（如按小时统计日志数据）时，常需支持多维度、可选的查询参数（如 site、size、network、region、http_request_status 等）。若采用原始方式——为每种参数组合编写独立的 if 分支（如问题中多达 30+ 种情况），不仅代码臃肿、难以阅读，更严重的是极易引入逻辑遗漏、维护成本极高，且违背“开闭原则”。
幸运的是，bson.M 本质就是 map[string]interface{} 的类型别名（见 mgo 官方文档），天然支持动态键值插入。因此，最佳实践是：初始化一个空的 bson.M 映射，仅在参数非空/有效时追加对应条件，最后统一传入 $match 阶段。
以下是重构后的推荐写法（兼容原逻辑，含关键优化）：
func GetAllHourly(dbsession *mgo.Session, year, month, day, site, size, network, region string, code int) (items []MassAggregation, err error) {
defer dbsession.Close()
// 1. 构建动态查询条件映射
query := bson.M{
"year": year,
"month": month,
"day": day,
}
// 2. 按需添加可选条件（顺序无关，语义清晰）
if site != "" {
query["site"] = site
}
if size != "" {
query["size"] = size
}
if region != "" {
query["aws_region"] = region
}
if code != -1 {
query["http_request_status"] = code
}
if network != "" {
// 兼容旧逻辑：openx3 → openx
if network == "openx3" {
network = "openx"
}
query["network"] = &bson.RegEx{Pattern: network, Options: "i"}
}
// 3. 组装聚合管道：单个 $match + $group
pipeline := []bson.M{
{"$match": query},
{
"$group": bson.M{
"_id": bson.M{
"aws_region": "$aws_region",
"http_request_status": "$http_request_status",
"hour": "$hour",
"network": "$network",
"site": "$site",
"size": "$size",
"zone": "$zone",
"extra": "$extra",
},
"total": bson.M{"$sum": "$count"},
},
},
}
// 4. 执行聚合查询
err = dbsession.DB("logs").C("prod").Pipe(pipeline).All(&items)
return
}
✅ 优势总结：
- 极简可读：从 30+ 行嵌套 if-else 缩减为 10 行清晰的条件判断；
-
零遗漏风险：每个参数独立判断，无需枚举所有组合，新增字段只需一行
if xxx != "" { query["xxx"] = xxx }； -
易于扩展：支持任意新增过滤字段（如增加
zone或extra参数），无需重构主干逻辑； -
性能无损：MongoDB 会自动优化单个
$match中的复合条件，与拆分为多个$match阶段相比，通常更高效（减少管道阶段开销）； -
语义明确：
query变量直观表达“本次查询的完整筛选条件”，符合直觉。
⚠️ 注意事项：
- 正则匹配（
&bson.RegEx）仍需保留，但建议对network做预处理（如大小写归一化或白名单校验），避免注入风险； - 若后续需支持范围查询（如
code >= 400）或数组匹配（如size in ["300x250", "728x90"]），可在对应if块内使用bson.M{"$gte": ...}或bson.M{"$in": [...]}等操作符，bson.M完全兼容； - 使用
mgo时请注意其已归档，生产环境建议迁移到官方驱动go.mongodb.org/mongo-driver/mongo，其bson.D/bson.M同样支持动态构建。
这种“声明式条件组装”模式，是构建高可维护性 MongoDB 查询层的通用范式，值得在各类聚合接口中推广使用。
