MongoDB同时按两个字符串字段排序并过滤双NO记录的实现方法

实现思路 #

共有3种可落地的实现思路，可根据实际场景选择：

• 动态计算权重方案：使用聚合管道动态计算每条记录的排序优先级，适合规则灵活、查询频率低的场景
• 普通查询内置表达式方案：直接在find查询的sort阶段使用条件表达式实现排序，不需要走聚合流程，写法更简洁
• 预计算权重方案：提前在文档中新增权重字段，写入/更新数据时同步计算权重值，查询时直接按权重排序，性能最高，适合查询频率高、规则稳定的场景

具体实现代码 #

方案1：聚合管道实现（扩展性最高） #

db.yourCollectionName.aggregate([
  // 过滤掉isAvailable和isSuper均为NO的记录
  {
    $match: {
      $or: [
        { isAvailable: "YES" },
        { isSuper: "YES" }
      ]
    }
  },
  // 计算自定义排序权重
  {
    $addFields: {
      customSortWeight: {
        $switch: {
          branches: [
            // 双YES优先级最高，权重设为1
            { case: { $and: [ { $eq: ["$isAvailable", "YES"] }, { $eq: ["$isSuper", "YES"] } ] }, then: 1 },
            // 任意一个为YES优先级次之，权重设为2
            { case: { $or: [ { $eq: ["$isAvailable", "YES"] }, { $eq: ["$isSuper", "YES"] } ] }, then: 2 }
          ],
          default: 3
        }
      }
    }
  },
  // 按权重升序排序，权重越小优先级越高
  {
    $sort: {
      customSortWeight: 1,
      // 相同优先级下可追加其他排序规则，比如按ID升序，不需要可删除
      ID: 1
    }
  },
  // 可选：隐藏返回结果中的权重字段
  {
    $project: {
      customSortWeight: 0
    }
  }
])

方案2：普通find查询实现（写法更简洁） #

db.yourCollectionName.find(
  // 过滤条件：排除双NO的记录
  { $or: [ { isAvailable: "YES" }, { isSuper: "YES" } ] }
).sort({
  // 内置条件表达式计算排序优先级
  $expr: {
    $cond: {
      if: { $and: [ { $eq: ["$isAvailable", "YES"] }, { $eq: ["$isSuper", "YES"] } ] },
      then: 1,
      else: 2
    }
  },
  // 相同优先级追加排序规则，不需要可删除
  ID: 1
})

方案3：预计算权重实现（性能最高） #

1. 先给已有数据批量初始化权重字段：

db.yourCollectionName.updateMany(
  {},
  [
    {
      $set: {
        customSortWeight: {
          $switch: {
            branches: [
              { case: { $and: [ { $eq: ["$isAvailable", "YES"] }, { $eq: ["$isSuper", "YES"] } ] }, then: 1 },
              { case: { $or: [ { $eq: ["$isAvailable", "YES"] }, { $eq: ["$isSuper", "YES"] } ] }, then: 2 }
            ],
            default: 3
          }
        }
      }
    }
  ]
)

2. 后续写入/更新isAvailable、isSuper字段时同步更新customSortWeight，查询时直接过滤+排序即可：

// 查询语句
db.yourCollectionName.find(
  { customSortWeight: { $ne: 3 } }
).sort({ customSortWeight: 1, ID: 1 })

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