D:\nodejs_mysql>npm init -y  # 快速傳創(chuàng)建包管理文件

D:\nodejs_mysql>npm i mysql -D # 安裝mysql 依賴包

const mysql = require("mysql");

// 創(chuàng)建 一個mysql 連接池
const pool = mysql.createPool({
  host: "127.0.0.1", // 連接數(shù)據庫的地址 127.0.0.1 為本地的mysql
  user: "root", //  連接數(shù)據庫的用戶名 root 是最高權限
  password: "", //  連接數(shù)據庫的密碼
  database: "apm", // 操作的數(shù)據庫名
  port: "3306", // 連接mysql的端口號
  multipleStatements: true, // 運行一次執(zhí)行多條sql語句 可以忽略此項
});

function select() {
  // 請求連接mysql
  pool.getConnection((err, connection) => {
    // 未連接成功 報錯
    if (err) throw err;
    // 得到一個 連接對象 調用 query 方法 可以執(zhí)行 sql 語句
    let sql = "select * from goods";
    // 運行sql語句 query 第二個參數(shù)為 sql語句需要的參數(shù) ，沒有可以不寫
    connection.query(sql, (errors, results, fields) => {
      // 釋放連接
      connection.release();
      // 如果運行sql語句有報錯 拋出錯誤
      if (errors) throw errors;
      console.log(results);
      console.log(fields);
    });
  });
}

select();

//  index.js
// ... 忽略前面的創(chuàng)建mysql連接池代碼
function query(sql, params) {
  return new Promise((resolve, reject) => {
    pool.getConnection((err, connection) => {
      // 未連接成功 報錯
      if (err) return reject(err);
      // 得到一個 連接對象 調用 query 方法 可以執(zhí)行 sql 語句
      // 運行sql語句 query 第二個參數(shù)為 sql語句需要的參數(shù) ，沒有可以不寫
      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

// .... 部分創(chuàng)建 pool 的代碼

/**
 * mysql 事務處理
 * @param {Array} sqls 需要執(zhí)行的sql語句
 * @param {Array} params 對應上面sql語句的參數(shù)
 * @returns {Promise} 返回一個Promise
 */
function transaction(sqls, params) {
  return new Promise((resolve, reject) => {
    pool.getConnection(function (err, connection) {
      // 連接失敗 promise直接返回失敗
      if (err) {
        return reject(err);
      }
      // 如果 語句和參數(shù)數(shù)量不匹配 promise直接返回失敗
      if (sqls.length !== params.length) {
        connection.release(); // 釋放掉
        return reject(new Error("語句與傳值不匹配"));
      }
      // 開始執(zhí)行事務
      connection.beginTransaction((beginErr) => {
        // 創(chuàng)建事務失敗
        if (beginErr) {
          connection.release();
          return reject(beginErr);
        }
        console.log("開始執(zhí)行事務，共執(zhí)行" + sqls.length + "條語句");
        // 返回一個promise 數(shù)組
        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執(zhí)行的狀態(tài) 檢查
        Promise.all(funcAry)
          .then((arrResult) => {
            // 若每個sql語句都執(zhí)行成功了 才會走到這里 在這里需要提交事務，前面的sql執(zhí)行才會生效
            // 提交事務
            connection.commit(function (commitErr, info) {
              if (commitErr) {
                // 提交事務失敗了
                console.log("提交事務失敗:" + commitErr);
                // 事務回滾，之前運行的sql語句不生效
                connection.rollback(function (err) {
                  if (err) console.log("回滾失?。? + err);
                  connection.release();
                });
                // 返回promise失敗狀態(tài)
                return reject(commitErr);
              }

              connection.release();
              // 事務成功 返回 每個sql運行的結果 是個數(shù)組結構
              resolve(arrResult);
            });
          })
          .catch((error) => {
            // 多條sql語句執(zhí)行中 其中有一條報錯 直接回滾
            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 是數(shù)組格式 與sqls里的sql語句里 ? 一一對應
  [5, 3],
];

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