Node.js+MySQL如何promisify pool.getConnection用async/await实现事务

你之前的预期代码无法运行的根本原因是：仅对pool.getConnection做了promisify处理，但从连接池取出的connection实例上的beginTransaction、query、commit、release方法本身还是回调风格，没有被转成Promise形式，自然无法直接搭配await使用。不需要混合两种编码风格，也不用强制换库，只需要补全对应方法的promisify绑定即可。

const mysql = require('mysql');
const util = require('util');

// 初始化连接池，替换为实际数据库配置
const pool = mysql.createPool({
  host: 'localhost',
  user: 'your_username',
  password: 'your_password',
  database: 'your_db_name',
  connectionLimit: 10
});

// 将getConnection转为Promise版本，绑定pool上下文避免this指向错误
pool.getConnection = util.promisify(pool.getConnection).bind(pool);

/**
 * 事务执行封装
 * @param {Function} task 事务内具体逻辑，接收已完成promisify的连接实例
 * @returns 事务执行结果
 */
async function runTransaction(task) {
  // 从连接池获取独立连接
  const connection = await pool.getConnection();
  try {
    // 将当前连接上所有事务相关方法批量转为Promise版本，绑定connection上下文
    ['beginTransaction', 'query', 'commit', 'rollback'].forEach(method => {
      connection[method] = util.promisify(connection[method]).bind(connection);
    });

    // 开启事务
    await connection.beginTransaction();
    // 执行传入的事务逻辑，所有操作复用当前连接
    const result = await task(connection);
    // 无错误则提交事务
    await connection.commit();
    return result;
  } catch (err) {
    // 出现错误先回滚
    await connection.rollback();
    throw err;
  } finally {
    // 无论执行成功/失败，最终都将连接放回连接池
    connection.release();
  }
}

// 调用示例
async function bizOperation() {
  const insertSql = 'INSERT INTO user (name) VALUES (?)';
  const updateSql = 'UPDATE account SET balance = balance - 100 WHERE user_id = ?';
  
  try {
    const [insertResult, updateResult] = await runTransaction(async (conn) => {
      const res1 = await conn.query(insertSql, ['test_user']);
      const res2 = await conn.query(updateSql, [res1.insertId]);
      return [res1, res2];
    });
    console.log('事务执行成功', insertResult, updateResult);
  } catch (err) {
    console.error('事务执行失败，已自动回滚', err);
  }
}

bizOperation();

• 所有promisify操作必须用.bind()绑定原方法所属的实例上下文，否则mysql驱动内部的this指向会错乱，直接抛出类型错误。
• 不要用全局promisify后的pool.query实现事务：pool.query是驱动封装的快捷方法，内部会自动获取连接、执行SQL、释放连接，每次调用都是独立连接，无法保证多个SQL在同一个连接中执行，根本不满足事务的基础要求。
• 事务逻辑必须加try/catch/finally结构：SQL执行报错时要先触发回滚，最后必须释放连接，否则连接会一直被占用，最终导致连接池耗尽。

如果不想手动做promisify绑定，可以直接替换为mysql2库，它内置了Promise API，原生支持async/await写法，连接池、事务的API设计和mysql库基本兼容，迁移成本极低。

内容的提问来源于stack exchange，提问作者Jonathan