如何使MongoDB $in查询结果按指定id顺序返回？

这问题我熟！MongoDB默认不会跟着$in里的数组顺序返回结果——它要么按文档的存储顺序，要么按索引顺序来返回，所以你才会得到不符合预期的排序。要实现你想要的自定义顺序，有几种实用的方法，我按推荐度给你列出来：

方法一：用聚合管道添加临时排序字段（通用且高效） #

这是最靠谱的方案，从MongoDB 3.4版本开始就支持，全程在数据库端完成排序，性能拉满。思路是先给每个匹配的文档加一个临时字段，值是它的id在你目标顺序数组里的索引位置，然后按这个索引排序，最后把临时字段删掉就行。

代码示例：

db.yourCollection.aggregate([
  // 先过滤出符合条件的文档
  { $match: { id: { $in: [2, 3, 1] } } },
  // 添加临时字段sortOrder，值为当前id在目标数组中的索引
  { $addFields: { sortOrder: { $indexOfArray: [[2, 3, 1], "$id"] } } },
  // 按sortOrder升序排列，这样就会跟着目标数组的顺序走
  { $sort: { sortOrder: 1 } },
  // 移除临时的sortOrder字段，返回干净的结果
  { $project: { sortOrder: 0 } }
])

方法二：MongoDB 5.2+ 可用的简化写法 #

如果你用的是MongoDB 5.2及以上版本，可以用$sortArray来简化操作，本质和方法一逻辑类似：

db.yourCollection.aggregate([
  { $match: { id: { $in: [2, 3, 1] } } },
  // 先把所有匹配的文档聚合到一个数组里
  { $group: { _id: null, docs: { $push: "$$CURRENT" } } },
  // 按目标数组的索引排序这个文档数组
  { $addFields: {
      docs: {
        $sortArray: {
          input: "$docs",
          sortBy: { $indexOfArray: [[2, 3, 1], "$id"] }
        }
      }
    }
  },
  // 把数组拆回单个文档
  { $unwind: "$docs" },
  // 输出最终结果
  { $replaceRoot: { newRoot: "$docs" } }
])

方法三：客户端侧排序（不推荐大数据量场景） #

如果你的数据量很小，也可以在查询到结果后，在代码里手动排序。比如用JavaScript的话：

const desiredOrder = [2, 3, 1];
// 先查询出所有匹配的文档
const rawResults = await db.yourCollection.find({ id: { $in: desiredOrder } }).toArray();
// 按目标顺序重新映射结果
const sortedResults = desiredOrder.map(targetId => rawResults.find(doc => doc.id === targetId));

⚠️ 注意：这个方法的缺点很明显——如果某个id在集合里不存在，会出现undefined；而且数据量大的时候，客户端排序的性能远不如数据库端处理。

总结一下，优先选方法一，兼容性好、效率高，几乎能覆盖所有场景。

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