如何在 mongo 查询中获取子数组的长度

Question

示例数据如下：

db.infos.find()
{
    "groupId": "1111",
    "customerId": "A100",
    "tracks": [{
        "trackId": "234",
        "infos": [{
                "location": {
                    "address": "street1",
                    "city": "test",
                    "country": "US"
                }
            },
            {
                "location": {
                    "address": "street2",
                    "city": "test",
                    "country": "US"
                }
            },
            {
                "location": {
                    "address": "street3",
                    "city": "test",
                    "country": "US"
                }
            }
        ]
    }]
}
{
    "groupId": "2222",
    "customerId": "A100",
    "tracks": [{
        "trackId": "345",
        "infos": [{
                "location": {
                    "address": "street4",
                    "city": "test",
                    "country": "US"
                }
            },
            {
                "location": {
                    "address": "street5",
                    "city": "test",
                    "country": "US"
                }
            },
            {
                "location": {
                    "address": "street5",
                    "city": "test",
                    "country": "US"
                }
            }
        ]
    }]
}
{
    "groupId": "2222",
    "customerId": "A100",
    "tracks": [{
        "trackId": "666",
        "infos": [{
                "location": {
                    "address": "street4",
                    "city": "test",
                    "country": "US"
                }
            },
            {
                "location": {
                    "address": "street5",
                    "city": "test",
                    "country": "US"
                }
            },
            {
                "location": {
                    "address": "street5",
                    "city": "test",
                    "country": "US"
                }
            }
        ]
    }]
}

我们需要一个查询来获取 groupId 级别的“infos”子数组的长度。在上面的示例数据中，我们需要以下输出： 1111, 3 2222, 6

在下面尝试过，但它不起作用：

 db.infos.aggregate([{"$project": {"groupId": "$groupId", "samples": "$tracks.infos"}}, {"$group": {"_id": "$groupId", "samples": {"$push": "$samples"}}}, {"$project": {"_id": 1, "samples": {"$size": "$samples"}}}], {   "allowDiskUse": true });

还有大量数据，上面的查询抛出 ExceededMemoryLimit 异常：

2021-07-10T07:20:14.415+0000 E QUERY    [js] Error: command failed: {
    "ok" : 0,
    "errmsg" : "$push used too much memory and cannot spill to disk. Memory limit: 104857600 bytes",
    "code" : 146,
    "codeName" : "ExceededMemoryLimit"
} : aggregate failed :
_getErrorWithCode@src/mongo/shell/utils.js:25:13
doassert@src/mongo/shell/assert.js:18:14
_assertCommandWorked@src/mongo/shell/assert.js:580:17
assert.commandWorked@src/mongo/shell/assert.js:673:16
DB.prototype._runAggregate@src/mongo/shell/db.js:260:9
DBCollection.prototype.aggregate@src/mongo/shell/collection.js:1074:12
@(shell):1:1

Answer 1

这应该与 unwind 一起使用：

db.collection.aggregate([
  {
    $unwind: "$tracks"
  },
  {
    $group: {
      "_id": "$groupId",
      "count": {
        "$sum": {
          "$size": "$tracks.infos"
        }
      }
    }
  }
])

即使您的轨道数组有多个对象，这也会将它们加起来。 Playground

如何在 mongo 查询中获取子数组的长度

How to get length of sub array in mongo query

mongodb

mongodb-query

aggregation-framework

spring-data-mongodb