Mgo v2聚合查询如何灵活运用动态条件？

Mgo v2聚合查询动态条件构建技巧

本文介绍如何在MongoDB的mgo v2驱动程序中，灵活构建聚合查询的$match阶段动态条件，实现根据实际情况筛选数据的目的。

下图展示了动态条件构建的流程示意图：

示例代码中，使用bson.m构建$match条件：

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

bank_id，knowledge_points和difficulty字段为可选匹配条件。如果这些字段为空或无效，则不应包含在$match条件中，避免查询结果偏差。

解决方法的核心在于理解bson.m的本质：map[string]interface{}。我们可以动态添加键值对。高效的解决方案如下：

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
}
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
}

这段代码声明map[string]interface{}类型的query变量，根据bankId和difficulty以及knowledge的值，选择性地添加键值对。最后，将query作为$match条件的值，实现动态条件构建。只有当bankId或difficulty或knowledge有有效值时，对应的条件才会添加到查询中，确保查询的灵活性和准确性。 这提升了代码的灵活性和可维护性，允许根据需求动态构建$match条件。