MongoDB：如何按指定优先级筛选并返回首个符合条件的文档？

给定如下MongoDB文档结构：

{
   "name": "somename",
   "level": "123",
   "nested":[ 
        {
          "somefield": "test",
          "file": {
             "level":"123"
         }
     },
     {
         "somefield": "test2",
         "file": {
            "level":"124"
         }
     },
     {
         "somefield": "test3",
         "file": {
            "level":"123"
         }
     }
  ]
}

经过$unwind和$match阶段匹配根级level与nested.file.level相等的条件后，得到2个文档：

[
  {
    "level": "123",
    "name": "somename",
    "nested": {
      "file": {
        "level": "123"
      },
      "somefield": "test"
    }
  },
  {
    "level": "123",
    "name": "somename",
    "nested": {
      "file": {
        "level": "123"
      },
      "somefield": "test3"
    }
  }
]

需要按指定优先级顺序（test > test2 > test3 > test4...）匹配somefield字段，仅返回第一个符合优先级的文档。此前尝试用$switch编写$project阶段，但该方法会对每个文档单独处理，返回多个匹配结果，无法实现需求。

核心思路是给每个符合条件的文档标记优先级权重，按权重排序后取第一个即可，具体步骤如下：

1. 标记优先级权重 #

用$addFields阶段，通过$indexOfArray根据指定的优先级列表，获取每个文档nested.somefield对应的优先级索引——索引越小，优先级越高：

$addFields: {
  priority: {
    $indexOfArray: [["test", "test2", "test3", "test4", "test5"], "$nested.somefield"]
  }
}

• 匹配到优先级项时，返回对应数组索引（比如test对应0，test3对应2）；
• 无匹配项时返回-1，后续可通过$match过滤这类文档。

2. 过滤无效项并排序 #

添加$match过滤掉优先级为-1的文档，再用$sort按priority升序排列（优先级高的文档排在前面）：

$match: {
  priority: { $ne: -1 }
},
$sort: { priority: 1 }

3. 取首个匹配文档 #

用$limit: 1获取排序后的第一个文档，再通过$project整理最终需要的字段：

$limit: 1,
$project: {
  _id: 0,
  name: 1,
  level: 1,
  setting: "$nested"
}

完整聚合管道 #

整合所有步骤后的完整管道：

[
  // 原有筛选阶段
  { $unwind: "$nested" },
  { $match: { $expr: { $eq: ["$level", "$nested.file.level"] } } },
  
  // 优先级处理阶段
  {
    $addFields: {
      priority: {
        $indexOfArray: [["test", "test2", "test3", "test4", "test5"], "$nested.somefield"]
      }
    }
  },
  { $match: { priority: { $ne: -1 } } },
  { $sort: { priority: 1 } },
  { $limit: 1 },
  {
    $project: {
      _id: 0,
      name: 1,
      level: 1,
      setting: "$nested"
    }
  }
]

执行结果 #

最终返回优先级最高的匹配文档：

{
  "name": "somename",
  "level": "123",
  "setting": {
    "file": { "level": "123" },
    "somefield": "test"
  }
}

补充说明 #

• 调整优先级顺序只需修改$indexOfArray中的数组顺序；
• 新增优先级值直接往数组里添加对应字段即可；
• 该方法能全局处理所有符合条件的文档，确保只返回优先级最高的那一个。

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