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

上述案例是我截取項目的一段代碼，從上面代碼顯然可以很容易封裝自己的邏輯，可以執行不同路徑的事務流程。

Node Mysql事務處理封裝