mongoDB 4.0事务回滚的辛酸历程探究

前言

主管前几天发现mongoDB已经升级到4.0了，迫不及待得让我实现他期待已久的事务回滚，发现还是有很多坑啊！

下面是我将已有的本地mongoDB升级到支持事务回滚的历程，分享出来，有错误的地方欢迎指正！

以mac为例哈

部署mongodb事务回滚

1.准备工作

升级mongodb至4.0.0

$ brew upgrade mongodb

升级或安装mongodb.js v3.1.0 以上

$ npm i mongodb --save-dev

坑

升级完上述后，db.js里连接mongo时，

会警告，让你在connect的option里加一个字段

useNewUrlParser:true，

这时如果有用户验证，还需要在connect的option里再加一个字段

authSource:用户所在的db，一般为admin

否则会报验证失败，找不到用户的错误，比如

 const mongoClient = await MongoClient.connect(mongoClientUrl, {
  auth: {
  user: config.dbUserName,
  password: config.dbUserPassword,
  },
  authSource:'admin',
  useNewUrlParser:true,
 });

2.将已有的数据库改造成复制集

目前事务回滚只能在复制集上操作，单独的mongodb server是不能操作事务的

关掉所有的mongod

在平时启动mongod的命令后添加--replSet rs0，比如

$ mongod -dbpath ./db --port 27017 --replSet rs0

再开个shell，创一个不同端口的mongo实例，比如

$ mongod -dbpath ./db_repl --port 27018 --replSet rs0

连接27017的mongo实例，并设置

$ mongo
$ rs.initiate()
$ rs.add('localhost:27018');

完成

3.写回滚代码

在db.js里新增了一个方法

 export const getSession = async function() {
  return await state.mongoClient.startSession();
 };

每次在需要回滚的mongo代码前调用这个方法拿到session，开始回滚的标记

 const session = await db.getSession();
 session.startTransaction({
  readConcern: {level: 'snapshot'},
  writeConcern: {w: 'majority'},
 });

在每次调用mongodb.js的方法操作数据库时，都要带上session，比如

 db
 .collection(this.collecitonName)
 .insertOne(doc,{session});

在你处理错误并觉得需要回滚了，执行

await session.abortTransaction();

在你觉得没问题，一起正常结束时，执行

await session.commitTransaction();

我的代码里可能有一些封装的代码没有放上去，导致不一定能理解。我只是举个栗子去实现，具体的代码实现可以看参考链接1

4.总结

useNewUrlParser这个属性会在url里识别验证用户所需的db，未升级前是不需要指定的，升级到一定要指定，不管是在url后面，还是用authSource

事务回滚只能在复制集上操作，我猜测实现的原理可能是这样：先记录主节点的session，然后回滚的话，通过这个session查找副节点的数据快照，然后将这快照再应用到主节点上，实现回滚。当然，实际情况应该相当复杂，不然mongoDB也不会用3年时间来实现这个操作。

5.参考链接

• mongoDB官方关于事务的介绍
• mongoDB中文社区关于replSet的介绍
• 如果不想进行第二步，run-rs可以直接为你创建一个新的mongoDB复制集

总结