MongoDB如何基于指定字段确保内嵌数组元素组合唯一

问题场景 #

现有MongoDB集合文档包含events数组字段，数组内每个元素包含eventName、actorId、detail三个属性，示例文档结构如下：

{
  _id: ObjectId("5a934e000102030405000000"),
  date: ISODate("2022-06-03T00:00:00.000Z"),
  events: [
    {
      eventName: "Add",
      actorId: "1",
      detail: "new actor"
    },
    {
      eventName: "Vote",
      actorId: "2",
      detail: "up"
    },
    {
      eventName: "Vote",
      actorId: "3",
      detail: "down"
    },
    {
      eventName: "Vote",
      actorId: "4",
      detail: "cork"
    }
  ]
}

需求为：根据_id定位到指定文档后，向events数组批量新增元素，仅当待新增元素的eventName与actorId组合在现有数组元素中不存在时才插入，保证该字段组合在单文档的events数组内唯一。
以如下待插入数组为例：

[
  {
    eventName: "Add",
    actorId: "3",
    detail: "action"
  },
  {
    eventName: "Vote",
    actorId: "2",
    detail: "up"
  },
  {
    eventName: "Vote",
    actorId: "5",
    detail: "up"
  }
]

最终仅第1、3个元素会被成功插入，{eventName: "Vote", actorId: "2"}的组合已存在于原数组中，会被自动过滤。

推荐实现方案（MongoDB 4.2+ 支持） #

使用聚合管道更新实现原子化的过滤+插入操作，无需应用层提前查询文档，无并发重复风险，是该场景的最优解。

注意：原生$addToSet操作符仅在数组元素完全一致时才会去重，无法满足仅按eventName、actorId两个字段判断重复的需求，不可直接用于该场景。

实现代码如下：

// 替换为实际目标文档ID
const targetDocId = ObjectId("5a934e000102030405000000");
// 替换为实际待插入的事件数组
const pendingInsertEvents = [
  { eventName: "Add", actorId: "3", detail: "action" },
  { eventName: "Vote", actorId: "2", detail: "up" },
  { eventName: "Vote", actorId: "5", detail: "up" }
];

db.yourCollectionName.updateOne(
  // 匹配目标文档
  { _id: targetDocId },
  [
    // 步骤1：提取现有events的eventName+actorId组合，生成唯一键列表临时字段
    {
      $set: {
        _existingEventKeys: {
          $reduce: {
            input: "$events",
            initialValue: [],
            in: {
              $concatArrays: [
                "$$value",
                // 用特殊分隔符拼接两个字段作为唯一键，分隔符需选择不会出现在两个字段值内的字符
                [{ $concat: ["$$this.eventName", "|||", "$$this.actorId"] }]
              ]
            }
          }
        }
      }
    },
    // 步骤2：过滤待插入数组，仅保留唯一键不存在于现有列表的元素，拼接到原events数组
    {
      $set: {
        events: {
          $concatArrays: [
            "$events",
            {
              $filter: {
                input: pendingInsertEvents,
                cond: {
                  $not: [
                    {
                      $in: [
                        { $concat: ["$$this.eventName", "|||", "$$this.actorId"] },
                        "$_existingEventKeys"
                      ]
                    }
                  ]
                }
              }
            }
          ]
        }
      }
    },
    // 步骤3：清理临时字段
    { $unset: "_existingEventKeys" }
  ]
);

方案优势 #

• 全流程原子执行，不存在并发写入导致重复插入的问题
• 过滤逻辑完全在数据库侧执行，减少应用层与数据库的交互次数
• 支持任意数量的批量插入，自动完成重复项过滤

低版本兼容方案（MongoDB <4.2） #

如果使用的MongoDB版本不支持聚合管道更新，可采用两步操作：

1. 先根据_id查询目标文档，提取现有events数组中所有eventName+actorId组合
2. 在应用层过滤待插入数组，剔除重复项后，使用$push + $each将剩余元素插入数组

该方案存在并发风险，若业务并发较高，可在更新时追加数组元素不存在的查询条件，避免并发写入重复数据。

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