MongoDB通过聚合/更新管道在数组指定索引插入元素

场景说明 #

在MongoDB聚合管道、支持管道的更新操作（4.2及以上版本）中，可通过原生数组操作符实现指定索引位置的元素插入，无需将文档读取到应用层处理，保证操作原子性与执行效率。
以下为示例场景：

初始示例文档 #

[
  {
    _id: ObjectId("62c2e94e65f32725f8f62b79"),
    updatedAt: ISODate("2022-06-29T13:10:36.659Z"),
    createdAt: ISODate("2022-06-29T08:06:51.264Z"),
    userID: 1,
    myImage: "imageC",
    images: [
      "imageA",
      "imageB",
      "imageD",
      "imageE",
      "imageF"
    ]
  }
]

目标效果 #

将myImage字段的值插入到images数组的索引2位置，插入后文档结构如下：

[
  {
    _id: ObjectId("62c2e94e65f32725f8f62b79"),
    updatedAt: ISODate("2022-06-29T13:10:36.659Z"),
    createdAt: ISODate("2022-06-29T08:06:51.264Z"),
    userID: 1,
    myImage: "imageC",
    images: [
      "imageA",
      "imageB",
      "imageC",
      "imageD",
      "imageE",
      "imageF"
    ]
  }
]

实现逻辑 #

核心思路是将原数组按插入位置拆分为前后两段，和待插入元素拼接为新数组，用到的操作符均为MongoDB管道通用操作符，4.2及以上版本支持更新管道，聚合管道全版本支持该逻辑：

• 用$slice截取原数组从起始位置到插入索引前的所有元素，作为新数组前半段
• 将待插入元素包装为数组，作为新数组中间段
• 用$slice截取原数组从插入索引到末尾的所有元素，作为新数组后半段
• 用$concatArrays将三段数组按顺序拼接，得到插入完成的目标数组

可直接复用的代码 #

更新管道场景（原子更新文档） #

db.collection.updateOne(
  // 替换为实际的文档匹配条件
  { _id: ObjectId("62c2e94e65f32725f8f62b79") },
  [
    {
      $set: {
        images: {
          $concatArrays: [
            // 插入位置前的元素，第二个参数为插入索引值，示例为2
            { $slice: ["$images", 2] },
            // 待插入元素，可替换为常量、其他字段引用、多元素数组
            ["$myImage"],
            // 插入位置后的元素，第二个参数与前面的插入索引保持一致即可
            { $slice: ["$images", 2, { $size: "$images" }] }
          ]
        }
      }
    }
  ]
)

聚合管道场景（加工查询结果） #

在聚合的$project/$set阶段使用完全相同的数组拼接逻辑即可，不需要修改语法。

通用扩展：如果需要一次性插入多个元素，只需要把中间待插入的单元素数组替换为包含多个元素的数组即可；如果插入位置是动态值（比如来自文档其他字段），只需要把代码里写死的索引值2替换为对应的字段引用即可，逻辑无需调整。

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