Início / coding / Perguntas / 76969284
Accepted
Neighbourhood
Asked: 2023-08-24 20:31:44 +0800 CST

Operação mediana do Mongo em Golang

  • 772

Tenho uma coleção no mongo (go) cujo tipo é:

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

Preciso avaliar o valor mediano da classificação dos últimos 5 registros (por receivedAtcampo de tempo) para determinado usuário (por seu user_id). Eu já consegui isso:

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})

Mas não sei como obter a mediana de classificação para essas 5 linhas. E não tenho certeza sobre a maneira correta de fazer isso. Ajuda, obrigado

mongodb

1 respostas

  1. Best Answer

    Após o $limitestágio, uma opção desde o mongodb versão 7.0 é $groupcom $medianacumulador

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

    Para versões mais antigas, você pode

    1. $sortporrating
    2. $groupe $pushtodos ratings para uma matriz (todos os 5 após o limite)
    3. $projecto item no meio da matriz

    Será assim:

    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

relate perguntas