mongo mgo v2中 聚合查询 动态条件的实现

在mongo mgo v2中进行聚合查询时，经常需要使用动态条件，即只有在指定条件存在时才会对其进行匹配。本文将介绍如何实现此需求。

原始代码

where := bson.m{"$match": bson.m{"user_id": userid, "bank_id": bankid, "knowledge_points": bson.m{"$in": knowledge}, "difficulty": difficulty}}

问题

上述代码中的knowledge_points和difficulty是可选条件，即只有当它们存在相应值时才会进行匹配。如何仅在存在值时应用这些条件？

解决方案

var query map[string]interface{}
query = make(map[string]interface{})
query["user_id"] = userId
if bankId > 0 {
    query["bank_id"] = bankId
}
if difficulty != "" {
    query["difficulty"] = difficulty
}

where := bson.M{"$match": query}

解释

• 创建一个map[string]interface{}类型的变量query，用于存储最终的查询条件。
• 向query中添加固定参数user_id（前提：userid必须存在）。
• 如果bankid和difficulty有值，则分别向query中添加 "bank_id" 和 "difficulty" 字段。
• 将query赋值给where条件，这样仅在条件存在时才会应用它们。