在Go语言中执行并转换聚合查询为BSON

大家好，我们又见面了啊~本文《在Go语言中执行并转换聚合查询为BSON》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

这是我想从代码运行的以下查询。

db.apnsummary.aggregate([{"$match":{"mid":{"$in":["81"]}}},{"$group":{"_id":"$mid ","点击":{"$sum":"$open"},"已交付":{"$sum":"$delivered"},"失败":{"$sum":"$failed"}, "freqdrop":{"$sum":"$freqdrop"},"已发布":{"$sum":"$已发布"},"已发送":{"$sum":"$已发送"}}}] ）；

从代码准备的查询：-

[{"$match":{"date":{},"mid":{"$in":[81]}}},{"$group":{"_id":"$mid","clicked":{"$sum":"$open"},"delivered":{"$sum":"$delivered"},"failed":{"$sum":"$failed"},"freqdrop":{"$sum":"$freqdrop"},"published":{"$sum":"$published"},"sent":{"$sum":"$sent"}}}]

当我执行此查询时，我得到结果，但是当我从代码执行查询时，它没有提供结果，但给出了空白数据。 我的代码如下：-

func GetApnDataFromMongo(data utils.Data, clientName string) (utils.Results, error) {
    result := utils.Results{}

    date := bson.M{}
    messageId := bson.M{}
    if len(data.MessageId) > 0 {
        messageId = bson.M{"$in": convertToInt(data.MessageId)}
    }

    if data.DateFrom != "" && data.DateTo != "" {
        date = bson.M{"$gte": data.DateFrom, "$lte": data.DateTo}
    }

    primary_collection := bson.M{"$match": bson.M{"mid": messageId, "date": date}}
    group_collection := bson.M{"$group": bson.M{"_id": "$mid", "published": bson.M{"$sum": "$published"}, "sent": bson.M{"$sum": "$sent"}, "failed": bson.M{"$sum": "$failed"}, "freqDrop": bson.M{"$sum": "$freqDrop"}, "delivered": bson.M{"$sum": "$delivered"}, "clicked": bson.M{"$sum": "$open"}}}

    fullquery := []bson.M{primary_collection, group_collection}
    jsonString, _ := json.Marshal(fullquery)
    logrus.Info(fmt.Sprintf("Final query prepared is as following %+v", string(jsonString)))
    ctx := context.Background()
    conn, err := utils.GetMongoConnection()
    if err != nil {
        logrus.Errorf("Error connecting to mongo %+v", err)
        return result, err
    }
    logrus.Info("clientName %+v", clientName)
    resultData, err := conn.Database(clientName).Collection("apnSummary").Aggregate(ctx, mongo.Pipeline(primary_collection, group_collection))
    if err != nil {
        logrus.Errorf("Error querying %+v", err)
        return result, err
    }
    mongodata := []utils.MonogResponse{}
    resultData.All(ctx, &mongodata)
    fmt.Println(mongodata)
    result = PrepareMongoResponse(mongodata)
    return result, nil
}

func PrepareMongoResponse(data []utils.MonogResponse) utils.Results {
    result := utils.Results{}
    for _, res := range data {
        result.Failure.Mid = append(result.Failure.Mid, res.Mid)
        result.Delivered.Mid = append(result.Delivered.Mid, res.Mid)
        result.Frequency.Mid = append(result.Frequency.Mid, res.Mid)
        result.Sent.Mid = append(result.Sent.Mid, res.Mid)
        result.Click.Mid = append(result.Click.Mid, res.Mid)
        result.Failure.Count = append(result.Failure.Count, res.Failed)
        result.Delivered.Count = append(result.Delivered.Count, res.Delivered)
        result.Frequency.Count = append(result.Frequency.Count, res.FreqDrop)
        result.Sent.Count = append(result.Sent.Count, res.Sent)
        result.Click.Count = append(result.Click.Count, res.Clicked)
    }
    return result
}

func convertToInt(data []string) []int {
    intData := []int{}
    for _, dt := range data {
        intV, _ := strconv.Atoi(dt)
        intData = append(intData, intV)
    }
    return intData
}

正确答案

如果 date 过滤器缺失，请勿添加 "date": {} 过滤器，该过滤器会过滤 "date" 为空对象的文档，这显然不适用于您的情况。如果您没有日期过滤器，请从 $match 阶段省略 date！

messageid 和所有其他过滤器也是如此。

你可以这样做：

matchDoc := bson.M{}
if len(data.MessageId) > 0 {
    matchDoc["mid"] = bson.M{"$in": convertToInt(data.MessageId)}
}
if data.DateFrom != "" && data.DateTo != "" {
    matchDoc["date"] = bson.M{"$gte": data.DateFrom, "$lte": data.DateTo}
}
primary_collection := bson.M{"$match": matchDoc}

注意：您的原始查询之所以有效，是因为您遗漏了 date:{} 过滤器！

今天关于《在Go语言中执行并转换聚合查询为BSON》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！