如何实现MongoDB查询返回指定用户居中、前后各2条的共5条用户数据

核心逻辑 #

要实现返回目标用户及前后各2位共5条排序后的数据，核心分为两步：

• 先通过唯一字段username定位到目标用户的score值以及唯一标识（用于同分场景排序）
• 基于score排序规则，分别拉取高于目标的前2位、低于目标的后2位，加上目标本身合并后重排，即可得到目标用户在中间的5条结果

为了避免score同分导致排序不稳定，所有排序逻辑都加入_id作为第二排序字段，你也可以替换为其他你需要的唯一排序字段。

方案1：单聚合查询（性能最优，适合生产环境） #

直接通过一次聚合查询完成所有逻辑，无需多次和数据库交互：

// 替换为你要查询的目标用户名
const targetUsername = "your_target_username";

db.users.aggregate([
  // 匹配目标用户，获取其score值
  { $match: { username: targetUsername } },
  { $project: { targetScore: "$score", targetId: "$_id", _id: 0 } },
  // 关联查询符合要求的所有用户
  { $lookup: {
    from: "users",
    let: { targetScore: "$targetScore", targetId: "$targetId" },
    pipeline: [
      // 查询score大于等于目标的用户，取前3条（包含目标本身）
      { $match: {
        $expr: {
          $or: [
            { $gt: ["$score", "$$targetScore"] },
            { $and: [
              { $eq: ["$score", "$$targetScore"] },
              { $gte: ["$_id", "$$targetId"] }
            ]}
          ]
        }
      }},
      { $sort: { score: 1, _id: 1 } },
      { $limit: 3 },
      // 合并查询score小于目标的用户，取后2条
      { $unionWith: {
        coll: "users",
        pipeline: [
          { $match: {
            $expr: {
              $or: [
                { $lt: ["$score", "$$targetScore"] },
                { $and: [
                  { $eq: ["$score", "$$targetScore"] },
                  { $lt: ["$_id", "$$targetId"] }
                ]}
              ]
            }
          }},
          { $sort: { score: -1, _id: -1 } },
          { $limit: 2 }
        ]
      }},
      // 最终统一排序得到正确顺序
      { $sort: { score: 1, _id: 1 } }
    ],
    as: "finalResult"
  } },
  // 展开结果直接返回用户数据
  { $unwind: "$finalResult" },
  { $replaceRoot: { newRoot: "$finalResult" } }
])

如果需要按score降序排序，只需要把所有sort配置中的score: 1改为score: -1，同时对应调整$gte/$lte的匹配逻辑即可。

方案2：分两次查询（逻辑简单，易调试） #

适合数据量不大或者需要灵活调整逻辑的场景：

const targetUsername = "your_target_username";
// 第一步：查询目标用户
const targetUser = db.users.findOne({ username: targetUsername });
if (!targetUser) return [];

// 第二步：查询比目标排名高的前2位 + 目标本身，共3条
const higherList = db.users.find({
  $or: [
    { score: { $gt: targetUser.score } },
    { score: targetUser.score, _id: { $gte: targetUser._id } }
  ]
}).sort({ score: 1, _id: 1 }).limit(3).toArray();

// 第三步：查询比目标排名低的后2位
const lowerList = db.users.find({
  $or: [
    { score: { $lt: targetUser.score } },
    { score: targetUser.score, _id: { $lt: targetUser._id } }
  ]
}).sort({ score: -1, _id: -1 }).limit(2).toArray();

// 合并结果得到最终5条数据，目标用户固定在第3位
const result = lowerList.reverse().concat(higherList);

性能优化建议 #

给score和*_id*加联合索引，可大幅提升查询性能：

db.users.createIndex({ score: 1, _id: 1 })

如果你的第二排序字段不是_id，把索引中的_id替换为对应的字段即可。

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