MongoDB 的存储库模式 - 一个事务处理多个工作单元

MongoDB 是一种流行的非关系型数据库，它采用了一种称为存储库模式的架构，以实现高效的事务处理。存储库模式是一种将多个工作单元（也称为命令）组合在一起执行的方法，以确保数据的完整性和一致性。在本文中，我们将了解存储库模式的工作原理，并通过一个实际案例来说明其应用。

// 导入 MongoDB 驱动程序
const { MongoClient } = require('mongodb');
// 定义数据库连接 URL
const url = 'mongodb://localhost:27017';
// 定义数据库名称
const dbName = 'mydb';
// 创建一个存储库对象
const repository = {
  async executeTransaction(commands) {
    // 连接数据库
    const client = new MongoClient(url);
    try {
      // 连接到数据库
      await client.connect();
      // 开始事务
      const session = client.startSession();
      try {
        // 开始事务
        session.startTransaction();
        // 执行所有工作单元
        for (const command of commands) {
          await command(session);
        }
        // 提交事务
        await session.commitTransaction();
      } catch (error) {
        // 回滚事务
        await session.abortTransaction();
        throw error;
      } finally {
        // 结束事务
        session.endSession();
      }
    } finally {
      // 断开数据库连接
      client.close();
    }
  }
};
// 定义工作单元
const insertDocument = async (session) => {
  const db = session.client.db(dbName);
  const collection = db.collection('documents');
  await collection.insertOne({ name: 'John Doe' });
};
const updateDocument = async (session) => {
  const db = session.client.db(dbName);
  const collection = db.collection('documents');
  await collection.updateOne({ name: 'John Doe' }, { $set: { age: 30 } });
};
const deleteDocument = async (session) => {
  const db = session.client.db(dbName);
  const collection = db.collection('documents');
  await collection.deleteOne({ name: 'John Doe' });
};
// 执行多个工作单元并应用事务处理
repository.executeTransaction([insertDocument, updateDocument, deleteDocument])
  .then(() => {
    console.log('事务执行成功');
  })
  .catch((error) => {
    console.error('事务执行失败:', error);
  });