告别回调地狱，在Node里优雅的访问MySQL（一）

/**
* 基于 promise 封装 MySQL 的基本操作，支持事务
* * 创建数据库连接
* * 事务相关
* * 提交SQL给数据库执行，得到返回结果
* * 共用函数
*/
class MySQLHelp {
constructor (info) {
// 创建普通连接的参数
this._info = {
host: 'localhost',
port: '3306',
user: 'root',
password: '',
database: ''
}
// 创建池子的参数
this._infoTran = {
host: 'localhost',
port: '3306',
user: 'root',
password: '',
database: '',
connectionLimit: 20
}
// 加参数
Object.assign(this._info, info)
Object.assign(this._infoTran, info)
// 预定一个池子，用于事务
this.pool = null
console.log(' ★ 初始化：不使用事务！')
// 不使用事务，开启连接获得数据库对象，其实也支持事务
this.db = mysql.createConnection(this._info)
//启动连接
this.db.connect((err) => {
if(err) {
console.error('连接数据发生错误：', err)
} else {
console.log('★ [MySQL 已经打开数据库，threadId：]', this.db.threadId)
}
})
}

/**
* 把 SQL 提交给数据库。支持事务。
* @param { string } sql sql语句
* @param { array } params 需要的参数，数组形式
* @param { connection } cn 如果使用事务的话，需要传递一个链接对象进来
* @returns Promise 形式
*/
query(sql, params=[], cn = this.db) {
const myPromise = newPromise((resolve, reject) => {
// 把SQL语句提交给数据库执行，然后等待回调
cn.query(sql, params, (err, res, fields) => {
if (err) { //失败
// 如果开始事务，自动回滚
if (cn !== this.db) {
cn.rollback((err) => {
console.log('执行SQL失败，数据回滚：', err)
})
}
reject(err)
return
}
resolve(res)
})
})
return myPromise
}

/**
* 在池子里面获取一个链接，创建对象
*/
_poolCreateConnection() {
const myPromise = newPromise((resolve, reject) => {
console.log('初始化：使用事务')
// 如果没有池子则创建一个。好在不是异步
if (this.pool === null) {
this.pool = mysql.createPool(this._infoTran)
}
// 从池子里面获取一个链接对象，这个是异步
this.pool.getConnection((err, connection) => {
if(err) {
reject(err)
} else {
resolve(connection)
}
})
})
return myPromise
}

/**
* 内部开启事务
*/
_beginStran(_cn) {
const myPromise = newPromise((resolve, reject) => {
_cn.beginTransaction((err) => {
if (err) { //失败
console.log('[★ ★ MySQL 开启事务时报错：] --- ', err)
reject(err)
return
}
console.log('[★ MySQL 事务已经开启：] - ')
resolve(_cn)
})
})
return myPromise
}

/**
* 开启一个事务，Promise 的方式
*/
begin() {
const myPromise = newPromise((resolve, reject) => {
console.log('★ 开启事务，promise 模式')
this._poolCreateConnection().then((_cn) => {
// 开启事务
this._beginStran(_cn)
.then(() => {resolve(_cn)}) // 返回连接对象
.catch((err) => {reject(err)})
})
})
return myPromise
}