mysqljs在koa2中的正确姿势

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