Go mongodb 驱动程序忽略聚合中的管道

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《Go mongodb 驱动程序忽略聚合中的管道》，文章讲解的知识点主要包括，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

基于 https://github.com/simagix/mongo-go-examples/blob/master/examples/aggregate_array_test.go 上找到的示例，我创建了下面的代码，该代码在集合上运行管道：

pipeline := `[
    {
        $group: {
            _id: {
                year: { $year: "$action_date" },
                month: { $month: "$action_date" },
                day: { $dayOfMonth: "$action_date" },
            },
            count: { $sum: 1 }
        }
    }
]`

var doc bson.M
var cur *mongo.Cursor
collection := apiContext.MongoClient.Database(apiContext.DatabaseName).Collection(util.ActionCollection)

opts := options.Aggregate()
if cur, err = collection.Aggregate(ctx, MongoPipeline(pipeline), opts); err != nil {
}
defer cur.Close(ctx)
for cur.Next(ctx) {
    cur.Decode(&doc)
    fmt.Println(doc["count"])
}

我没有收到任何错误，但输出不是我所期望的。我获取了集合中的所有文档，而不是根据日期返回文档计数。

我不知道我错过了什么，但我不应该得到分组结果吗？

解决方案

我不知道我错过了什么，但我不应该得到分组结果吗？

这里有几件事需要指出。该代码示例使用 github.com/simagix/keyhole/mdb 模块来表示 MongoPipeline。此函数应该采用字符串格式的 mongodb 管道 $group 并转换为 bson。这是 mongo-go-driver 用于发送到 mongodb 服务器的格式。

不幸的是，mongopipeline 函数不会返回您的代码可以捕获的任何 error，但该函数正在使用 encoding/json 模块将管道转换为 bson，并且存在错误。您收到的 json 解析错误是：

• in有效字符“$”查找对象键字符串的开头 =>因为键需要用“”包裹
• in 查找对象键字符串开头的有效字符“}” => 因为 } 后面有多余的逗号。

由于这些错误，mongopipeline 返回空数组 []，这意味着未定义的聚合管道，并将返回集合中的所有文档。

正确的管道 json 字符串是（您可以使用 jsonlint.com 进行验证）

[{
    "$group": {
        "_id": {
            "year": {
                "$year": "$action_date"
            },
            "month": {
                "$month": "$action_date"
            },
            "day": {
                "$dayofmonth": "$action_date"
            }
        },
        "count": {
            "$sum": 1
        }
    }
}]

根据您的用例，您可以跳过 keyhole/mdb 和 encoding/json 模块，并直接使用 bson。例如：

groupstage := `{
    "$group": {
        "_id": {
            "year": { "$year": "$action_date" },
            "month": { "$month": "$action_date" },
            "day": { "$dayofmonth": "$action_date" }
        },
        "count": { "$sum": 1 }
    }
}`


var group bson.m 
err = bson.unmarshalextjson([]byte(groupstage), false, &group)
if err!=nil {
    fmt.println(err)
}

var doc bson.m 

// you can add another stage i.e. bson.a{group, sort, project}
cursor, err := collection.aggregate(context.background(), bson.a{group})
defer cursor.close(context.background())
if err!=nil {
    log.fatal(err)
}

或者，您也可以直接将 mongo.Pipeline 与 bson.M 或 bson.D 一起使用：

pipeline := mongo.Pipeline{
        {{"$group", bson.D{
                {"_id",   bson.D{
                        {"year", bson.D{{"$year", "$action_date"}}},
                        {"month", bson.D{{"$month", "$action_date"}}},
                        {"day", bson.D{{"$dayOfMonth", "$action_date"}}}
                }},
                {"count", bson.D{{"$sum", 1}}},
        }}},
}

上面的示例是使用当前稳定版 MongoDB Go driver v1.1.3 编写的。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Go mongodb 驱动程序忽略聚合中的管道》文章吧，也可关注golang学习网公众号了解相关技术文章。