Суммарные значения всех ключей в поддокументе

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

{
    '_id': 'doc1',
    'store_A': {'apples': 50, 'oranges':20},
    'store_B': {'oranges': 15}
}
{
    '_id': 'doc2',
    'store_A': {'oranges':10},
    'store_B': {'apples': 15}
}

Как мне написать команду агрегирования, чтобы получить общее количество фруктов для каждого магазина по всем документам в коллекции БЕЗ перечисления всех разрешенных видов фруктов?

Результат должен выглядеть так:

{
    '_id': 'Result',
    'store_A_total': {'apples': 50, 'oranges': 30},
    'store_B_total': {'apples': 15, 'oranges': 15}
}

Этот запрос работает, но все типы фруктов должны быть указаны явно:

db.collection.aggregate(
{'$group': {'_id': 'Result',
    'store_A_apples': {'$sum': '$Store_A.apples'},
    'store_A_oranges': {'$sum': '$store_A.oranges'},
    'store_B_apples': {'$sum': '$store_B.apples'},
    'store_B_oranges': {'$sum': '$store_B.oranges'}
}},
{'$project': {
    'store_A': {'apples': '$store_A_apples','oranges': '$store_A_oranges'},
    'store_B': {'apples': '$store_B_apples','oranges': '$store_B_oranges'}
}})

Есть ли лучший способ структурировать эти документы, чтобы упростить запросы такого типа?


mongodb aggregation-framework
person SuperAce99    schedule 18.07.2013    source источник

Ответы (1)


arrow_upward
5
arrow_downward

В среде агрегации mongodb нет способа обрабатывать ключ внутри документа как данные, которые вы можете исследовать или манипулировать. Обходной путь — превратить то, что вы используете здесь в качестве ключей (например, тип фруктов и название магазина), в такие значения:

{
    "_id" : "doc1",
    "stores":[
        {
            // store name is a value
            "name":"store_A",
            "inventory": [
            {
                // so is fruit type
                "type" : "apple",
                "count" : 50
            },
            {
                "type" : "orange",
                "count" : 20
            }
            ]
        },
        {
            "name": "store_B",
            "inventory": [
            {
                "type" : "orange",
                "count" : 15
            }
            ]
        }
    ]
}

Это позволяет вам легче работать с этими данными в агрегированном виде:

db.coll.aggregate([
    // split documents by store name
    {$unwind:"$stores"},
    // split documents further by fruit type
    {$unwind:"$stores.inventory"},
    // group documents together by store/fruit type, count quantities of fruit
    {$group:{"_id":{"store":"$stores.name", "fruit":"$stores.inventory.type"},
             "count":{$sum:"$stores.inventory.count"}}},
    // reformat the data to look more like your specification
    {$project:{
        "store":"$_id.store",
        "fruit":"$_id.fruit",
        "_id":0,
        "count":1}}])

Вывод выглядит так:

{
    "result" : [
        {
            "count" : 15,
            "store" : "store_B",
            "fruit" : "apple"
        },
        {
            "count" : 15,
            "store" : "store_B",
            "fruit" : "orange"
        },
        {
            "count" : 30,
            "store" : "store_A",
            "fruit" : "orange"
        },
        {
            "count" : 50,
            "store" : "store_A",
            "fruit" : "apple"
        }
    ],
    "ok" : 1
}
person llovett    schedule 17.09.2013