MongoDB聚合查询：排序后获取目标文档前后文档的问题

问题分析 #

你需要实现的逻辑是：先按type升序、name升序完成二级排序，再针对某个虚拟文档（未实际存入集合）找到它在排序序列中的位置，获取其前后各2个文档。当前的查询逻辑错误，导致部分场景返回结果不符合预期——比如目标为{ name: 'K', type: 'Effect' }时，正确后续文档应为{ name: 'J', type: 'Magic' }，但当前查询错误返回了{ name: 'V', type: 'Trap' }。

错误根源在于$match条件逻辑：你使用了同时满足type >= 目标type和name > 目标name，而正确逻辑应为**type > 目标type，或者type == 目标type且name > 目标name**。

测试数据 #

db.test.insertMany([
   { name: 'V', type: 'Trap' }, 
   { name: 'J', type: 'Magic' },
   { name: 'B', type: 'Effect' }
])

预期排序结果 #

执行以下聚合管道可得到正确的二级排序：

db.test.aggregate([{ "$sort": { type: 1, name: 1 } }])

结果：

{ name: 'B', type: 'Effect' }
{ name: 'J', type: 'Magic' }
{ name: 'V', type: 'Trap' }

正确的后续文档查询 #

调整$match为逻辑或条件，匹配所有在目标文档之后的项，再取第一个：

[
  { $sort: { type: 1, name: 1 } },
  {
    $match: {
      $or: [
        { type: { $gt: <target_type> } },
        { type: <target_type>, name: { $gt: <target_name> } }
      ]
    }
  },
  { $limit: 1 }
]

测试验证：

• 目标{ name: 'K', type: 'Trap' }：匹配type == 'Trap'且name > 'K'得到V，返回正确
• 目标{ name: 'C', type: 'Effect' }：匹配type > 'Effect'得到J和V，排序后取第一个J，返回正确
• 目标{ name: 'K', type: 'Effect' }：匹配type > 'Effect'得到J和V，排序后取第一个J，返回正确

正确的前置文档查询 #

前置文档逻辑为type < 目标type，或者type == 目标type且name < 目标name，可通过倒序取第一个实现：

[
  { $sort: { type: 1, name: 1 } },
  {
    $match: {
      $or: [
        { type: { $lt: <target_type> } },
        { type: <target_type>, name: { $lt: <target_name> } }
      ]
    }
  },
  { $sort: { type: -1, name: -1 } }, // 倒序后取第一个等价于原排序的最后一个匹配项
  { $limit: 1 }
]

也可以用$group和$last直接获取原排序的最后一个匹配项：

[
  { $sort: { type: 1, name: 1 } },
  {
    $match: {
      $or: [
        { type: { $lt: <target_type> } },
        { type: <target_type>, name: { $lt: <target_name> } }
      ]
    }
  },
  { $group: { _id: null, lastDoc: { $last: "$$ROOT" } } },
  { $replaceRoot: { newRoot: "$lastDoc" } }
]

扩展：一次性获取前后各2个文档 #

若要批量获取目标位置前后各2个文档，可通过添加序号字段后筛选：

[
  { $sort: { type: 1, name: 1 } },
  // 为每个文档添加排序后的自增序号
  { $setWindowFields: {
      sortBy: { type: 1, name: 1 },
      output: { index: { $documentNumber: {} } }
    }
  },
  // 计算目标文档的虚拟序号
  { $facet: {
      targetIndex: [
        { $match: {
            $or: [
              { type: { $lt: <target_type> } },
              { type: <target_type>, name: { $lt: <target_name> } }
            ]
          }
        },
        { $count: "count" }
      ],
      allDocs: [ { $match: {} } ]
    }
  },
  { $set: { targetPos: { $add: [ { $arrayElemAt: [ "$targetIndex.count", 0 ] }, 1 ] } } },
  { $unwind: "$allDocs" },
  // 筛选目标位置前后各2个文档
  { $match: {
      $expr: {
        $and: [
          { $gte: [ "$allDocs.index", { $subtract: [ "$targetPos", 2 ] } ] },
          { $lte: [ "$allDocs.index", { $add: [ "$targetPos", 2 ] } ] }
        ]
      }
    }
  },
  { $replaceRoot: { newRoot: "$allDocs" } },
  { $sort: { index: 1 } }
]

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