Mongoose条件更新：单查询实现数组的选择性$pull操作

需求描述 #

现有查询用于从User文档的requests和followers数组中移除指定userId：

User.updateOne(
    { _id: otherUserId },
    { $pull: { requests: userId, followers: userId } }
)

需要修改为：若userId存在于requests数组中，仅移除requests中的该值；若userId不在requests数组中，则移除followers中的该值，要求用单个查询实现。

错误代码分析 #

尝试的以下代码出现CastError：

User.updateOne(
    { _id: otherUserId },
    { $pull: { requests: userId, followers: {$nin: [new mongoose.Types.ObjectId(userId), "$requests"]} }}
)

原因是$pull的条件中不能直接用"$requests"引用文档内的字段，$nin里混合了ObjectId类型值和字段引用，导致MongoDB无法正确转换类型，触发类型转换错误。

正确实现方式 #

可以利用MongoDB 4.2及以上版本支持的聚合管道式更新，通过$cond判断userId是否存在于requests数组，动态生成对应的更新操作：

User.updateOne(
  { _id: otherUserId },
  [
    {
      $set: {
        requests: {
          $cond: [
            { $in: [mongoose.Types.ObjectId(userId), "$requests"] },
            { $filter: {
                input: "$requests",
                cond: { $ne: ["$$this", mongoose.Types.ObjectId(userId)] }
              } },
            "$requests"
          ]
        },
        followers: {
          $cond: [
            { $in: [mongoose.Types.ObjectId(userId), "$requests"] },
            "$followers",
            { $filter: {
                input: "$followers",
                cond: { $ne: ["$$this", mongoose.Types.ObjectId(userId)] }
              } }
          ]
        }
      }
    }
  ]
)

逻辑说明 #

1. 用$in判断目标userId是否在requests数组中
2. 对于requests字段：如果存在则用$filter移除该userId，否则保持原数组
3. 对于followers字段：如果userId在requests中则保持原数组，否则用$filter移除该userId

若项目使用的MongoDB版本低于4.2，无法使用聚合管道更新，则只能拆分两次查询：先检查requests是否包含userId，再执行对应移除操作，但这不符合单个查询的需求。

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