如何通过单次MongoDB查询实现ID数组比对，替代循环多次查询

原代码存在的问题 #

• 首先存在异步逻辑错误：forEach 循环不支持异步等待，你在循环内加的 await 不会生效，函数最终会返回空数组
• 循环调用 findOne 会产生N次数据库请求，数据量大时性能损耗非常明显

单查询实现方案 #

可以用 MongoDB 的 $in 运算符一次性匹配所有目标值，仅发起一次数据库查询即可得到所有匹配结果。如果需要保持返回结果和原输入数组的顺序一致，需要额外做一层映射处理，因为 $in 返回的结果默认不会和传入数组的顺序对齐。

优化后代码：

async function myFunction(array) {
    if (!array || !array.length) return []
    // 一次查询拿到所有匹配的文档
    const docs = await dbModel.find({ element: { $in: array } })
    // 生成映射表，方便后续按原数组顺序匹配
    const docMap = new Map(docs.map(doc => [doc.element, doc]))
    // 按原数组顺序返回结果，找不到匹配的位置返回null，和原findOne逻辑对齐
    return array.map(element => docMap.get(element) || null)
}

该实现和你原有逻辑的输出完全一致，同时将N次数据库请求降为1次，性能提升非常明显。

内容的提问来源于stack exchange，提问作者Faisal Jamil Khan