MongoDB聚合中转换数组内字符串user_id为ObjectId并关联用户数据

针对嵌套数组里的字符串user_id转ObjectId并关联users表的需求，可通过MongoDB聚合管道实现，以下是两种常用方案：

方案一：展开数组后关联重组（适合小数据量场景） #

通过两次$unwind展开嵌套数组，转换ID类型后执行关联，最后重新组合回原结构，逻辑直观易懂。

db.your_collection.aggregate([
  // 展开外层的newAssessmentReport数组
  { $unwind: "$newAssessmentReport" },
  // 展开内层的investment_report数组
  { $unwind: "$newAssessmentReport.investment_report" },
  // 将字符串user_id转换为ObjectId类型
  {
    $set: {
      "newAssessmentReport.investment_report.user_id": {
        $toObjectId: "$newAssessmentReport.investment_report.user_id"
      }
    }
  },
  // 关联users表，匹配用户ID
  {
    $lookup: {
      from: "users",
      localField: "newAssessmentReport.investment_report.user_id",
      foreignField: "_id",
      as: "newAssessmentReport.investment_report.user_info"
    }
  },
  // 将lookup返回的数组转为单个用户对象
  {
    $set: {
      "newAssessmentReport.investment_report.user_info": {
        $arrayElemAt: ["$newAssessmentReport.investment_report.user_info", 0]
      }
    }
  },
  // 重新组合investment_report数组
  {
    $group: {
      _id: {
        _id: "$_id",
        report_id: "$newAssessmentReport._id"
      },
      newAssessmentReport: { $first: "$newAssessmentReport" },
      investment_report: { $push: "$newAssessmentReport.investment_report" }
    }
  },
  // 更新报告对象中的investment_report字段
  {
    $set: {
      "newAssessmentReport.investment_report": "$investment_report"
    }
  },
  // 重新组合newAssessmentReport数组
  {
    $group: {
      _id: "$_id._id",
      newAssessmentReport: { $push: "$newAssessmentReport" }
    }
  }
])

方案二：嵌套管道式关联（适合大数据量场景） #

通过$map遍历数组转换ID类型，一次性查询所有需要的用户数据，再通过$filter匹配到对应项，避免频繁展开和重组，性能更优。

db.your_collection.aggregate([
  // 遍历两层数组，将所有user_id字符串转为ObjectId
  {
    $set: {
      "newAssessmentReport": {
        $map: {
          input: "$newAssessmentReport",
          as: "report",
          in: {
            // 保留原报告的所有字段
            ..."$$report",
            investment_report: {
              $map: {
                input: "$$report.investment_report",
                as: "item",
                in: {
                  ..."$$item",
                  user_id: { $toObjectId: "$$item.user_id" }
                }
              }
            }
          }
        }
      }
    }
  },
  // 一次性查询所有关联的用户数据
  {
    $lookup: {
      from: "users",
      let: { reports: "$newAssessmentReport" },
      pipeline: [
        {
          $match: {
            $expr: {
              $in: ["$_id", { $reduce: {
                input: "$$reports",
                initialValue: [],
                in: { $concatArrays: ["$$value", { $map: { input: "$$this.investment_report", as: "ir", in: "$$ir.user_id" } }] }
              } }]
            }
          }
        }
      ],
      as: "users_list"
    }
  },
  // 将用户信息匹配到对应的investment_report项中
  {
    $set: {
      "newAssessmentReport": {
        $map: {
          input: "$newAssessmentReport",
          as: "report",
          in: {
            ..."$$report",
            investment_report: {
              $map: {
                input: "$$report.investment_report",
                as: "item",
                in: {
                  ..."$$item",
                  user_info: {
                    $arrayElemAt: [
                      { $filter: { input: "$users_list", cond: { $eq: ["$$this._id", "$$item.user_id"] } } },
                      0
                    ]
                  }
                }
              }
            }
          }
        }
      }
    }
  },
  // 移除临时的用户列表字段
  { $unset: "users_list" }
])

注意事项 #

• 替换your_collection为你实际操作的集合名称。
• $toObjectId要求字符串为有效的MongoDB ObjectId格式，若存在无效ID，可通过$cond添加判断逻辑过滤。
• 若不需要保留原user_id字段，可在$map中直接替换或后续用$unset移除。

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