Node Mysql事务处理封装

寂寞大萝卜

　　node回调函数的方式使得数据库事务貌似并没有像java、php那样编写简单，网上找了一些事务处理的封装并没有达到自己预期的那样简单编写，还是自己封装一个吧。
　　封装的大体思路很简单：函数接受一个事务处理的函数，如何处理事务的逻辑由该函数确定，而不是像网上封装的事务是对组装多条预处理SQL和参数。该事务处理函数要求返回Promise。这样我们就可以通过该Promise来确定是提交事务还是回滚事务了。
　　根据上述思路，实现代码：
　　

let mysql = require('mysql')　　
let config = require('../config')
　　

　　
let pool = mysql.createPool(config.database) //config.databa数据库配置
　　
let trans = (tran) => {
　　return new Promise((resolve, reject) => {  //返回promise提供事务成功与失败的接口
　　pool.getConnection((err, conn) => {
　　if(err) {
　　reject(err)
　　}else {
　　conn.beginTransaction((err) => { //开始事务处理
　　if(err) {
　　conn.release()
　　reject(err)
　　}else {
　　let promise = tran(conn)  //调用事务处理函数
　　promise.then(result => {
　　conn.commit(err => {  //事务处理函数resolve则提交事务
　　if(err) {
　　reject(err)
　　}else {
　　resolve(result)
　　}
　　})
　　}).catch(err => {
　　conn.rollback(() => {  //事务处理函数reject则回滚事务
　　conn.release()
　　reject(err)
　　})
　　})
　　}
　　})
　　}
　　})
　　})
　　
}
　　

　　事务处理函数接受数据库连接对象conn，这里我封装了数据库执行语句：
　　

trans.query = (conn, sql, params) => {　　return new Promise((resolve, reject) => {
　　conn.query(sql, params,(err, result) => {
　　if(err) {
　　reject(err)
　　}else {
　　resolve(result)
　　}
　　})
　　})
　　
}
　　

　　上面事务处理封装完成，应用就和正常Promise使用相同：
　　

trans((conn) => {　　return trans.query(conn, db_user.register.user, [username, password, new Date().getTime()])
　　.then(result => {
　　return trans.query(conn,db_user.login, [username])
　　}).then(result => {
　　return trans.query(conn,db_user.register.profile,[result[0].id, nickname])
　　})
　　}).then(result=>{
　　console.log(‘事务提交成功’)
　　}).catch(err => {
　　console.log(‘事务提交失败’)
　　})
　　

　　上述案例是我截取项目的一段代码，从上面代码显然可以很容易封装自己的逻辑，可以执行不同路径的事务流程。