如何在MongoDB中仅获取更新后的数组对象ID列表？

集合结构示例 #

{
  _id: xxx,
  shipment: {
    _id: xxx
  },
  messages: [
    { 
      _id: 123,
      origin: 'driver',
      isRead: false,
      ...
    },
    { 
      _id: 234,
      origin: 'driver',
      isRead: false,
      ...
    },
    { 
      _id: 345,
      origin: 'driver',
      isRead: true,
      ...
    },
    { 
      _id: 456,
      origin: 'dispatch',
      isRead: false,
      ...
    },
  ]
}

当前问题 #

使用findOneAndUpdate更新数组中符合条件的元素时，会返回完整文档，但我们只需要本次更新的元素ID列表（如[123,234]），希望避免多次查询或冗余对比操作。

最优解决方案 #

利用MongoDB单文档操作的原子性，通过返回更新前的旧文档提取本次要更新的元素ID，这些ID对应的元素必然已被成功更新：

// 执行更新并返回更新前的旧文档
const oldDoc = await MessageModel.findOneAndUpdate(
  { _id: messageId },
  { $set: { 'messages.$[elem].isRead': true } },
  {
    arrayFilters: [{ 'elem.isRead': false, 'elem.origin': 'driver' }],
    new: false // 设置为false，返回更新前的旧文档
  }
);

// 从旧文档中过滤出本次被更新的元素ID
const updatedIds = oldDoc.messages
  .filter(msg => msg.origin === 'driver' && !msg.isRead)
  .map(msg => msg._id);

// 最终得到目标ID列表：[123, 234]
console.log(updatedIds);

方案说明 #

1. 原子性保障：MongoDB会对目标文档加锁，直到更新完成，确保获取旧文档和执行更新的过程中，不会有其他操作修改该文档的messages数组。
2. 高效无冗余：仅一次数据库操作即可完成更新并获取所需ID，无需额外查询或复杂聚合。
3. 逻辑直观：旧文档中符合更新条件的元素，就是本次被修改的对象，直接过滤提取即可。

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