如何通过 Sequelize 交易保存方法 - JavaScript

Sequelize 是 Node.js 中使用广泛的 ORM 工具之一，它提供了轻松管理关系型数据库的能力。在很多情况下我们需要进行多个操作才能完成一次完整的数据库事务，例如在转账的时候需要先验证账户余额是否充足，然后才能完成转账的操作。这时我们需要使用 Sequelize 中的 transactions 功能来确保这些操作都能原子性地整合在一起。

创建数据库事务

在 Sequelize 中，事务是通过 sequelize.transaction 方法来创建的。该方法的使用方式如下：

const transaction = await sequelize.transaction();
// 代码片段1
await transaction.commit();

其中，sequelize 是初始化 Sequelize 实例时的返回值。我们可以通过 sequelize.transaction 方法创建一个事务对象，该对象有两种常见的使用方式：

1. 在单个 Sequelize 方法中使用。

await Model.create(data, { transaction });

使用该方式创建的操作都会在事务的作用域内运行，当事务成功提交时，这些操作会被提交并进行持久化。如果事务执行过程中发生了错误，这些操作也会被回滚。

2. 在多个 Sequelize 方法中使用。

await sequelize.transaction(async t => {
  // 代码片段2
});

使用该方式创建的操作需要手动传入事务对象，且在多个 Sequelize 方法中进行操作。当事务成功提交时，这些操作会被提交并进行持久化。如果事务执行过程中发生了错误，这些操作也会被回滚。

事务的提交与回滚

在事务执行过程中，如果所有 Sequelize 操作执行成功，则可以通过 await transaction.commit() 方法手动提交事务。如果事务过程中发生了错误，则可以通过 await transaction.rollback() 方法手动回滚事务。

我们可以将 Sequelize 操作放置在一个事务中执行，当我们需要进行多个操作时，如果其中一个操作执行失败，则所有的操作都会被回滚。

完整示例代码

下面是一个使用 Sequelize 进行数据库事务的示例代码：

// sequelize 实例初始化
const Sequelize = require('sequelize');
const sequelize = new Sequelize('database', 'username', 'password', {
  host: 'localhost',
  dialect: 'mysql',
});

// 定义 Model
const User = sequelize.define('User', {
  name: Sequelize.STRING,
  balance: Sequelize.INTEGER,
});

// 创建初始数据
await User.create({ name: 'user1', balance: 1000 });
await User.create({ name: 'user2', balance: 2000 });

// 数据库事务处理
await sequelize.transaction(async t => {
  // 查询账户余额
  const user1 = await User.findOne({ where: { name: 'user1' }, transaction: t });
  const user2 = await User.findOne({ where: { name: 'user2' }, transaction: t });
  const amount = 500;

  // 检查账户余额是否充足
  if (user1.balance < amount) {
    throw new Error('Insufficient balance');
  }

  // 转账操作，扣除 user1 的余额，增加 user2 的余额
  await user1.decrement('balance', { by: amount, transaction: t });
  await user2.increment('balance', { by: amount, transaction: t });
});

// 查询转账后的账户余额
const user1_balance = await User.findOne({ where: { name: 'user1' } }).balance;
const user2_balance = await User.findOne({ where: { name: 'user2' } }).balance;

console.log(`user1's balance after transaction: ${user1_balance}`);
console.log(`user2's balance after transaction: ${user2_balance}`);

在上述代码中，我们使用 Sequelize 中的 decrement 和 increment 方法来修改账户余额，同时使用查询方法 findOne 来查询账户余额。使用事务之后，我们在程序执行过程中，可以保证查询、修改等操作之间的一致性，保证操作的原子性和完整性。