寂寞大萝卜 发表于 2018-10-2 06:44:32

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, )
  .then(result => {
  return trans.query(conn,db_user.login, )
  }).then(result => {
  return trans.query(conn,db_user.register.profile,.id, nickname])
  })
  }).then(result=>{
  console.log(‘事务提交成功’)
  }).catch(err => {
  console.log(‘事务提交失败’)
  })
  

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


页: [1]
查看完整版本: Node Mysql事务处理封装