node是基于异步的,因此在进行数据库查询操作的通常是通过回调来操作查询结果。但是在有了es7的async/await,基本不再需要回调了,所以本篇是基于async/await对mysql进行一次操作封装,让查询更加方便。(node版本需>=7.0)。
原理
简单来说,async/await的实现原理是基于promise,根据promise的状态来判断是否真正返回,因此我们可以在mysql真正查询到结果后将promise状态切换为resolve,返回结果。如出现错误通过reject返回错误信息,reject需要用try/catch进行捕获。
代码
'use strict'; const mysql = require( 'mysql' ); var pool = mysql.createPool( { connectionLimit : 50, host : '', user : '', password : '', database : '', multipleStatements : true //是否允许执行多条sql语句 } ); //将结果已对象数组返回 var row=( sql , ...params )=>{ return new Promise(function(resolve,reject){ pool.getConnection(function(err,connection){ if(err){ reject(err); return; } connection.query( sql , params , function(error,res){ connection.release(); if(error){ reject(error); return; } resolve(res); }); }); }); }; //返回一个对象 var first=( sql , ...params )=>{ return new Promise(function(resolve,reject){ pool.getConnection(function(err,connection){ if(err){ reject(err); return; } connection.query( sql , params , function(error,res){ connection.release(); if(error){ reject(error); return; } resolve( res[0] || null ); }); }); }); }; //返回单个查询结果 var single=(sql , ...params )=>{ return new Promise(function(resolve,reject){ pool.getConnection(function(err,connection){ if(err){ reject(err); return; } connection.query( sql , params , function(error,res){ connection.release(); if(error){ reject( error ); return; } for( let i in res[0] ) { resolve( res[0][i] || null ); return; } resolve(null); }); }); }); } //执行代码,返回执行结果 var execute=(sql , ...params )=>{ return new Promise(function(resolve,reject){ pool.getConnection(function(err,connection){ if(err){ reject(err); return; } connection.query( sql , params , function(error,res){ connection.release(); if(error){ reject(error); return; } resolve( res ); }); }); }); } //模块导出 module.exports = { ROW : row , FIRST : first , SINGLE : single , EXECUTE : execute }
使用示例
const mysql = require('./mysql.js'); (async ()=>{ let s = await mysql.row(sql,params); console.log(s); })();