简介
mongodb 的事务是依靠 mongodb 连接的客户端 session 实现,事务执行的流程大致是 建立 session,通过 session startTransaction 启动事务,如果一系列事务都完成,那么 commitTransaction 完成事务操作,并结束当前事务 session;如果一系列事务中有任意事件失败, 那么 abortTransaction 中止事务,内部将已完成的任务回退到修改之前,并结束当前事务 session。
session = client.startSession();
session.startTransaction();
session.commitTransaction();
session.abortTransaction();
session.endSession();
复制代码
场景描述
当前有两个用户,A 用户有余额 50 软妹币,B 用户有余额 10 软妹币,A 用户给 B 转账 10 ,场景设定 转账很安全,网络也很畅通,没有黑客拦截,没有发生意外,这次转账成功了,这时 A 用户余额剩下40 ,B 用户余额有20。A 感觉很安全,这时又给 B 转账,A 忘记自己余额有多少,给 B 转了 50 ,结果出错了。
在没有事务的情况下,操作数据库是这样的,1.A 账户的余额 -50,2. B 账户增加50. 当 A 余额不足时或在操作 A 账户成功后网络发生错误,B 账户的金额没能正确修改。
在有事务的情况下,即使在操作 A 账户金额后出现错误,则事务会将整个转账过程回退到修改之前。
下载 mongodb4 并解压
wget https://fastdl.mongodb.org/osx/mongodb-osx-ssl-x86_64-4.0.0.tgz
tar -xzvf mongodb-osx-ssl-x86_64-4.0.0.tgz
cd mongodb-osx-ssl-x86_64-4.0.0
复制代码
!!! Transaction 只适用复制集 Replica Set, 所以要先搭建mongodb 复制集
启动多个 mongodb 实例
// 创建data目录
mkdir -p data/1301
mkdir -p data/1302
mkdir -p data/1303
// 起三个 mongodb 实例
./bin/mongod --replSet shard1 --dbpath=./data/1301 --port=1301
./bin/mongod --replSet shard1 --dbpath=./data/1302 --port=1302
./bin/mongod --replSet shard1 --dbpath=./data/1303 --port=1303
复制代码
// 配置复制集
./bin/mongo --port 1301
rsconf = {_id: "shard1", members: [ { _id: 0, host: "127.0.0.1:1301" } ] }
rs.initiate( rsconf )
rs.add("127.0.0.1:1302")
rs.add("127.0.0.1:1303")
复制代码
//查看是否是主节点
rs.isMaster()
// 查看复制集状态
rs.status()
复制代码
编码
mkdir mongodb4
cd mongodb4
npm init
npm i mongodb -S
vi app.js
复制代码
//app.js
(async function() {
// 连接DB
const { MongoClient } = require('mongodb');
const uri = 'mongodb://localhost:1301/dbfour';
const client = await MongoClient.connect(uri, { useNewUrlParser: true });
const db = client.db();
await db.dropDatabase();
console.log('(1) 首先 删库 dbfour, then 跑路\n')
// 插入两个账户并充值一些金额
await db.collection('Account').insertMany([
{ name: 'A', balance: 50 },
{ name: 'B', balance: 10 }
]);
console.log('(2) 执行 insertMany, A 充值 50, B 充值 10\n')
await transfer('A', 'B', 10); // 成功
console.log('(3) 然后 A 给 B 转账 10\n')
try {
// 余额不足 转账失败
console.log('(4) A 再次转账给 B 50\n')
await transfer('A', 'B', 50);
} catch (error) {
//error.message; // "Insufficient funds: 40"
console.log(error.message)
console.log('\n(5) A 余额不够啊,所以这次转账操作不成功')
}
// 转账逻辑
async function transfer(from, to, amount) {
const session = client.startSession();
session.startTransaction();
try {
const opts = { session, returnOriginal: false };
const A = await db.collection('Account').
findOneAndUpdate({ name: from }, { $inc: { balance: -amount } }, opts).
then(res => res.value);
if (A.balance < 0) {
// 如果 A 的余额不足,转账失败 中止事务
// `session.abortTransaction()` 会撤销上面的 `findOneAndUpdate()` 操作
throw new Error('Insufficient funds: ' + (A.balance + amount));
}
const B = await db.collection('Account').
findOneAndUpdate({ name: to }, { $inc: { balance: amount } }, opts).
then(res => res.value);
await session.commitTransaction();
session.endSession();
return { from: A, to: B };
} catch (error) {
// 如果错误发生,中止全部事务并回退到修改之前
await session.abortTransaction();
session.endSession();
throw error; //使其调用者 catch error
}
}
})()
复制代码
查看结果
node app.js
复制代码