mysqljs在koa2中的正確姿勢

Happi發表於2019-02-16

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

相關文章