MongoDB聚合管道实现数组子文档Upsert：$push失效问题及最优方案

为什么聚合管道中else分支用$push不生效？ #

MongoDB里的$push分两种用途：

• 一种是聚合阶段运算符：只能用在$group阶段，用于分组时构建数组，不能在$set/$addFields这类修改字段的聚合阶段里使用。
• 另一种是更新运算符：作为顶级运算符用于更新命令中，直接往数组里追加元素，但它不能嵌套在聚合管道的阶段（比如$cond分支）中使用。

你在聚合管道的$cond else分支里用$push，本质是错误地把聚合阶段的$push当成了更新数组的手段，这不符合聚合管道的语法规则，所以无法生效。

最优实现方式 #

推荐两种方案，按需选择：

方案一：分两步原子操作（简洁高效） #

先尝试更新匹配指定ID的子文档，如果没有找到匹配项，再执行添加操作：

// 假设guestId是主文档ID，memberId是目标子文档ID，updatedMember是更新后的子文档数据，newMember是新子文档数据
const updatedGuest = await Guest.findOneAndUpdate(
  { _id: guestId, 'members._id': memberId },
  { $set: { 'members.$': updatedMember } },
  { new: true }
);

if (!updatedGuest) {
  await Guest.findOneAndUpdate(
    { _id: guestId },
    { $push: { members: newMember } },
    { new: true }
  );
}

这个方案逻辑简单，性能较好，适合大部分场景。

方案二：单聚合管道原子操作（一次完成） #

如果需要严格的原子性（避免两次操作之间的并发问题），可以用聚合管道生成新的数组并替换原字段：

await Guest.findOneAndUpdate(
  { _id: guestId },
  [
    {
      $set: {
        members: {
          $cond: {
            // 判断数组中是否存在目标子文档ID
            if: { $in: [memberId, '$members._id'] },
            // 存在则遍历数组，替换目标子文档
            then: {
              $map: {
                input: '$members',
                as: 'm',
                in: {
                  $cond: {
                    if: { $eq: ['$$m._id', memberId] },
                    then: updatedMember,
                    else: '$$m'
                  }
                }
              }
            },
            // 不存在则将新子文档追加到数组末尾
            else: { $concatArrays: ['$members', [newMember]] }
          }
        }
      }
    }
  ],
  { new: true }
);

这个方案通过聚合管道的数组表达式，在一次更新操作中完成“更新或插入”的逻辑，完全原子化，适合高并发场景。

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