Как создать массив на выходе ответа с использованием агрегата в Mongodb - PullRequest
Новая
       4

Как создать массив на выходе ответа с использованием агрегата в Mongodb

1 голос
/ 02 апреля 2020

В моей коллекции есть список объектов с такой структурой: 

[
 {
    "country": "colombia",
    "city":"medellin",
    "calification": [
        {
            "_id": 1,
            "stars": 5
        },
        {
            "_id": 2,
            "stars": 3
        }
    ]
},
{
    "country": "colombia",
    "city":"manizales",
    "calification": [
        {
            "_id": 1,
            "stars": 5
        },
        {
            "_id": 2,
            "stars": 5
        }
    ]
},

{
    "country": "argentina",
    "city":"buenos aires",
    "calification": [
        {
            "_id": 1,
            "stars": 5
        },
    ]
},
{
    "country": "perú",
    "city":"cusco",
    "calification": [
        {
            "_id": 3,
            "stars": 3
        },
    ]
  }
]

Я пытаюсь создать фильтр, чтобы выходные данные представляли собой количество массивов для каждой страны. это пример вывода, который я хочу.

avg будет результатом sum 'stars' / calification.length 

{
  "colombia": [
    {
      "city": "medellin",
      "avg": 4,
      "calification": [
        {
          "_id": 1,
          "stars": 5
        },
        {
          "_id": 2,
          "stars": 3
        }
      ]
    },
    {
      "city": "manizales",
      "avg": 5,
      "calification": [
        {
          "_id": 1,
          "stars": 5
        },
        {
          "_id": 2,
          "stars": 3
        }
      ]
    }
  ],
  "argentina": {
    "city": "buenos aires",
    "avg": 5,
    "calification": [
      {
        "_id": 1,
        "stars": 5
      }
    ]
  },
  "peru": {
    "city": "cusco",
    "avg": 4,
    "calification": [
      {
        "_id": 1,
        "stars": 4
      }
    ]
  }
}

Я пытаюсь сделать это: 

    Alcalde.aggregate([

        {
            $addFields: {
                colombia: {
                    "$push": {

                        "$cond": [{ $eq: ["$country", "'Colombia'"] }, true, null]


                    }


                }

            }
        },
        {
            $project: { colombia: "$colombia" }
        }
    ]

как я могу сделай это

1 Ответ

1 голос
/ 02 апреля 2020

Мы можем сделать его более элегантным.

MongoDB имеет оператор $ avg , давайте его использовать. Кроме того, мы можем использовать оператор $group для группировки городов для одного и того же country.

. В конце, применяя $ replace Root + $ arrayToObject ** мы преобразуем в желаемый результат.

** это потому, что мы не можем использовать такое выражение: {"$country":"$city"}

$replaceRoot                                 $arrayToObject

data : {              {                  [                             {
  "key" : "val",  -->   "key" : "val",     {k:"key", v: "val"},  -->     "key" : "val",
  "key2" : "val2"       "key2" : "val2"    {k:"key2", v: "val2"}         "key2" : "val2"
}                     }                  ]                             }

Попробуйте это: 

Alcalde.aggregate([
  {
    $group: {
      _id: "$country",
      city: {
        $push: {
          "city": "$city",
          "avg": { $avg: "$calification.stars"},
          "calification": "$calification"
        }
      }
    }
  },
  {
    $replaceRoot: {
      newRoot: {
        $arrayToObject: [ [{ "k": "$_id", "v": "$city"}] ]
      }
    }
  }
])

MongoPlayground

РЕДАКТИРОВАТЬ: Generi c способ заполнения city внутренний объект

$$ROOT является переменной, которая хранит root документ $mergeObjects добавляет / переопределяет поля к конечному объекту 

Alcalde.aggregate([
  {
    $group: {
      _id: "$country",
      city: {
        $push: {
          $mergeObjects: [
            "$$ROOT",
            {
              "avg": { "$avg": "$calification.stars" }
            }
          ]
        }
      }
    }
  },
  {
    $project: {
      "city.country": 0
    }
  },
  {
    $replaceRoot: {
      newRoot: {
        $arrayToObject: [
          [ { "k": "$_id", "v": "$city" } ]
        ]
      }
    }
  }
])

MongoPlayground

...