MongoDB 聚合查询分组方式

Question

假设我有一个包含以下信息的 MongoDB 集合：

{
  cust_id: "abc123",
  ord_date: ISODate("2012-11-02T17:04:11.102Z"),
  state: 'CA',
  price: 50,
  item: apple,
  color: red
}
{
  cust_id: "abc123",
  ord_date: ISODate("2012-11-02T17:04:11.102Z"),
  state: 'WA',
  price: 25,
  item: apple,
  color: green
}
{
  cust_id: "abc123",
  ord_date: ISODate("2012-11-02T17:04:11.102Z"),
  state: 'CA',
  price: 75,
  item: orange,
  color: orange
}
{
  cust_id: "def456",
  ord_date: ISODate("2012-11-02T17:04:11.102Z"),
  state: 'OR',
  price: 75,
  item: apple,
  color: red
}

我想对按状态分组的订单总价进行汇总，其中商品为“苹果”，颜色为“红色”。我的查询是：

{
  $match: {$and: [{item : "apple"}, {color : "red"}]},
  $group: {_id: {state: "$state", cust_id: "$cust_id"}, total: {$sum: "$price"}}
}

但是，我希望我的结果 cust_id 包含在 _id 中，它是一个数组/映射/某个结构，其中包含构成我的汇总总数的所有客户 ID 的列表。因此我希望我的输出包含

cust_id {'abc123', 'def456'}

有没有办法处理这个 mongo 聚合/查询？或者也许是构建此查询的更好方法，以便我可以汇总按州分组的红苹果的成本，并包括属于该类别的所有客户？我把它放在 _id 部分是为了提取信息，但其中包含任何这些数据并不重要。我只是想要一种按状态分组并使用上述聚合选择获取所有客户 ID 的集合的方法。

Answer 1

是的，在您的聚合 $group 管道中，您可以使用 $addToSet 聚合运算符将 cust_ids 添加到数组中，同时您仍然可以按状态分组：

db.collection.aggregate([
    {
        "$match": {
            "item": "apple", 
            "color" : "red"
        }
    },
    {
        "$group": {
            "_id": "$state",
            "cust_id": {
                "$addToSet": "$cust_id"
            },
            "total": {
                "$sum": "$price"
            }
        }
    }
]);

输出：

/* 1 */
{
    "result" : [ 
        {
            "_id" : "OR",
            "cust_id" : [ 
                "def456"
            ],
            "total" : 75
        }, 
        {
            "_id" : "CA",
            "cust_id" : [ 
                "abc123"
            ],
            "total" : 50
        }
    ],
    "ok" : 1
}