MongoDB中根据ID列表查询嵌套数组指定对象的异常问题

数据库结构 #

{
  '_id': ObjectId('3f05e2aa794e17504a6674a7'),  
  'lt': [
    {'_id': ObjectId('6f05e2aa794e177b456674a9'), 'name': 'text1'}, 
    {'_id': ObjectId('2f05e2aa794e1765286674a8'),'name': 'text3'}
  ]
}

{
  '_id': ObjectId('3f05e3aa791e17f23b6674aa'), 
  'lt': [
    {'_id': ObjectId('7f05e2aa494e17f5b36674ac'), 'name': 'text12'}, 
    {'_id': ObjectId('5f05e2aa794e1707006674ab'), 'name': 'text2'}
  ]
}

目标ID列表 #

lists = ["6f05e2aa794e177b456674a9", "2f05e2aa794e1765286674a8", "5f05e2aa794e1707006674ab"]

预期结果 #

{
  'lt': [
    {'_id': ObjectId('6f05e2aa794e177b456674a9'), 'name': 'text1'}, 
    {'_id': ObjectId('2f05e2aa794e1765286674a8'),'name': 'text3'}
  ]
}
{
  'lt': [
    {'_id': ObjectId('5f05e2aa794e1707006674ab'), 'name': 'text2'}
  ]
}

现有问题 #

单值查询时用$elemMatch可以正常返回对应元素，但改成ID列表查询后，投影阶段的$elemMatch只会返回每个文档中第一个匹配的元素，无法获取所有符合条件的元素。

错误代码示例：

lists = ["6f05e2aa794e177b456674a9", "2f05e2aa794e1765286674a8", "5f05e2aa794e1707006674ab"]
obj_ids = list(map(lambda x: ObjectId(x), lists))

fx = mycol.find(
    { "lt": { "$elemMatch": { "_id": {"$in" : obj_ids }}}},
    { "lt": { "$elemMatch": { "_id": {"$in": obj_ids }}},
     "_id":0, "lt._id":1 , "lt.name":1}
  ).limit(5)

for x in fx:
    print(x)

解决方案 #

使用MongoDB的聚合框架，通过$filter操作符在投影阶段筛选数组中所有匹配ID的元素，替代$elemMatch。

修改后的代码：

from bson.objectid import ObjectId

lists = ["6f05e2aa794e177b456674a9", "2f05e2aa794e1765286674a8", "5f05e2aa794e1707006674ab"]
obj_ids = list(map(lambda x: ObjectId(x), lists))

pipeline = [
    # 过滤出包含至少一个匹配元素的文档
    {
        "$match": {
            "lt._id": {"$in": obj_ids}
        }
    },
    # 筛选lt数组中所有符合ID条件的元素
    {
        "$project": {
            "_id": 0,
            "lt": {
                "$filter": {
                    "input": "$lt",
                    "as": "item",
                    "cond": {"$in": ["$$item._id", obj_ids]}
                }
            }
        }
    },
    {"$limit": 5}
]

fx = mycol.aggregate(pipeline)

for x in fx:
    print(x)

代码说明 #

1. $match阶段：筛选出lt数组中存在目标ID的文档，减少后续处理的数据量。这里直接用"lt._id": {"$in": obj_ids}替代$elemMatch，效果相同且更简洁。
2. $project阶段：使用$filter遍历lt数组，保留所有_id在目标列表中的元素，确保返回每个文档中所有符合条件的子元素。
3. $limit：限制返回结果数量，与原逻辑保持一致。

内容的提问来源于stack exchange，提问作者GreatFilter