MongoDB新手问询：无需两阶段提交维护帖子评论数的方法

嘿，作为MongoDB新手碰到这种跨集合的一致性问题太正常了！我给你几个实用的方案，都不用手动搞复杂的两阶段提交，一起来看看：

1. 用MongoDB内置事务（推荐，4.0+支持） #

MongoDB从4.0版本开始支持多文档事务，它已经帮你封装好了两阶段提交的逻辑，你不用手动去实现。只要你的部署是副本集或者分片集群，就可以用这个方案，保证两个集合的操作要么都成功，要么都失败。

举个Node.js驱动的例子：

const session = await client.startSession();
session.startTransaction();

try {
  // 1. 插入/删除评论
  await db.collection('Comment').insertOne(
    { postId: postId, content: '新评论' },
    { session }
  );
  // 2. 同步更新Post的评论数
  await db.collection('Post').updateOne(
    { _id: postId },
    { $inc: { commentsNumber: 1 } },
    { session }
  );

  await session.commitTransaction();
} catch (error) {
  await session.abortTransaction();
  console.error('操作失败，已回滚:', error);
} finally {
  session.endSession();
}

这个方案能保证强一致性，适合对数据一致性要求高的场景。

2. 调整数据模型：把评论嵌入Post文档 #

如果你的评论数量不是特别多（不会让Post文档超过MongoDB的16MB限制），那嵌入式模型是MongoDB推荐的设计思路。把评论直接存在Post文档里，这样更新评论和评论数是同一个文档的原子操作，根本不会有一致性问题。

比如Post的结构可以是这样：

{
  "_id": ObjectId("xxx"),
  "title": "我的第一篇帖子",
  "content": "帖子内容...",
  "comments": [
    { "_id": ObjectId("yyy"), "content": "沙发！", "createdAt": ISODate("2024-05-20T10:00:00Z") },
    { "_id": ObjectId("zzz"), "content": "不错的帖子", "createdAt": ISODate("2024-05-20T10:05:00Z") }
  ],
  "commentsNumber": 2
}

添加评论的时候，用一次原子操作搞定：

await db.collection('Post').updateOne(
  { _id: postId },
  {
    $push: { comments: { _id: new ObjectId(), content: '新评论', createdAt: new Date() } },
    $inc: { commentsNumber: 1 }
  }
);

删除评论同理，用$pull和$inc组合操作，完全原子，省心又高效。

3. 用变更流（Change Streams）实现最终一致性 #

如果必须分开两个集合，又不想用事务，那可以用变更流监听Comment集合的变化，异步同步Post的评论数。这个方案是最终一致性，适合对实时性要求不是特别高的场景。

思路是：监听Comment集合的插入、删除事件，当有评论被添加或删除时，自动更新对应的Post的commentsNumber。还要注意处理幂等性，避免重复处理同一个事件。

举个例子：

const commentStream = db.collection('Comment').watch([
  { $match: { operationType: { $in: ['insert', 'delete'] } } }
]);

commentStream.on('change', async (change) => {
  const postId = change.fullDocument?.postId || change.documentKey.postId; // 根据操作类型取postId
  const increment = change.operationType === 'insert' ? 1 : -1;

  try {
    await db.collection('Post').updateOne(
      { _id: postId },
      { $inc: { commentsNumber: increment } }
    );
  } catch (error) {
    console.error('同步评论数失败:', error);
    // 可以在这里添加重试逻辑，或者记录失败事件后续处理
  }
});

这个方案不用在业务代码里写两次操作，而是让数据库自己监听变化来同步，缺点是可能有短暂的不一致，但大部分场景下完全够用。

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