MongoDB聚合管道如何实现条件触发排序否则保留文档原顺序

你要的条件排序逻辑可以通过以下聚合管道实现，分为高版本兼容方案和低版本兼容方案两种： #

高版本方案（MongoDB 5.2+ 推荐） #

核心思路是先给所有文档标记原始插入顺序，再按规则做条件排序：

[
  // 标记文档原始顺序
  {
    $setWindowFields: {
      sortBy: { $natural: 1 },
      output: {
        original_idx: { $documentNumber: {} }
      }
    }
  },
  // 条件排序：has_presence为true按name排序，为false保留原始顺序
  {
    $sort: {
      has_presence: -1,
      name: 1,
      original_idx: 1
    }
  },
  // 移除辅助字段，还原输出结构
  {
    $unset: "original_idx"
  }
]

低版本兼容方案（MongoDB 3.2+ 可用） #

如果使用的MongoDB版本不支持$setWindowFields，可以用分组拆分数组的方式生成原始顺序标记：

[
  // 收集所有文档到数组，保留原始插入顺序
  {
    $group: {
      _id: null,
      doc_list: { $push: "$$ROOT" }
    }
  },
  // 拆分数组同时生成原始顺序索引
  {
    $unwind: {
      path: "$doc_list",
      includeArrayIndex: "original_idx"
    }
  },
  // 还原文档结构
  {
    $replaceRoot: {
      newRoot: { $mergeObjects: [ "$doc_list", { original_idx: "$original_idx" } ] }
    }
  },
  // 条件排序，逻辑同上
  {
    $sort: {
      has_presence: -1,
      name: 1,
      original_idx: 1
    }
  },
  // 移除辅助字段
  {
    $unset: "original_idx"
  }
]

逻辑说明 #

两种方案的排序规则完全一致：

• has_presence为true的组内，按name字段升序排列，完全匹配场景1的要求
• has_presence为false的组内，按original_idx也就是原始插入顺序升序排列，完全匹配场景2的要求
• 如果不需要把has_presence: true的文档整体排在前面，调整$sort阶段的字段优先级即可。

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