具有多个子组的mongodb聚合

Question

我有一个包含与此类似的文档的集合：

[
    {
        "_id": ObjectId("..."),
        "date": ISODate("..."),
        "type": "TypeA",
        "color": "ColorA",
        "soldFor": 12.15
    },
    {
        "_id": ObjectId("..."),
        "date": ISODate("..."),
        "type": "TypeA",
        "color": "ColorB",
        "soldFor": 13.15
    },
    {
        "_id": ObjectId("..."),
        "date": ISODate("..."),
        "type": "TypeB",
        "color": "ColorA",
        "soldFor": 12.15
    },
    {
        "_id": ObjectId("..."),
        "date": ISODate("..."),
        "type": "TypeB",
        "color": "ColorB",
        "soldFor": 12.15
    }
]

我知道这不是存储此类信息的好方法，但不幸的是我对此没有影响。 我需要从集合中得到的是这样的：

[
    2017: {
        typeA: {
            colorA: {
                sum: 125.00
            },
            colorB: {
                sum: 110.00
            }            
        },
        typeB: {
            colorA: {
                sum: 125.000
            }
        }        
    },
    2016: {
        typeA: {
            colorB: {
                sum: 125.000
            }
        }
    }
]

目前我有两个小组赛阶段，所有内容都按年份分组，但我不知道如何获得另外两个小组。建立总和将是一件好事，但我确信我可以弄清楚如何在一个小组中完成。

到目前为止，我的管道如下所示：

[
    {
      $group: {
        _id: { type: '$type', color: '$color', year: { $year: '$date' } },
        docs: {
          $push: '$$ROOT'
        }
      }
    },
    {
      $group: {
        _id: { year: '$_id.year' },
        docs: {
          $push: '$$ROOT'
        }
      }
    }
]

结果如下：

[
    {
        "_id": {
            "year": 2006
        },
        "docs": {
            "_id": {
                "type": "typeA",
                "color": "colorA",
                "year": 2006
            },
            "docs": [
                {
                    ... root document
                }
            ]
        }
    },
    {
        "_id": {
            "year": 2016
        },
        "docs": [
            {
                "_id": {
                    "type": "typeA",
                    "color": "colorB",
                    "year": 2016
                },
                "docs": [
                    {
                        ... root document
                    }
                ]
            }
            ... more docs with three keys in id
        ]
    }
]

非常感谢您的帮助！

Answer 1

使用 MongoDB 3.4.4 及更高版本中的一组运算符，即 $addFields、$arrayToObject 和 $replaceRoot ，你可以像下面这样组成一个管道来得到想要的结果：

[
    { "$group": {
        "_id": { 
            "year": { "$year": "$date" },
            "type": "$type",
            "color": "$color"
        },
        "count": { "$sum": "$soldFor" }
    } },
    { "$group": {
        "_id": {
            "year": "$_id.year",
            "type": "$_id.type"
        },
        "counts": {
            "$push": {
                "k": "$_id.color",
                "v": { "sum": "$count" }
            }
        }
    } },
    { "$addFields": {
        "counts": { "$arrayToObject": "$counts" }
    } },
    { "$group": {
        "_id": "$_id.year",
        "counts": {
            "$push": {
                "k": "$_id.type",
                "v": "$counts"
            }
        }
    } },
    { "$addFields": {
        "counts": { "$arrayToObject": "$counts" }
    } },
    { "$group": {
        "_id": null,
        "counts": {
            "$push": {
                "k": { "$substr": ["$_id", 0, -1 ]},
                "v": "$counts"
            }
        }
    } },
    { "$replaceRoot": { 
        "newRoot": { 
            "$mergeObjects": [ 
                { "$arrayToObject": "$counts" }, 
                "$$ROOT" 
             ] 
        } 
    } },
    { "$project": { "counts": 0 } }
]