MongoDB 将总计添加到聚合管道中的 sortByCount()

Question

我已按国家/地区对所有用户进行分组，但我还希望有一行显示总计(在我们的用例中，用户被标记到单个国家/地区)。

数据模型/样本输入

该集合充满了代表国家/地区(name)的对象，每个对象都在 users 下的数组中包含一个用户对象列表。

{ _id: ObjectId("..."),
  name: 'SG',
  type: 'COUNTRY',
  increment: 200,
  users: 
   [ ObjectId("..."),
     ObjectId("..."),
     ...

查询

db.collection.aggregate([{$match:{type:"COUNTRY"}},{$unwind:"$users"},{$sortByCount:"$name"}])

当前结果

{ _id: 'SG', count: 76 } 
{ _id: 'IN', count: 6 }  
{ _id: 'US', count: 4 }  
{ _id: 'FR', count: 3 }  
{ _id: 'UK', count: 2 } 
{ _id: 'RU', count: 1 } 
{ _id: 'CO', count: 1 } 
{ _id: 'DK', count: 1 } 
{ _id: 'ID', count: 1 } 
{ _id: 'PH', count: 1 }

预期结果

{ _id: 'SG', count: 76 } 
{ _id: 'IN', count: 6 }  
{ _id: 'US', count: 4 }  
{ _id: 'FR', count: 3 }  
{ _id: 'UK', count: 2 } 
{ _id: 'RU', count: 1 } 
{ _id: 'CO', count: 1 } 
{ _id: 'DK', count: 1 } 
{ _id: 'ID', count: 1 } 
{ _id: 'PH', count: 1 }
{ _id: null, count: 96 } <<< TOTAL COUNT ADDED

有什么技巧可以在不诉诸复杂或肮脏技巧的情况下实现这一目标吗？

Answer 1

您还可以尝试使用 $facet 按国家/地区名称和总计数计算计数，然后将它们组合在一起。像这样的事情:

db.collection.aggregate([
  {
    $match: {
      type: "COUNTRY"
    }
  },
  {
    "$unwind": "$users"
  },
  {
    "$facet": {
      "groupCountByCountry": [
        {
          "$sortByCount": "$name"
        }
      ],
      "totalCount": [
        {
          "$group": {
            "_id": null,
            "count": {
              "$sum": 1
            }
          }
        }
      ]
    }
  },
  {
    "$project": {
      array: {
        "$concatArrays": [
          "$groupCountByCountry",
          "$totalCount"
        ]
      }
    }
  },
  {
    "$unwind": "$array"
  },
  {
    "$replaceRoot": {
      "newRoot": "$$ROOT.array"
    }
  }
])

这里是 Playground link .