Mgo v2：灵活构建聚合查询的动态条件
本文介绍如何在MongoDB的Mgo v2驱动程序中，为聚合查询构建动态条件，尤其针对可选匹配字段的处理。 我们面临的问题是：一个聚合查询中，部分条件是可选的，只有当这些条件有值时才需要匹配。
问题：原始代码片段
以下代码片段展示了原始的、静态的聚合查询：
where := bson.M{"$match": bson.M{"user_id": userid, "bank_id": bankid, "knowledge_points": bson.M{"$in": knowledge}, "difficulty": difficulty}}
num := bson.M{"$sample": bson.M{"size": questionnum}}
err := repo.collection().Pipe([]bson.M{where, num}).All(&b)
if err != nil {
return nil, err
}
其中user_id是必选条件，bank_id，knowledge_points和difficulty是可选条件。如果这些可选字段为空或无效，则不应该参与匹配。
解决方案：动态条件构建
为了实现动态条件，我们利用bson.M (map[string]interface{}) 的特性，动态添加键值对：
var query bson.M
query = make(bson.M)
query["user_id"] = userId
if bankId > 0 {
query["bank_id"] = bankId
}
if difficulty != "" {
query["difficulty"] = difficulty
}
if len(Knowledge) > 0 {
query["knowledge_points"] = bson.M{"$in": Knowledge}
}
where := bson.M{"$match": query}
num := bson.M{"$sample": bson.M{"size": questionNum}}
err := repo.collection().Pipe([]bson.M{where, num}).All(&b)
if err != nil {
return nil, err
}
这段代码首先创建一个空的bson.M变量query。然后，添加必选条件user_id。 之后，通过if语句判断可选条件是否有效，如有值则添加到query中。最后，query作为$match的参数构建where条件，实现了动态条件的构建。 只有当可选条件有有效值时，才会参与数据库匹配。
通过这种方法，我们有效地避免了冗余的查询条件，并提升了代码的灵活性和可维护性。 这对于处理用户输入或其他动态数据场景下的聚合查询非常实用。
