MongoDB跨不同集群多库原子事务实现方案咨询

当前实现的问题 #

你现在的代码是分别在两个集群的数据库里开启独立事务，这种方式无法保证真正的原子性。核心问题在于：

• 当第一个事务（db1Session.commitTransaction()）提交成功后，如果第二个事务提交失败，第一个已经提交的操作没办法回滚，会出现两个库数据不一致的情况。
• 反过来，如果第二个事务提交成功但第一个提交失败，同样会导致数据不一致。
  因为两个事务完全独立，没有协调机制来确保它们要么同时成功，要么同时失败。

MongoDB官方推荐的处理方式 #

MongoDB本身的分布式事务能力仅支持同一个分片集群内的不同数据库/集合，不支持完全独立的两个集群之间的原子性事务。针对跨集群的原子性需求，通常有两种可行方案：

1. 补偿事务（最终一致性方案） #

如果必须用两个独立集群，只能通过最终一致性来实现业务层面的“原子性”，常见的是TCC（Try-Confirm-Cancel）模式：

• Try阶段：对两个数据库执行预操作，比如锁定资源、预留数据状态（标记为待确认）。
• Confirm阶段：如果两个预操作都成功，正式提交修改（把预留状态改为最终状态）。
• Cancel阶段：如果任意一个预操作失败，对已执行预操作的数据库执行回滚（取消预留状态，恢复原始数据）。

示例简化代码：

async function crossClusterUpdate() {
  let db1RollbackNeeded = false;
  let db2RollbackNeeded = false;

  try {
    // Try阶段：预操作db1，标记为待确认
    await db1.collection('xxx').updateOne(
      { _id: xxx },
      { $set: { status: 'pending' } },
      { session: db1Session }
    );
    db1RollbackNeeded = true;

    // Try阶段：预操作db2，标记为待确认
    await db2.collection('yyy').updateOne(
      { _id: yyy },
      { $set: { status: 'pending' } },
      { session: db2Session }
    );
    db2RollbackNeeded = true;

    // Confirm阶段：正式提交db1
    await db1.collection('xxx').updateOne(
      { _id: xxx },
      { $set: { status: 'completed', ...finalData } },
      { session: db1Session }
    );
    db1RollbackNeeded = false;

    // Confirm阶段：正式提交db2
    await db2.collection('yyy').updateOne(
      { _id: yyy },
      { $set: { status: 'completed', ...finalData } },
      { session: db2Session }
    );
    db2RollbackNeeded = false;

  } catch (error) {
    // Cancel阶段：回滚已完成的预操作
    if (db1RollbackNeeded) {
      await db1.collection('xxx').updateOne(
        { _id: xxx },
        { $set: { status: 'original' } },
        { session: db1Session }
      );
    }
    if (db2RollbackNeeded) {
      await db2.collection('yyy').updateOne(
        { _id: yyy },
        { $set: { status: 'original' } },
        { session: db2Session }
      );
    }
    throw error;
  }
}

2. 合并集群（如果业务允许） #

如果业务架构可以调整，把两个数据库迁移到同一个MongoDB分片集群中，就可以直接使用MongoDB的分布式事务：

• 只需开启一个跨库的会话事务，MongoDB会自动协调分片间的事务，确保原子性。

示例代码：

// 注意：db1和db2必须属于同一个分片集群
const session = await db1.startSession(); // 同一个会话即可操作同集群内的所有库
try {
  session.startTransaction();
  
  // 操作db1
  await db1.collection('xxx').updateOne({ _id: xxx }, { $set: data }, { session });
  // 操作db2（同集群内的另一个库）
  await db2.collection('yyy').updateOne({ _id: yyy }, { $set: data }, { session });

  await session.commitTransaction();
} catch (error) {
  await session.abortTransaction();
  throw error;
}

总结 #

• 你的当前实现不可靠，无法保证跨集群的原子性。
• 跨独立集群只能用最终一致性方案（如TCC）来模拟原子性。
• 同分片集群内可以直接用MongoDB原生分布式事务，这是最优解。

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