MongoDB查询：从指定元素开始返回排序后的集合切片

当然可以实现！你可以借助MongoDB的聚合框架，通过几个阶段的组合来完成这个需求，完全匹配你描述的场景。下面是具体的实现方案：

核心思路 #

1. 先按section字段对所有文档进行升序排序，得到你需要的有序文档列表。
2. 将排序后的文档合并到一个数组中，方便后续定位和截取操作。
3. 找到目标sequence（即你说的X）在有序数组中的位置。
4. 从该位置开始截取2个文档，同时支持边界循环（比如当目标是最后一个文档时，自动取第一个文档作为第二个结果）。

具体聚合查询代码 #

假设你的集合名为yourCollection，把下面代码中的X替换为你需要的目标sequence值即可：

db.yourCollection.aggregate([
  // 阶段1：按section字段升序排序，得到有序文档列表
  { $sort: { section: 1 } },
  // 阶段2：将所有排序后的文档存入一个数组，方便后续操作
  { $group: { _id: null, docs: { $push: "$$ROOT" } } },
  // 阶段3：定位目标位置并截取2个文档（支持循环边界）
  {
    $project: {
      targetDocs: {
        $slice: [
          { $concatArrays: ["$docs", "$docs"] }, // 拼接数组实现循环取数
          { $indexOfArray: ["$docs.sequence", X] }, // 找到目标sequence的索引
          2 // 截取数量
        ]
      },
      _id: 0
    }
  },
  // 阶段4：将截取的数组展开为单个文档
  { $unwind: "$targetDocs" },
  // 阶段5：将展开的文档作为根文档输出
  { $replaceRoot: { newRoot: "$targetDocs" } }
])

代码解释 #

• $sort：按section的字母顺序排序，和你例子中的排序结果完全一致。
• $group：把排序后的所有文档收集到docs数组里，这样我们可以通过数组操作来定位和截取。
• $project：
  • $indexOfArray：遍历docs数组的sequence字段，找到等于X的文档的索引位置。
  • $concatArrays：把docs数组和自身拼接，这样当目标位置是数组末尾时，截取操作会自动从数组开头取数，实现循环效果。
  • $slice：从目标索引开始，截取2个元素。
• $unwind 和 $replaceRoot：把截取到的数组展开成独立的文档，最终输出你需要的结果格式。

验证你的示例场景 #

1. 当X=5时：
   排序后的数组是[A,B,C,D,E,F]，sequence=5对应文档E，索引为4。$slice从索引4取2个元素，得到[E,F]，完全符合你的需求。
2. 当X=1时：
   sequence=1对应文档B，索引为1，$slice取到[B,C]，和你预期的结果一致。

注意事项 #

• 确保sequence字段是唯一的，否则$indexOfArray只会返回第一个匹配的索引，可能导致结果不符合预期。
• 如果不需要循环边界的功能（比如目标是最后一个文档时只返回它自己），可以去掉$concatArrays，直接用"$docs"作为$slice的第一个参数。

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