mongodb 聚合查询：实现动态条件匹配

mongodb 中的聚合查询允许对文档进行复杂的数据操作和聚合。在使用聚合查询时，可以选择使用 $match 操作符来根据指定条件筛选文档。

问题说明

在聚合查询中，有时需要匹配可选条件。例如，"knowledge_points" 和 "difficulty" 字段可能是可选条件，仅当它们具有值时才需要进行匹配。

代码示例

mongodb-query

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

下载

以下代码展示了如何使用 bson.m 类型来实现动态条件匹配：

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}

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{} 类型的查询 map。然后，我们可以根据需要将不同的条件添加到查询 map 中。例如，如果 bankid 大于 0，则将 "bank_id" 添加到查询 map 中。

接下来，我们使用查询 map 构建 where 查询，并将它与采样查询 num 组合起来。这样，我们在进行聚合查询时，只会根据查询 map 中存在的条件进行匹配。

通过这种方式，我们可以动态地添加和移除条件，从而实现强大的可配置查询。