截止到今天,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。