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是可选条件。如果这些可选字段为空或无效，则不应该参与匹配。

mongodb-query

用于调试和故障排查，查询MongoDB数据库。用户需要时可以： (1) 列出所有数据库； (2) 列出数据库中的集合； (3) 执行MongoDB查询或命令。

下载

解决方案：动态条件构建

为了实现动态条件，我们利用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条件，实现了动态条件的构建。 只有当可选条件有有效值时，才会参与数据库匹配。

通过这种方法，我们有效地避免了冗余的查询条件，并提升了代码的灵活性和可维护性。 这对于处理用户输入或其他动态数据场景下的聚合查询非常实用。