如何使用MongoDB聚合框架计算满足特定条件的组数？

Question

我有一个使用几个组的 MongoDB 聚合操作：

{ $group: { "_id": { "vid": "$visitor_id", "pid":"$project.id" } } }
{ $group: { "_id": "$_id.vid", "count": { $sum: 1} } }

返回以下数据：

{
  "results": [
    {
      "_id": "user1",
      "count": 1
    }, 
    {
      "_id": "user2",
      "count": 2
    },
    {
      "_id": "user3",
      "count": 1
    }
  ]
}

我如何才能返回具有 1 个以上项目（count字段）的用户总数。在这种情况下，它将类似于：

{
  total: 1
}

因为只有user21 个以上的项目（count> 1）。

我尝试添加以下组操作无济于事：

{ 
  $group: { 
    "_id": null, 
    count: {
      $sum: {$cond: [{$gt: ['$_id.count', 1]}, 1, 0]}
    } 
  } 
}

有任何想法吗？

Answer 1

您只需将以下三个聚合阶段添加到管道中即可实现所需的结果：

{$match: {count: {$gt: 1}}}
{$group: {_id: null, total: {$sum: 1}}}
{$project: {_id: 0, total: 1}}

1. 正如chridam 所提到的，您可以有一个$match阶段来过滤拥有超过 1 个项目的所有用户。
2. 然后使用另一个$group阶段来统计此类用户的总数。
3. 第三步只是化妆品，只留下领域total。