MongoDB/Mongoose更新数组内元素并返回更新后的元素

要实现单查询返回更新后的notification元素，你可以利用MongoDB 4.2及以上版本支持的聚合管道更新，配合投影精准提取目标元素，无需返回整个文档。

具体实现 #

假设你的User和Notification Schema结构如下：

const userSchema = new mongoose.Schema({
  _id: mongoose.Schema.Types.ObjectId,
  notifications: [{
    _id: mongoose.Schema.Types.ObjectId,
    status: String,
    // 其他自定义字段
  }]
});
const User = mongoose.model('User', userSchema);

执行更新并直接获取目标notification的代码示例：

const updatedResult = await User.findOneAndUpdate(
  // 匹配用户ID和目标通知ID
  {
    _id: 目标用户ID,
    'notifications._id': 目标通知ID
  },
  // 用聚合管道定位并修改指定通知的status
  [
    {
      $set: {
        'notifications': {
          $map: {
            input: '$notifications',
            as: 'n',
            in: {
              $cond: [
                { $eq: ['$$n._id', 目标通知ID] },
                { ...'$$n', status: 'read' }, // 替换为你需要的状态：'read'或'confirmed'
                '$$n'
              ]
            }
          }
        }
      }
    }
  ],
  // 配置返回规则：只保留更新后的目标通知
  {
    new: true,
    projection: {
      notifications: { $elemMatch: { _id: 目标通知ID } }
    }
  }
);

// 从返回的数组中提取单个通知元素
const updatedNotification = updatedResult?.notifications[0];

关键细节 #

• 用$map遍历数组精准修改目标元素，比传统的$set: {'notifications.$.status': ...}更适配后续的投影逻辑。
• projection里的$elemMatch会过滤数组，只保留匹配目标ID的通知，最后取数组第一个元素就是更新后的结果。
• 原生MongoDB驱动的写法逻辑一致，只需调整语法适配驱动即可。

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