Node Mysql事务处理封装

Posted 2020-10-29

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(‘事务提交失败’)
     })

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