如何在TypeScript中实现Mongoose批量删除并返回文档ID？

你当前的实现思路是可行的——因为MongoDB原生的deleteMany仅返回删除的文档数量，不会返回被删文档的具体ID，所以必须先查询出符合条件的文档ID，再执行删除操作。不过你的代码存在几个可优化的点：

• 未给deleteMany添加await，会导致函数提前返回，无法确保删除操作完成
• 两次数据库操作之间存在竞态条件：如果在查询和删除的间隙，有其他操作修改/删除了符合条件的文档，会导致实际删除的ID和你返回的不一致
• 存在额外的数据库查询开销

下面给出几种不同场景下的优化实现：

场景一：需要严格一致性（避免竞态） #

如果你的MongoDB部署支持事务（副本集或分片集群），可以用事务包裹查询和删除操作，确保两个步骤原子执行：

async function deleteAndReturnIds(searchQuery: Partial<Model>) {
  const session = await mongoose.startSession();
  session.startTransaction();
  try {
    // 事务内查询目标文档ID
    const docs = await Model.find(searchQuery).session(session).lean();
    const ids = docs.map(({ _id }) => _id);
    
    // 同一事务内执行删除
    await Model.deleteMany({ _id: { $in: ids } }).session(session);
    
    await session.commitTransaction();
    return ids;
  } catch (err) {
    await session.abortTransaction();
    throw err;
  } finally {
    session.endSession();
  }
}

场景二：无并发修改风险（简化实现） #

如果你的业务场景不存在并发修改目标文档的情况，可以简化代码，同时补上await确保删除完成：

async function deleteAndReturnIds(searchQuery: Partial<Model>) {
  const docs = await Model.find(searchQuery).lean();
  const ids = docs.map(({ _id }) => _id);
  // 等待删除操作执行完毕
  await Model.deleteMany({ _id: { $in: ids } });
  return ids;
}

场景三：需要返回完整被删文档（小批量） #

如果除了ID还需要完整的被删文档，可以用findOneAndDelete循环处理，适合小批量删除场景：

async function deleteAndReturnIds(searchQuery: Partial<Model>) {
  const deletedDocs = [];
  let doc;
  while ((doc = await Model.findOneAndDelete(searchQuery))) {
    deletedDocs.push(doc);
  }
  return deletedDocs.map(({ _id }) => _id);
}

补充说明：MongoDB的deleteMany确实没有内置选项返回被删文档的ID或内容，这是因为批量删除操作设计上更注重性能，不会额外返回被删数据。所以先查询再删除是目前实现需求的唯一可行路径，你可以根据业务场景选择对应的方案。

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