Mongo单操作实现字段覆盖更新、数组去重合并及Upsert功能

可以通过单个操作实现需求，使用MongoDB 4.2+支持的聚合管道更新结合updateOne()方法，配合upsert参数即可完成。

原集合文档 #

{
  "key": "abc",
  "field1": "oldValue",
  "fieldN": ["a","b"]
}

待更新的新文档内容 #

{
  "key": "abc",
  "field1": "newValue",
  "fieldN": ["b","c"]
}

具体操作命令（Mongo Shell） #

db.yourCollectionName.updateOne(
  { key: "abc" }, // 匹配条件：key等于abc
  [
    {
      $set: {
        // 直接覆盖需要更新的普通字段
        field1: "newValue",
        // 其他需要覆盖的字段可在此添加，比如 field2: "newVal"
        // 对fieldN执行去重合并：用$setUnion合并原有数组与新数组
        fieldN: { $setUnion: ["$fieldN", ["b", "c"]] }
      }
    }
  ],
  { upsert: true } // 开启upsert：无匹配文档时自动插入
)

效果说明 #

1. 当匹配文档存在时：
   • field1会被覆盖为newValue
   • fieldN会合并去重为["a", "b", "c"]（$setUnion自动处理重复值，顺序由MongoDB内部排序规则决定）
2. 当匹配文档不存在时：
   • 自动插入新文档，其中field1为newValue，fieldN为["b", "c"]（因为原数组不存在时$fieldN为null，$setUnion会直接返回新数组）

灵活扩展写法 #

如果需要更新的字段较多，可通过变量简化代码，避免手动逐个编写字段：

// 定义新文档数据
const newData = {
  key: "abc",
  field1: "newValue",
  fieldN: ["b", "c"],
  // 其他需要覆盖的字段
  field2: "newFieldValue"
};

// 提取除key和fieldN外的普通更新字段
const updateFields = { ...newData };
delete updateFields.key;
delete updateFields.fieldN;

// 执行更新
db.yourCollectionName.updateOne(
  { key: newData.key },
  [
    {
      $set: {
        ...updateFields,
        fieldN: { $setUnion: ["$fieldN", newData.fieldN] }
      }
    }
  ],
  { upsert: true }
)

注意事项 #

• 需确保MongoDB版本为4.2或以上，该版本开始支持聚合管道作为更新操作的参数
• $setUnion处理的数组元素需为同一类型，否则可能出现不符合预期的结果

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