Mongo中的值运算在Golang中操作

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《Mongo中的值运算在Golang中操作》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

我在 mongo (go) 中有一个集合，其类型是：

type CreateFeedbackRequest struct {
    UserID     string    `json:"user_id" validate:"required"`
    WaybillID  uint64    `json:"waybill_id" validate:"required"`
    Rating     int       `json:"rating" validate:"required"`
    Comment    string    `json:"comment"`
    ReceivedAt time.Time `json:"received_at" validate:"required"`
}

我需要评估某个用户（通过他的 user_id）的最后 5 条记录（通过 receivedAt 时间字段）的评分中值。 我已经得到了这个：

matchStage := bson.D{{"$match", bson.D{{"_id", userID}}}}
sortStage := bson.D{{"$sort", bson.D{{"created_at", 1}}}}
limitStage := bson.D{{"$limit", tripsCount}}

cursor, err := r.c.Aggregate(ctx, mongo.Pipeline{matchStage, sortStage, limitStage})

但我不知道如何获得这 5 行的评分中位数。我不确定我这样做的正确方法。求助，谢谢

正确答案

在 $limit 阶段之后，自 mongodb 版本 7.0 以来的一个选项是 $group 与 $median 累加器

groupgStage := bson.D{{"$group", bson.D{
  {"_id", 0}, 
  {"median", bson.D{{"$median", 
    bson.D{{"$input", "$rating"}, {"method", "approximate"}}
  }}}
}}}

对于旧版本，您可以

1. $sort 通过 rating
2. $group 和 $push 所有 rating 到一个数组（限制后全部 5 个）
3. $project 数组中间的项目

它看起来像这样：

sortRatingStage := bson.D{{"$sort", bson.D{{"rating", 1}}}}
groupStage := bson.D{{"$group", bson.D{{"_id", 0}, {"ratings", bson.D{{"$push", "ratings"}}}}}}
projectStage := bson.D{{"$project", bson.D{
  {"_id", 0}, 
  {median, bson.D{{"$arrayElemAt", bson.D{
    {"$ratings", bson.D{{"$floor", bson.D{
      {"$divide", bson.A{{bson.D{{"$size", "$ratings"}}, 2}}}
    }}}}
  }}}}
}}}}

到这里，我们也就讲完了《Mongo中的值运算在Golang中操作》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！