我有几个文档遵循这种结构:

{
  "queue-type": <integer>,
  "participants": [{
     "id": <integer>,
     "level": <level>,
     "flags": <integer>
  }]
}

participants.id 上有一个多键索引。

代码中有一个find查询，如下:db.queues.find({"participants.id": {"$in": [2, 3, 4]}})，结果如下:

{"queue-type": 1, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}
{"queue-type": 25, "participants": [{"id": 5, "level": 10, "flags":4},{"id": 15, "level": 10, "flags":8},{"id": 4, "level": 10, "flags":8}]}

有没有办法同时检索用于匹配查询的元素？像这样的东西:

{"queue-type": 1, "_matched": 2, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}
{"queue-type": 25, "_matched": 4, "participants": [{"id": 5, "level": 10, "flags":4},{"id": 15, "level": 10, "flags":8},{"id": 4, "level": 10, "flags":8}]}

PS:我试图避免遍历 [2, 3, 4] 和 participants 数组，因为它们要大得多。

例子: 队列

{"queue-type": 1, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}
{"queue-type": 2, "participants": [{"id": 3, "level": 10, "flags":0}]}
{"queue-type": 3, "participants": [{"id": 4, "level": 10, "flags":4},{"id": 5, "level": 10, "flags":8}]}
{"queue-type": 4, "participants": [{"id": 7, "level": 10, "flags":4},{"id": 8, "level": 10, "flags":8},{"id": 9, "level": 10, "flags":8}]}

我要检索的结果:

db.queues.find({"participants.id": {"$in": [2]}});
{"queue-type": 1, "_matched": 2, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}

请注意“_matched”元素与搜索查询中给出的“participant.id”相同

另一个例子:

db.queues.find({"participants.id": {"$in": [2, 3, 6]}});
{"queue-type": 1, "_matched": 2, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}
{"queue-type": 2, "_matched": 3, "participants": [{"id": 3, "level": 10, "flags":0}]}

多个匹配示例:

db.queues.find({"participants.id": {"$in": [1, 2, 3]}});
{"queue-type": 1, "_matched": 1, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}
{"queue-type": 1, "_matched": 2, "participants": [{"id": 1, "level": 10, "flags":4},{"id": 2, "level": 10, "flags":8}]}
{"queue-type": 2, "_matched": 3, "participants": [{"id": 3, "level": 10, "flags":0}]}

一个不好的解决方案是简单地复制'participants'数据('participants-cpy')然后运行:

db.queues.find({"participants-cpy.id": {"$in": [2]}}, {"participants-cpy.$":1, "participants":1, "_id": 1, "queue-type":1})

这可用于检索用于“匹配”查询的元素，但这会生成重复的数据——这非常糟糕:p

最佳答案

常规find查询在这里不起作用。您需要使用聚合框架。在您的管道中，您需要使用 $match 仅选择那些与您的查询条件匹配的文档。管道运算符(operator)。

管道中的下一个也是最后一个阶段是 $project您使用的阶段将新字段“_matched”添加到您的文档中。如果您考虑一下，您会意识到新字段只不过是一个数组，其中包含出现在“participantsId”数组/列表中的元素以及文档中“participants”字段中的所有“id”。

要获得该值，您只需使用 $map 对 participantsId 数组和来自“participants”的“id”数组执行集合交集操作。和 $setIntersection运算符(operator)。请注意，结果数组仅包含唯一条目，因为 $setIntersection 过滤掉重复项。

participantsId = [1, 2, 3]
db.queues.aggregate([
    { "$match": { "participants.id": { "$in": participantsId } } }, 
    { "$project": {
        "_matched": {
            "$setIntersection": [
                { "$map": { 
                    "input": "$participants", 
                    "as": "p", 
                    "in": "$$p.id"
                }}, 
                participantsId
            ]
        }, 
        "queue-type": 1, 
        "participants": 1
    }}
])

关于mongodb - 从 $in 查询中返回匹配的元素，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/37458142/