MongoDB按单字段分组并关联用户信息的技术实现问题

1. 按REF_NotifierID分组聚合 #

先执行分组操作，将同一通知者ID下的用户ID聚合为指定结构，使用$group和$push实现：

db.notifications.aggregate([
  {
    $group: {
      _id: "$REF_NotifierID",
      notifiedUsers: {
        $push: {
          REF_UserID: "$REF_UserID"
          // 可添加其他需要保留的字段，如通知时间、类型等
        }
      }
    }
  }
])

这段代码会将所有REF_NotifierID相同的记录分组，把每组内的REF_UserID组装成notifiedUsers数组，结构符合需求。

2. 关联users表补全用户信息 #

直接使用$lookup无法正确关联数组内的REF_UserID，因为$lookup默认不会遍历数组元素。以下两种方法可以解决这个问题：

方法1：嵌套管道式$lookup（推荐） #

通过$lookup的pipeline参数批量查询用户信息，再用$map和$filter将用户信息映射回对应的数组元素：

db.notifications.aggregate([
  // 第一步：分组聚合
  {
    $group: {
      _id: "$REF_NotifierID",
      notifiedUsers: { $push: { REF_UserID: "$REF_UserID" } }
    }
  },
  // 第二步：批量查询users表中所有相关用户
  {
    $lookup: {
      from: "users",
      let: { userIds: "$notifiedUsers.REF_UserID" },
      pipeline: [
        { $match: { $expr: { $in: ["$_id", "$$userIds"] } } },
        { $project: { firstName: 1, lastName: 1, _id: 1 } }
      ],
      as: "userDetails"
    }
  },
  // 第三步：将用户信息合并到notifiedUsers数组
  {
    $addFields: {
      notifiedUsers: {
        $map: {
          input: "$notifiedUsers",
          as: "user",
          in: {
            $mergeObjects: [
              "$$user",
              { $arrayElemAt: [
                  { $filter: {
                      input: "$userDetails",
                      cond: { $eq: ["$$this._id", "$$user.REF_UserID"] }
                    }
                  },
                  0
                ]
              }
            ]
          }
        }
      },
      userDetails: "$$REMOVE" // 清理临时字段
    }
  }
])

方法2：拆分数组再关联重组 #

先拆分组后的数组，关联用户信息后重新分组：

db.notifications.aggregate([
  // 第一步：分组聚合
  {
    $group: {
      _id: "$REF_NotifierID",
      notifiedUsers: { $push: { REF_UserID: "$REF_UserID" } }
    }
  },
  // 拆分数组，每条记录对应一个用户ID
  { $unwind: "$notifiedUsers" },
  // 关联users表获取单条用户信息
  {
    $lookup: {
      from: "users",
      localField: "notifiedUsers.REF_UserID",
      foreignField: "_id",
      as: "userInfo"
    }
  },
  // 将用户信息合并到notifiedUsers对象
  {
    $addFields: {
      "notifiedUsers.firstName": { $arrayElemAt: ["$userInfo.firstName", 0] },
      "notifiedUsers.lastName": { $arrayElemAt: ["$userInfo.lastName", 0] },
      userInfo: "$$REMOVE"
    }
  },
  // 重新按通知者ID分组，重组数组
  {
    $group: {
      _id: "$_id",
      notifiedUsers: { $push: "$notifiedUsers" }
    }
  }
])

常见问题说明 #

如果之前使用$lookup未得到预期结果，通常是因为分组后的notifiedUsers是数组，直接指定localField: "notifiedUsers.REF_UserID"会将整个数组作为查询条件，而非遍历每个元素。上述两种方法分别通过嵌套管道批量查询、拆分数组单条关联的方式解决了这个问题。

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