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(‘事务提交失败’)
})
上述案例是我截取项目的一段代码,从上面代码显然可以很容易封装自己的逻辑,可以执行不同路径的事务流程。

猜你喜欢

转载自blog.51cto.com/14451335/2421065