如何在MongoDB聚合中设置排序优先级

我来帮你搞定这个自定义排序的需求！要实现你想要的结果，我们可以通过在聚合管道中添加辅助字段来区分缺货商品，再结合多字段排序来达成目标。

完整聚合查询代码 #

db.collection.aggregate([
  {
    $addFields: {
      // 创建辅助字段：缺货标记，0表示有库存，1表示无库存
      isOutOfStock: { $cond: { if: { $eq: ["$itemsInStock", 0] }, then: 1, else: 0 } }
    }
  },
  {
    $sort: {
      // 优先按缺货标记升序（有库存的在前，缺货的在后）
      isOutOfStock: 1,
      // 然后按wishlist降序（true的商品排在前面）
      wishlist: -1,
      // 最后按productName升序排序
      productName: 1
    }
  },
  {
    // 可选：如果不需要保留辅助字段，可以移除它
    $project: {
      isOutOfStock: 0
    }
  }
])

代码解释 #

让我们拆解一下这个管道的作用：

• $addFields阶段：通过$cond条件判断生成isOutOfStock字段，把itemsInStock为0的商品标记为1，其余为0。这个字段是实现“缺货商品排最后”的关键。
• $sort阶段：排序优先级依次是：
  • isOutOfStock: 1：升序排列，确保有库存的商品（标记0）全部排在缺货商品（标记1）前面；
  • wishlist: -1：降序排列，让wishlist: true的商品排在wishlist: false的前面；
  • productName: 1：升序排列，同组内按商品名称字母顺序排序。
• $project阶段（可选）：如果不需要在结果中保留辅助字段isOutOfStock，可以用这个阶段把它移除，让输出结果和你期望的完全一致。

验证结果 #

运行这个查询后，输出结果会和你期望的完全匹配：

[
  { wishlist: true, productName: "product1", itemsInStock: 5 },
  { wishlist: true, productName: "product4", itemsInStock: 17 },
  { wishlist: false, productName: "product2", itemsInStock: 25 },
  { wishlist: false, productName: "product5", itemsInStock: 12 },
  { wishlist: true, productName: "product6", itemsInStock: 0 },
  { wishlist: false, productName: "product3", itemsInStock: 0 }
]

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