截止到今天,mysqljs在github上已经获取到了10000+star了,可以说是实实在在最流行的mysql驱动了,但是要把mysqljs应用到koa2中,似乎不太方便,koa2使用最新的语法async, await,而mysqljs,却还停留在callback阶段。
今天这篇文章就是要解决这个问题,非常简单。
1、实际开发中,我们肯定是使用连接池的形式,所以,我们选择了mysql.createPool这个方法:
var mysql = require('mysql');
var pool = mysql.createPool(...);
pool.getConnection(function(err, connection) {
// Use the connection
connection.query('SELECT something FROM sometable', function (error, results, fields) {
// And done with the connection.
connection.release();
// Handle error after the release.
if (error) throw error;
// Don't use the connection here, it has been returned to the pool.
});
});
2、使用Promise,对上面的方法稍加改造即可:
var mysql = require('mysql');
var pool = mysql.createPool(...);
const q = function (sql, values) {
return new Promise((resolve, reject) => {
pool.getConnection((err, conn) => {
if (err) return reject(err)
conn.query(sql, values, (err, rows) => {
if (err) reject(err)
else resolve(rows)
conn.release()
})
})
})
}
经过以上封装,一个查询用户信息的操作就可以这样优雅的完成了:
async function getUserInfoById(id) {
let userInfo = await q('select * from user where id=?', [id])
console.log(userInfo)
}
3、受tornado的一个mysql操作库torndb的启发,可以这样做一个完整的封装:
const mysql = require('mysql')
const defautConfig = {
host: 'localhost',
user: 'root',
password: '',
database: 'test',
connectionLimit: 20
}
const AsyncMysqljs = function(config=defautConfig){
const pool = mysql.createPool(config)
const q = function (sql, values) {
return new Promise((resolve, reject) => {
pool.getConnection((err, conn) => {
if (err) return reject(err)
conn.query(sql, values, (err, rows) => {
if (err) reject(err)
else resolve(rows)
conn.release()
})
})
})
}
/*
从数据库中查询一条数据,返回值是对象,而非数组
最好在sql语句中加一个唯一的限制条件
*/
const get = (sql, values) => {
try {
return q(sql, values).then(rows => {
if (rows.length >= 1) {
return rows[0]
}
})
} catch (err) {
return new Promise((resolve, reject) => {
reject(err)
})
}
}
return {query: q, delete: q, update: q, insert: q, execute: q, get}
}
module.exports = AsyncMysqljs
具体代码请查看我的github项目asyncmysqljs,欢迎给建议或者star。