Mgo v2 MongoDB聚合查询：如何灵活处理可选查询条件？

本文探讨mgo v2库在MongoDB聚合查询中如何灵活处理可选查询条件。 问题在于如何动态构建$match阶段的查询条件，使其根据参数是否存在而调整。

原始代码使用bson.M直接构建$match条件，导致可选条件无法灵活控制。改进的关键在于利用map[string]interface{}的特性，动态构建查询条件。

问题： 如何修改以下代码，使knowledge_points和difficulty条件成为可选的？

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)

解决方案： 使用map[string]interface{}动态构建查询条件。

query := make(map[string]interface{})
query["user_id"] = userId
query["bank_id"] = bankId // bankId始终存在

if len(knowledge) > 0 {
    query["knowledge_points"] = bson.M{"$in": knowledge}
}
if difficulty != "" {
    query["difficulty"] = difficulty
}

where := bson.M{"$match": query}
num := bson.M{"$sample": bson.M{"size": questionNum}}

err := repo.collection().Pipe([]bson.M{where, num}).All(&b)

通过此方法，我们先创建一个空的map[string]interface{}，然后根据knowledge和difficulty的值，有选择地添加键值对。 最终，query只包含存在的条件，从而实现了动态条件的添加，提升了代码的灵活性和可维护性。 bank_id 因为始终存在，所以直接添加到query中。 请注意，变量名已调整为更符合Go语言规范的命名方式 (例如，knowledge 变为 knowledge，questionnum 变为 questionNum)。