MongoDB4 事務 簡單易懂的?

簡介

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 並解壓

www.mongodb.com/download-ce…

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
複製程式碼