MongoDB更新数组时，能否获取被移除元素的索引及类型？

一、更新操作中能否获取数组特定元素的索引？ #

MongoDB的常规更新操作（如updateOne、updateMany）本身不会直接返回数组中被操作元素的索引。如果需要获取索引，可通过以下两种方式实现：

• 先查询计算索引：通过聚合查询定位元素，使用$indexOfArray运算符计算目标元素的索引。
  示例代码：

  const post = await Post.aggregate([
    { $match: { _id: postId } },
    {
      $project: {
        targetIndex: {
          $indexOfArray: ["$reacts.publisher", req.query.publisher]
        }
      }
    }
  ]).next();
  // targetIndex即为匹配元素的索引，未找到则返回-1
  

• 聚合管道更新时计算：在MongoDB 4.2+版本的聚合管道更新中，也可通过$indexOfArray在更新阶段计算索引，但更新操作本身不会返回该索引，需结合返回更新前后文档的方式获取。

二、如何获取$pull操作中移除元素的ReactType？ #

你当前的updateOne语句无法直接获取被$pull移除元素的ReactType，因为updateOne仅返回匹配数、修改数等元数据，不包含被移除的具体数据。推荐以下两种替代方案：

方案1：原子查询+分步更新（适用于低并发场景） #

使用findOneAndUpdate原子获取更新前的文档，从中提取被移除元素的ReactType，再完成postPoints的更新：

// 原子获取更新前的文档并执行$pull操作
const oldPost = await Post.findOneAndUpdate(
  { _id: postId, "reacts.publisher": req.query.publisher },
  { $pull: { reacts: { publisher: req.query.publisher } } },
  { returnDocument: 'before' }
);

if (oldPost) {
  // 从旧文档中提取被移除的ReactType
  const removedReact = oldPost.reacts.find(r => r.publisher === req.query.publisher);
  const reactType = removedReact.ReactType;
  // 更新postPoints
  await Post.updateOne(
    { _id: postId },
    { $inc: { postPoints: -reactsEnum[reactType] } }
  );
}

注：此方案分两步执行，高并发场景下存在数据不一致风险，更推荐方案2。

方案2：聚合管道式更新（MongoDB 4.2+，原子操作） #

利用MongoDB 4.2支持的聚合管道更新，在一个原子操作中完成$pull逻辑和postPoints的计算，同时可通过返回文档验证结果：

// 原子完成移除元素和分值更新
const result = await Post.updateOne(
  { _id: postId },
  [
    {
      $set: {
        // 过滤保留非目标publisher的reacts元素（等效$pull）
        reacts: {
          $filter: {
            input: "$reacts",
            cond: { $ne: ["$$this.publisher", req.query.publisher] }
          }
        },
        // 计算更新后的postPoints：减去被移除元素对应的分值
        postPoints: {
          $subtract: [
            "$postPoints",
            {
              $let: {
                vars: {
                  removedReact: {
                    $arrayElemAt: [
                      {
                        $filter: {
                          input: "$reacts",
                          cond: { $eq: ["$$this.publisher", req.query.publisher] }
                        }
                      },
                      0
                    ]
                  }
                },
                in: { $ifNull: [reactsEnum[$$removedReact.ReactType], 0] }
              }
            }
          ]
        }
      }
    }
  ],
  { returnDocument: 'after' } // 返回更新后的文档，可验证结果
);

// 若需获取被移除的ReactType，可将returnDocument改为'before'，从旧文档中提取
const oldPost = await Post.findOneAndUpdate(/* 同上聚合管道 */, { returnDocument: 'before' });
const removedReactType = oldPost?.reacts.find(r => r.publisher === req.query.publisher)?.ReactType;

这种方式是完全原子的，解决了并发修改问题，同时能准确计算postPoints的增量。

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