node连接mysql查询事务处理的实现

D:\nodejs_mysql>npm init -y  # 快速传创建包管理文件

D:\nodejs_mysql>npm i mysql -D # 安装mysql 依赖包

const mysql = require("mysql");

// 创建 一个mysql 连接池
const pool = mysql.createPool({
  host: "127.0.0.1", // 连接数据库的地址 127.0.0.1 为本地的mysql
  user: "root", //  连接数据库的用户名 root 是最高权限
  password: "", //  连接数据库的密码
  database: "apm", // 操作的数据库名
  port: "3306", // 连接mysql的端口号
  multipleStatements: true, // 运行一次执行多条sql语句 可以忽略此项
});

function select() {
  // 请求连接mysql
  pool.getConnection((err, connection) => {
    // 未连接成功 报错
    if (err) throw err;
    // 得到一个 连接对象 调用 query 方法 可以执行 sql 语句
    let sql = "select * from goods";
    // 运行sql语句 query 第二个参数为 sql语句需要的参数 ，没有可以不写
    connection.query(sql, (errors, results, fields) => {
      // 释放连接
      connection.release();
      // 如果运行sql语句有报错 抛出错误
      if (errors) throw errors;
      console.log(results);
      console.log(fields);
    });
  });
}

select();

//  index.js
// ... 忽略前面的创建mysql连接池代码
function query(sql, params) {
  return new Promise((resolve, reject) => {
    pool.getConnection((err, connection) => {
      // 未连接成功 报错
      if (err) return reject(err);
      // 得到一个 连接对象 调用 query 方法 可以执行 sql 语句
      // 运行sql语句 query 第二个参数为 sql语句需要的参数 ，没有可以不写
      connection.query(sql, params, (errors, results, fields) => {
        // 释放连接
        connection.release();
        // 如果运行sql语句有报错 抛出错误
        if (errors) return reject(errors);
        resolve(results);
      });
    });
  });
}

query("select * from goods", null).then((result) => {
  console.log(result);
});
module.exports = {
  query,
};

// data.js
const index = require("./index.js");
var sql = "select * from goods";
index
  .query(sql, null)
  .then((result) => {
    // do anything ....
  })
  .catch((err) => {
    // error
    console.log(err);
  });

// index.js

// .... 部分创建 pool 的代码

/**
 * mysql 事务处理
 * @param {Array} sqls 需要执行的sql语句
 * @param {Array} params 对应上面sql语句的参数
 * @returns {Promise} 返回一个Promise
 */
function transaction(sqls, params) {
  return new Promise((resolve, reject) => {
    pool.getConnection(function (err, connection) {
      // 连接失败 promise直接返回失败
      if (err) {
        return reject(err);
      }
      // 如果 语句和参数数量不匹配 promise直接返回失败
      if (sqls.length !== params.length) {
        connection.release(); // 释放掉
        return reject(new Error("语句与传值不匹配"));
      }
      // 开始执行事务
      connection.beginTransaction((beginErr) => {
        // 创建事务失败
        if (beginErr) {
          connection.release();
          return reject(beginErr);
        }
        console.log("开始执行事务，共执行" + sqls.length + "条语句");
        // 返回一个promise 数组
        let funcAry = sqls.map((sql, index) => {
          return new Promise((sqlResolve, sqlReject) => {
            const data = params[index];
            connection.query(sql, data, (sqlErr, result) => {
              if (sqlErr) {
                return sqlResolve(sqlErr);
              }
              sqlReject(result);
            });
          });
        });
        // 使用all 方法 对里面的每个promise执行的状态 检查
        Promise.all(funcAry)
          .then((arrResult) => {
            // 若每个sql语句都执行成功了 才会走到这里 在这里需要提交事务，前面的sql执行才会生效
            // 提交事务
            connection.commit(function (commitErr, info) {
              if (commitErr) {
                // 提交事务失败了
                console.log("提交事务失败:" + commitErr);
                // 事务回滚，之前运行的sql语句不生效
                connection.rollback(function (err) {
                  if (err) console.log("回滚失败：" + err);
                  connection.release();
                });
                // 返回promise失败状态
                return reject(commitErr);
              }

              connection.release();
              // 事务成功 返回 每个sql运行的结果 是个数组结构
              resolve(arrResult);
            });
          })
          .catch((error) => {
            // 多条sql语句执行中 其中有一条报错 直接回滚
            connection.rollback(function () {
              console.log("sql运行失败： " + error);
              connection.release();
              reject(error);
            });
          });
      });
    });
  });
}
module.exports = {
  transaction,
};

// data.js
const index = require("./index.js");
var sqls = [
  "delete from goods where goods_id = ?", // 删除 语句
  "update goods set num = ? where goods_id = ?;", // 更新语句
];
var params = [
  [1], // parmas 是数组格式 与sqls里的sql语句里 ? 一一对应
  [5, 3],
];

index
  .transaction(sqls, params)
  .then((arrResult) => {
    // do anything ....
  })
  .catch((err) => {
    // error
    console.log(err);
  });