Mongo4.2分布式事务实现Overview

整体流程

Mongos侧

相关代码:transaction_router.cpp/h

在Mongos侧,每个OperationCtx会有一个TransactionRouter,记录事务上下文。

TransactionRouter::get(opCtx)

每次事务内CRUD操作,会被TransactionRouter记录,特别是参与此次事务的shard,会被记录在_participants中。

TransactionRouter::attachTxnFieldsIfNeeded(opCtx, shardId, cmObj) {_participants.emplace_back(shardId);}

当事务提交时,TransactionRouter在所有参与者中选择(第一个)一个coordinator,并将两阶段提交的工作 hand off 给 coordinator。

TransactionRouter::_handOffCommitToCoordinatoropCtx) { CoordinateCommitTransaction cmd;  cmd.setParticipants(_participantList);  sendTo(_coordinator, cmd).waitForReply();}

因此,Mongos不处理二阶段提交,甚至configSvr也不参与。二阶段提交的工作完全由多个shard阶段之间协作完成。基于TimestampOracle的分布式事务需要中心节点分配事务Id或者时间戳,但是基于HLC的系统就不需要。

Shard侧

相关代码:transaction_coordinator.cpp/h,transaction_coordinator_util.cpp/h

shard侧在收到CoordinateCommitTransactionCmd后,会有如下操作:

  • 将{lsid,txnNumber, participants} 持久化到config.transaction_coordinators表中
  • 广播prepare命令{lsid, txnNumber, participants} - 收集prepareresponse,如果deadline之前没有全部返回,则prepare失败 - 只要有一个参与者反对提交,就prepare失败 - commitTimestamp = max(prepareTimestamp)
  • 将prepare结果COMMITorABORT更新到step1写入的记录中 - 一旦这一步决定commit,二阶段提交就必须成功
  • 假设prepare结果是commit,广播通知participants进行commit - 没有超时,如果某个participant一直没返回(挂了/网络故障),就一直等直到有返回结果
  • 删除config.transaction_coordinators表中的对应记录

Oplog格式更改

为了支持超过16MB的事务,一个事务内的多条操作会记录在多条Oplog里。每条Oplog新增字段如下:

  • prepare: {true/false} // 表示这条Oplog是否是分布式事务的prepare阶段产生的,与非事务产生的oplog区分
  • prevOpTime // 用以指向本事务内上一条Oplog,用于做prepare阶段的事务的恢复
  • lsid/txnNumber //多文档事务特有
  • o.commitTimestamp/o.commitTransaction

举个例子,一个shard内,有两个分布式事务,产生了五条Oplog db.oplog.rs.find().sort({$natural:-1})

{ts:5,o:{commitTimestamp:4,commitTransaction:1}, prevOpTime:4, lsid:1, txnNumber:1}{ts:4, applyOps:[...], prevOpTime:1,prepare:true, lsid:1, txnNumber:1}{ts:3,o:{commitTimestamp:2,commitTransaction:1}, prevOpTime:2,lsid:0, txnNumber:2}{ts:2, applyOps:[...], prevOpTime:-1,prepare:true,lsid:0, txnNumber:2}{ts:1, applyOps:[...], prevOpTime:-1,prepare:true, lsid:1, txnNumber:1}

总结:每个事务的Oplog不再放在一个OplogEntry里,甚至不再连续存放,而是通过prevOpTime串起来。

二阶段提交的故障恢复

分为Coordinator的故障和Participant的故障

Coordinator的故障

二阶段提交的信息持久化在transaction_coordinators表中。TransactionCoordinatorService::onStepUp从表里恢复 所有pending状态的事务,继续执行

auto coordinatorDocs = txn::readAllCoordinatorDocs(opCtx) for (const auto& doc : coordinatorDocs) {auto coordinator = std::make_shared<TransactionCoordinator>(); coordinator.continueCommit(doc);}

Participant的故障

事务在prepare时,会将修改记录记录到oplog表中,上文我们已经分析过prepareLog的格式,因此所有prepare成功的事务的Oplog都已经持久化,就差一条CommitLog了。在ReplicationCoordinatorImpl::startup中,会重新构建 所有处于prepare状态的事务。每个事务,顺着prevOpTime就可以找到所有的Oplog,通过Oplog恢复出事务的状态。

for (const auto& doc : config.transactions.find()) { lastOplogTime = doc.lastOplogTime vector<OplogEntry> v;while(auto t = oplog.rs.find(lastOplogTime)) { v.push_back(t);lastOplogTime = v.prevOpTime}reconstructTransaction(v.reverse());}

上文说过,coordinator会等待所有participant返回commit结果,所以participant重启后,coordinator会重试让participant提交处于prepare状态的事务。

可以看到,与Mysql的本地XA不同的是,mongo分布式事务的prepare阶段是通过Oplog保证持久化的,而不是wal 和oplog的协同,wiredTiger层在prepare阶段不保证持久化,那wiredTiger的prepare在干嘛呢?

wiredTiger层

建议先读这篇这篇

我们知道,对于wiredTiger,未提交事务是不会被evict的,也不会进入checkpoint,在ARIES算法中,这叫做nosteal模式。可是,在极端情况下,prepare状态可能会驻留很长时间,prepare状态的事务会不会被evict出去呢?为 了防止缓存压力,wiredTiger允许prepare状态的事务被evict出去(注意:不是evict到已提交事务的btree表空间)。

另外,(可能是)为了防止快照数量膨胀,wiredTiger在prepare阶段就释放快照,我们知道wiredTiger的SI是通过 事务开始时拷贝全局快照来实现,提前释放快照会导致未提交的事务过早的被看到,wiredTiger于是又加了一个WT_PREPARE_CONFLICT状态,考虑如下的时序:

txn1.begin() txn1.insert("a", "b")txn1.prepare(prepare_ts = 10)txn2.begin(read_ts = 11) txn2.read("a")  // WT_PREPARE_CONFLICT txn3.begin(read_ts = 9)txn3.read("a")  // WT_NOT_FOUND

由于txn1在prepare阶段就已经释放了快照。因此txn2开始时,txn1的修改对其是可见的,虽然可见,但是出于一种 特殊的 PREPARE_CONFLICT状态,Mongo层在处理PREPARE_CONFLICT状态的数据时,会不停重试,mongo将所有对wiredTiger的读操作通过wiredTigerPrepareConflictRetry封装起来。

int ret = wiredTigerPrepareConflictRetry(opCtx, [&] { return c->search(c); });

然而,直接略过这条记录不行吗?为什么Mongo一定要不停retry呢?因为这样违背SI。快照已经在事务prepare时释 放了,而事务的commitTimestamp > prepareTimestamp,因此对于一个SI的读事务,直接忽略另一个prepare状态的记录,会导致另一个事务提交后,对本事务可见,或者不可见。举个例子如下,txn2是快照隔离级别的,两次读a的结果必须一致。因此第一次读a需要重试。

// 如果txn2 直接不读其他出于prepare状态的记录。txn1.begin() txn1.insert("a", "b")txn1.prepare(prepare_ts = 10)txn2.begin(read_ts = 11) txn2.read("a") txn1.commit(commit_ts = 11) txn2.read("a")

那么,能不能不要prepare阶段释放快照,而是commit后释放快照呢?如果这样,wiredTiger其实根本不需要提供出prepare接口出来。

孔德雨

原文发布于微信公众号 - Mongoing中文社区(mongoing-mongoing)

原文发表时间:2019-06-20

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券