如何用单条MongoDB查询更新嵌套数组元素并插入缺失项？

需求回顾 #

现有文档结构：

{
  _id: 123,
  trips: [
    { uuid: 1, locations: [{id:1}, {id:2}, {id:3}] },
    { uuid: 2, locations: [{id:4}, {id:5}] }
  ]
}

待处理数据：

const newTrips = [
  { uuid: 2, locations: [{id:4}, {id:5}, {id: 6}]},
  { uuid: 3, locations: [{id:7}]},
]

目标：更新uuid=2的locations数组，同时新增uuid=3的元素到trips数组，最终得到：

{
  _id: 123,
  trips: [
    { uuid: 1, locations: [{id:1}, {id:2}, {id:3}] },
    { uuid: 2, locations: [{id:4}, {id:5}, {id:6}] },
    { uuid: 3, locations: [{id:7}] }
  ]
}

问题分析 #

之前的循环updateOne方案需要多次请求，且添加upsert: true报错是因为位置操作符$无法匹配到uuid=3的元素，无法生成有效的更新路径。

单查询实现方案 #

利用MongoDB 4.2+支持的聚合管道作为更新操作，通过$map、$mergeObjects和$setUnion组合操作，实现一次更新完成两个需求：

collection.updateOne(
  { _id: 123 },
  [
    {
      $set: {
        trips: {
          $setUnion: [
            // 处理已有trips：匹配到newTrips里的项就更新locations，否则保留原项
            {
              $map: {
                input: "$trips",
                as: "trip",
                in: {
                  $mergeObjects: [
                    "$$trip",
                    // 从newTrips中找到同uuid的项，取其locations覆盖原数据
                    { $arrayElemAt: [
                      { $filter: {
                          input: newTrips,
                          cond: { $eq: ["$$this.uuid", "$$trip.uuid"] }
                      }},
                      0
                    ]}
                  ]
                }
              }
            },
            // 提取newTrips中在原trips里不存在的uuid项，作为新增元素
            { $filter: {
                input: newTrips,
                cond: {
                  $not: {
                    $in: ["$$this.uuid", "$trips.uuid"]
                  }
                }
            }}
          ]
        }
      }
    }
  ]
)

原理说明 #

1. 更新已有元素：通过$map遍历原trips数组，用$filter从newTrips中匹配相同uuid的项，再用$mergeObjects将原元素和新数据合并，实现locations的覆盖更新。
2. 新增不存在的元素：用$filter筛选出newTrips中不在原trips.uuid列表里的项，作为待新增的元素。
3. 合并结果：通过$setUnion将处理后的原数组和新增元素数组合并，自动完成去重（因uuid唯一，不会出现重复元素）。

注意事项 #

• 该方案要求MongoDB版本≥4.2，因为只有4.2及以上版本才支持在更新操作中使用聚合管道。
• 需确保newTrips内部的uuid唯一，否则$arrayElemAt只会取第一个匹配项。

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