Mongodb: как использовать агрегат для построения позиций на вложенных полях - PullRequest
Новая
       19

Mongodb: как использовать агрегат для построения позиций на вложенных полях

2 голосов
/ 29 марта 2020

Это моя коллекция: 

[
  {userId: "u1", data: { score1: 1, score2: 2, score3: 3 }, day: 1},
  {userId: "u1", data: { score1: 1, score2: 0, score3: 0 }, day: 2}, 
  {userId: "u1", data: { score1: 5, score2: 3, score3: 2 }, day: 3},
  {userId: "u2", data: { score1: 2, score2: 5, score3: 1 }, day: 1}, 
  {userId: "u2", data: { score1: 1, score2: 1, score3: 6 }, day: 2},
  {userId: "u2", data: { score1: 3, score2: 5, score3: 3 }, day: 3},
  {userId: "u3", data: { score1: 4, score2: 1, score3: 1 }, day: 1},
  {userId: "u3", data: { score1: 0, score2: 1, score3: 1 }, day: 2},
  {userId: "u3", data: { score1: 0, score2: 1, score3: 10 }, day: 3}
]

Я хотел бы построить следующие таблицы лидеров: 

{
  score1: [
    {"u1": 7}, // sum of all score1 for u1
    {"u2": 6}, // sum of all score1 for u2
    {"u3": 4}, // sum of all score1 for u3
  ],
  score2: [
    {"u2": 11}, // sum of all score2 for u2
    {"u1": 5}, // sum of all score2 for u1
    {"u3": 3}, // sum of all score2 for u3
  ],
  score3: [
    {"u3": 12}, // sum of all score3 for u3
    {"u2": 10}, // sum of all score3 for u2
    {"u1": 5}, // sum of all score3 for u1
  ],
}

До сих пор я мог группировать по userId и вычислять совокупность каждой оценки для 3 из них: 

db.myCollection.aggregate([
  {
    $group: {
      _id: "$userId",
      score1: { $sum: "$score1" },
      score2: { $sum: "$score2" },
      score3: { $sum: "$score3" }
    }
  }
])

Что дает мне: 

[
  { 
    _id: "u1",
    score1: 7,
    score2: 5,
    score3: 5
  },
  { 
    _id: "u2",
    score1: 6,
    score2: 11,
    score3: 10
  },
  { 
    _id: "u3",
    score1: 4,
    score2: 3,
    score3: 12
  },
]

Как я могу извлечь каждый тип счета и построить соответствующую таблицу лидеров?

Спасибо в авансовый.

1 Ответ

3 голосов
/ 29 марта 2020

Сначала я бы использовал $objectToArray в поле data и $unwind, чтобы у каждого документа был 1 пользователь и 1 балл. Затем сгруппируйте по userId и data.k (которые будут содержать "Score1", "Score2" и др. c.) И вычислите сумму. Затем перегруппируйте по имени партитуры и pu sh объект с k:userId, v:<score> в массив. Затем сгруппируйте еще раз по null и pu sh k:scoreName, v:<object with user scores> в массив. Наконец $arrayToObject для преобразования этого массива в нужный объект: 

db.collection.aggregate([
    {$addFields: {data: {$objectToArray: "$data"}}},
    {$unwind: "$data"},
    {$group: {
         _id: {userId: "$userId", scoreName: "$data.k"},
         score: {$sum:"$data.v"}
    }},
    {$group: {
         _id:"$_id.scoreName",
         data:{$push:{k:"$_id.userId", v:"$score"}}
    }},
    {$group: {
         _id: null,
         scores:{$push:{k:"$_id", v:{$arrayToObject:"$data"}}}
    }},
    {$replaceRoot:{newRoot:{$arrayToObject:"$scores"}}}
])

Детская площадка

...