登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-14 12:54:28 316浏览 收藏

大家好,我们又见面了啊~本文《在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学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>