如何使用 golang 在 mongodb 中使用 $lookup 获取计数值?

How to get the count value using $lookup in mongodb using golang?

使用聚合,我正在使用 golang 加入 mongodb 的两个集合。结果如下所示:-

输出:-

{
"response": {
    "code": 1,
    "api_status": 1,
    "message": "Success",
    "total_record": [
        {
            "_id": 1,
            "author_name": "mohit",
            "category": 232,
            "content": "This is the content",
            "date_time": 1524632713,
            "excerpt": "This is a  short text",
            "image": "pic.jpg",
            "resultField": [
                {
                    "_id": 6,
                    "comment": "this is a least comment",
                    "comment_on": 1524644601,
                    "email": "puneet@bookingkoala.com",
                    "name": "puneet",
                    "post_id": 1,
                    "reply_to": 1524644601,
                    "status": 1
                },
                {
                    "_id": 7,
                    "comment": "this is a least comment",
                    "comment_on": 1524647808,
                    "email": "puneet@bookingkoala.com",
                    "name": "puneet",
                    "post_id": 1,
                    "reply_to": 1524647808,
                    "status": 1
                }
            ],
            "status": 0,
            "tags": "this",
            "title": "how to do the code"
        },
        {
            "_id": 2,
            "author_name": "mohit",
            "category": 232,
            "content": "This is the content",
            "date_time": 1524632713,
            "excerpt": "This is a  short text",
            "image": "pic.jpg",
            "resultField": [
                {
                    "_id": 8,
                    "comment": "this is a least comment",
                    "comment_on": 1524648059,
                    "email": "puneet@bookingkoala.com",
                    "name": "puneet",
                    "post_id": 2,
                    "reply_to": 1524648059,
                    "status": 1
                }
            ],
            "status": 0,
            "tags": "this",
            "title": "how to do the code"
        },
        {
            "_id": 3,
            "author_name": "puneet",
            "category": 2,
            "content": "this is content",
            "date_time": 1524641086,
            "excerpt": "this is excerpt",
            "image": "pic.jpg",
            "resultField": [],
            "status": 1,
            "tags": "go",
            "title": "how to do the code"
        }
    ]
 }
}

这个输出被下面的golang代码获取:-

mongoSession := config.ConnectDb()
collection := mongoSession.DB(config.Database).C(config.BlogCollection)
pipeline := []bson.M{
    bson.M{"$match": bson.M{}},
    bson.M{"$lookup": bson.M{"from" : "comment", "localField" : "_id", "foreignField": "post_id","as": "resultField" }},
}
fmt.Println(pipeline)
pipe := collection.Pipe(pipeline)
resp := []bson.M{}
err = pipe.All(&resp)
if err != nil {
    fmt.Println("Errored: %#v \n", err)
}
fmt.Println(resp)
if err != nil {
    response = ResponseControllerList{
        config.FailureCode,
        config.FailureFlag,
        config.FailureMsg,
        nil,
        nil,
    }
} else {
    response = ResponseControllerList{
        config.SuccessFlag,
        config.SuccessFlag,
        config.SuccessMsg,
        nil,
        resp,
    }
}

问题是:- 我只需要 datacount 而不是 data。意味着在如上所示的输出中有显示数据的 resultField 但我只需要 count 值,如:- "resultField":[2] 但它正在显示数据。我怎样才能得到输出数据的计数值。提前谢谢你。

所以你的聚合实际上 returns resultField 字段中的所有 comment 文档,它隐式包含结果的数量,它是一个切片,你可以在 Go 中检查长度使用内置 len() 函数。

由于您只需要长度(comment 文档的数量),这就是您只想检索此数组大小的原因。为此,您可以使用 $addFields 阶段用作为该数组长度的数字替换 resultField 数组。

pipe := c.Pipe([]bson.M{
    {
        "$lookup": bson.M{
            "from":         "comment",
            "localField":   "_id",
            "foreignField": "post_id",
            "as":           "resultField",
        },
    },
    {
        "$addFields": bson.M{
            "resultField": bson.M{"$size": "$resultField"},
        },
    },
})

请注意,$addFields 阶段等同于 $project 阶段,明确指定输入文档中的所有现有字段并添加新字段。仅在 MongoDB 版本 3.4 后可用。