主页 / coding / 问题 / 76969284
Accepted
Neighbourhood
Asked: 2023-08-24 20:31:44 +0800 CST

Golang中的Mongo中值运算

  • 772

我在 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"`
}

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

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 行的评分中位数。我不确定我这样做的正确方法。帮忙，谢谢

mongodb

1 个回答

  1. Best Answer

    在该$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所有ratings 放入一个数组（限制后全部 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}}}
    }}}}
  }}}}
}}}}
    • 1

相关问题