MongoDB聚合查询：如何为用户添加排名并关联位置数据分组

我明白你的需求了——你已经通过聚合框架得到了按特定规则排序的用户列表，现在要把这个排名关联到包含user和location字段的集合，最终算出每个地区的最佳用户排名。确实MongoDB没有直接支持你设想的那种$rank带字典映射的语法，但我们有两种靠谱的方法来实现：

方法一：直接在聚合管道中使用数组映射排名 #

这种方法不需要额外创建集合，适合你的排名列表固定且长度不大的场景。核心思路是把已有的排序用户列表嵌入聚合管道，通过$indexOfArray获取用户在列表中的位置，再转换为排名。

完整聚合管道示例 #

假设你的目标集合名为user_locations，文档结构为{ user: "u1", location: "China" }：

db.user_locations.aggregate([
  // 步骤1：引入你之前得到的排序用户数组
  {
    $addFields: {
      rankedUsers: ["u1", "u2", "u3"] // 这里替换成你的实际排序用户列表
    }
  },
  // 步骤2：计算每个用户的排名（数组索引从0开始，所以加1得到实际排名）
  {
    $addFields: {
      UserRank: {
        $add: [
          { $indexOfArray: ["$rankedUsers", "$user"] },
          1
        ]
      }
    }
  },
  // 步骤3：过滤掉不在排名列表中的用户（可选，根据你的数据情况决定）
  {
    $match: {
      UserRank: { $ne: 0 } // 不在列表中的用户会得到0，直接过滤
    }
  },
  // 步骤4：按地区分组，取最小的排名作为该地区的最佳排名
  {
    $group: {
      _id: "$location",
      BestUserRank: { $min: "$UserRank" }
    }
  },
  // 步骤5：调整输出格式，让结果更清晰
  {
    $project: {
      _id: 0,
      Location: "$_id",
      BestUserRank: 1
    }
  }
])

方法二：通过临时集合关联排名 #

如果你的排名列表比较长，或者后续需要重复使用这份排名数据，可以先把排名存入一个临时集合，再通过$lookup关联查询。

步骤1：创建并插入排名数据 #

// 创建临时集合并插入排名记录
db.user_ranks.insertMany([
  { user: "u1", rank: 1 },
  { user: "u2", rank: 2 },
  { user: "u3", rank: 3 }
])

步骤2：执行聚合关联与分组 #

db.user_locations.aggregate([
  // 步骤1：关联临时排名集合
  {
    $lookup: {
      from: "user_ranks",
      localField: "user",
      foreignField: "user",
      as: "rankInfo"
    }
  },
  // 步骤2：展开关联得到的数组（每个用户只会匹配一条排名记录）
  {
    $unwind: "$rankInfo"
  },
  // 步骤3：提取排名字段到UserRank
  {
    $addFields: {
      UserRank: "$rankInfo.rank"
    }
  },
  // 步骤4：按地区分组求最小排名
  {
    $group: {
      _id: "$location",
      BestUserRank: { $min: "$UserRank" }
    }
  },
  // 步骤5：调整输出格式
  {
    $project: {
      _id: 0,
      Location: "$_id",
      BestUserRank: 1
    }
  }
])

两种方法最终都会输出你想要的结果：

Location BestUserRank
China    1
Brazil   1
Australia 3

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