Как сделать поворот на MongoDB? - PullRequest
Новая
       12

Как сделать поворот на MongoDB?

2 голосов
/ 03 апреля 2019

У меня есть коллекция 'Sales', и ее пример выглядит так: 

[
{cusID: 'a412q39x',
cusCountry: 'MEX',
itemPurchased: 'Toy_A'
},
{cusID: 'r760e11s',
cusCountry: 'USA',
itemPurchased: 'Toy_B'
},
{cusID: 'g723f01z',
cusCountry: 'USA',
itemPurchased: 'Toy_C'
},
{cusID: 'h277p01c',
cusCountry: 'CAN',
itemPurchased: 'Toy_B'
}
]

Это результат, которого я надеюсь достичь. 

[
{item: 'Toy_A',
USA: 4,
MEX: 2,
CAN: 1,
BRA: 0
},
{item: 'Toy_B',
USA: 3,
MEX: 0,
CAN: 2,
BRA: 1
}
]

Я пытался: 

{
$group:{_id:{toy:'$itemPurchased', country: $cusCountry'},'cnt':{'$sum': 1}}
}

Результат оказался не тем, что я хотел. 

[
{
_id.toy: 'Toy_A',
_id.country: 'BRA',
cnt: 43
},
{
_id.toy: 'Toy_A',
_id.country: 'USA',
cnt: 102
},
{
_id.toy: 'Toy_A',
_id.country: 'JPN',
cnt: 72
},
{
_id.toy: 'Toy_B',
_id.country: 'CAN',
cnt: 32
}
]

Я также экспериментировал с $ facet, но безрезультатно. Монго гуру, пожалуйста, просветите. Заранее спасибо.

1 Ответ

0 голосов
/ 03 апреля 2019

Попробуйте, как показано ниже: 

db.collection.aggregate([
    {
        $group: { _id: '$itemPurchased' ,  items: { $push: { "country" : "$cusCountry", "count": { $sum :1} } } }
    },
    {
        "$project": {
            "countryCounts": {
                "$arrayToObject": {
                    "$map": {
                        "input": "$items",
                        "as": "item",
                        "in": {
                            "k": "$$item.country",
                            "v": "$$item.count",
                        }
                    }
                }
            }

        }
    },
    { $replaceRoot: { newRoot:  { "$mergeObjects":[  "$countryCounts" , { "item":"$_id"} ] } } }
])

Вы получите результат, как показано ниже: 

{
    "USA" : 1,
    "item" : "Toy_C"
},

/* 2 */
{
    "USA" : 1,
    "CAN" : 1,
    "item" : "Toy_B"
},

/* 3 */
{
    "MEX" : 1,
    "item" : "Toy_A"
}
...