MongoDB：获得不同组之间价值最高的文档计数

Question

我面临MongoDB的问题。我需要计算由另一个字段最高值的字段分组的所有文档。

我给您一个例子：

{ id: 1, scanner: 'bandit', severity: 'low', version: 1 },
{ id: 2, scanner: 'bandit', severity: 'low', version: 1 },
{ id: 3, scanner: 'bandit', severity: 'medium', version: 1 },
{ id: 4, scanner: 'bandit', severity: 'medium', version: 2 },
{ id: 5, scanner: 'bandit', severity: 'high', version: 2 },
{ id: 6, scanner: 'semgrep', severity: 'critical', version: 2 },
{ id: 7, scanner: 'semgrep', severity: 'critical', version: 3 },
{ id: 8, scanner: 'semgrep', severity: 'info', version: 4 },
{ id: 9, scanner: 'semgrep', severity: 'info', version: 4 },

在这种情况下，我们有两个扫描仪（BANDIT和semgrep）。

“强盗”文档的最高版本是2，而“ semgrep”文档的最高版本是4。

我需要根据扫描仪和严重性对它们进行分组，具体取决于它们的最高版本，忽略了以前的版本，并计算它们。

结果应该与此相似：

[
    { "scanner": "bandit", "severity": "medium", "count": 1, "version": 2 },
    { "scanner": "bandit", "severity": "high", "count": 1, "version": 2 },
    { "scanner": "semgrep", "severity": "info", "count": 2, "version": 4 }
]

我正在挣扎，到达我无法忽略前版本的时候。任何帮助都将受到赞赏。

非常感谢

Answer 1

使用MongoDB v5.0+，

您可以使用$ setWindowfields来计算扫描仪分组的等级。然后，具有最高等级（IE等级= 1），您可以执行另一个$组和计数。

db.collection.aggregate([
  {
    "$setWindowFields": {
      "partitionBy": "$scanner",
      "sortBy": {
        "version": -1
      },
      "output": {
        "rank": {
          "$rank": {}
        }
      }
    }
  },
  {
    $match: {
      "rank": 1
    }
  },
  {
    $group: {
      _id: {
        "scanner": "$scanner",
        "severity": "$severity",
        "version": "$version"
      },
      count: {
        $sum: 1
      }
    }
  },
  {
    $project: {
      "_id": 0,
      "scanner": "$_id.scanner",
      "severity": "$_id.severity",
      "version": "$_id.version",
      "count": 1
    }
  }
])

这是 mongo Playground 供您参考。