如何在MongoDB中按数组批量查询？优化多用户消息ID查询性能

你的循环写法确实会产生和用户数相等的数据库请求，当用户量上去之后，性能肯定会掉链子。这里给你两个更优的解决方案，从性能最优到代码改动最小排序：

方案一：使用MongoDB聚合框架（推荐） #

通过聚合操作，只需要一次数据库请求就能完成所有用户的查询，这是性能最好的方式。核心思路是先匹配目标用户，然后按userId分组，每组内按_id倒序取前20条记录，最后提取messageId。

代码示例 #

const messageIds = {};

const pipeline = [
  // 匹配目标userId数组
  { $match: { userId: { $in: userIds } } },
  // 按userId分组，同时保留每组的记录（带_id用于排序）
  {
    $group: {
      _id: "$userId",
      messages: {
        $push: { messageId: "$messageId", _id: "$_id" }
      }
    }
  },
  // 对每组的记录按_id倒序排序，再截取前20条
  {
    $project: {
      _id: 0,
      userId: "$_id",
      latestMessages: {
        $slice: [
          { $sortArray: { input: "$messages", sortBy: { _id: -1 } } },
          20
        ]
      }
    }
  }
];

const results = await Thread.aggregate(pipeline).exec();

// 整理成需要的对象格式
results.forEach(item => {
  messageIds[item.userId] = item.latestMessages.map(msg => msg.messageId);
});

性能优化关键 #

一定要给Thread集合创建复合索引：

db.Thread.createIndex({ userId: 1, _id: -1 });

这个索引会让$match和后续的排序操作都能高效利用索引，避免全表扫描，大幅提升聚合速度。

方案二：并行执行查询（代码改动最小） #

如果你暂时不想用聚合框架，可以把循环里的异步请求改成并行执行，用Promise.all同时发起所有查询，减少总等待时间（毕竟循环里是串行等待每个请求完成，并行可以同时处理）。

代码示例 #

const promiseList = userIds.map(async userId => {
  const results = await Thread.find({ userId }).sort({ _id: -1 }).limit(20).exec();
  return [userId, results.map(result => result.messageId)];
});

const results = await Promise.all(promiseList);
const messageIds = Object.fromEntries(results);

注意事项 #

这种方式请求数还是和用户数一致，所以如果用户量很大（比如几百上千个），还是会给数据库造成压力，更适合用户量较小的场景。

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