MongoDB聚合条件Lookup问题：动态关联对应Staff字段数据

需求是在StaffOneToOne集合的聚合操作中，根据当前用户ID（req.user._id）动态关联pharmacists集合的数据：

• 当当前用户ID匹配fromStaffId时，关联toStaffId对应的药剂师数据
• 当当前用户ID匹配toStaffId时，关联fromStaffId对应的药剂师数据

原代码无法正常工作，现有代码和集合文档示例如下：

原聚合代码 #

StaffOneToOne.aggregate([
  {"$match": {$or: [{ fromStaffId: req.user._id }, { toStaffId: req.user._id }]}},
  {$cond:{
    if:{fromStaffId:req.user._id},
    then:{$lookup: {
      from: 'pharmacists',
      localField: 'fromStaffId',
      foreignField: '_id',
      as: 'p_list'
    }},
    else:{$lookup: {
      from: 'pharmacists',
      localField: 'fromStaffId',
      foreignField: '_id',
      as: 'p_list'}}
  }}
])

StaffOneToOne集合文档示例 #

{
    "_id" : ObjectId("6310901b86c6077c5d3be84f"),
    "message" : "Hello Hau",
    "date" : 1662029851620.0,
    "messageType" : 0,
    "staffType" : 0,
    "isSeen" : 0,
    "image" : "",
    "toStaffId" : ObjectId("60c8908ef1c80777f6c78460"),
    "fromStaffId" : ObjectId("608599c361480b4cc2dd4946"),
    "sentBy" : ObjectId("608599c361480b4cc2dd4946"),
    "createdAt" : ISODate("2022-09-01T10:57:31.902Z"),
    "updatedAt" : ISODate("2022-09-01T10:57:31.902Z"),
    "__v" : 0
}

问题分析 #

原代码存在两个核心错误：

1. $cond不能直接作为独立的聚合阶段使用，聚合阶段必须是MongoDB支持的阶段类型（如$addFields、$lookup等）
2. 条件判断和关联字段逻辑错误：if条件未用$eq做字段与变量的比较，且then/else中的localField未按需求切换为对应关联ID

正确解决方案 #

方案1：添加辅助字段后关联（可读性更强）

先通过$addFields生成标记目标关联ID的字段，再基于该字段执行$lookup：

StaffOneToOne.aggregate([
  // 筛选当前用户参与的对话记录
  {
    "$match": {
      "$or": [
        { fromStaffId: req.user._id },
        { toStaffId: req.user._id }
      ]
    }
  },
  // 生成目标关联ID：当前用户是发送方则取接收方ID，反之取发送方ID
  {
    "$addFields": {
      "targetStaffId": {
        "$cond": {
          if: { "$eq": ["$fromStaffId", req.user._id] },
          then: "$toStaffId",
          else: "$fromStaffId"
        }
      }
    }
  },
  // 关联pharmacists集合
  {
    "$lookup": {
      from: "pharmacists",
      localField: "targetStaffId",
      foreignField: "_id",
      as: "p_list"
    }
  },
  // 可选：移除辅助字段，精简结果结构
  {
    "$project": {
      "targetStaffId": 0
    }
  }
])

方案2：直接在lookup中使用条件表达式（更简洁）

通过lookup的pipeline参数，直接在关联阶段内完成条件判断：

StaffOneToOne.aggregate([
  {
    "$match": {
      "$or": [
        { fromStaffId: req.user._id },
        { toStaffId: req.user._id }
      ]
    }
  },
  {
    "$lookup": {
      from: "pharmacists",
      // 定义可在pipeline中使用的变量
      let: {
        currentUserId: req.user._id,
        fromId: "$fromStaffId",
        toId: "$toStaffId"
      },
      // 关联时的条件判断逻辑
      pipeline: [
        {
          "$match": {
            "$expr": {
              "$eq": [
                "$_id",
                { "$cond": [
                  { "$eq": ["$$currentUserId", "$$fromId"] },
                  "$$toId",
                  "$$fromId"
                ]}
              ]
            }
          }
        }
      ],
      as: "p_list"
    }
  }
])

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