h-orm/lib/mysql/index.js

H-orm is a mysql orm for nodejs

const mysql = require('mysql')
const colors = require('colors')
const debugTip = colors.bgYellow
const debugMsg = colors.cyan

module.exports = {
    pool: null,
    connConfig: {},
    debug: false,

    __getPool() {
        if (!this.pool) {
            this.debug = this.connConfig.debug
            this.pool = mysql.createPool({
                connectionLimit : this.connConfig.limit || 10,
                port            : this.connConfig.port || 3306,
                host            : this.connConfig.host || '',
                user            : this.connConfig.user || '',
                password        : this.connConfig.password || '',
                database        : this.connConfig.database || ''
            })
        }
        return this.pool

    },
    init(config) {
        if (this.debug) {
            console.log(debugMsg(config))
        }
        if (!Object.keys(config).length) {
            throw new Error('[h-rom]: db config muse hava value when you use init function')
        }
        this.connConfig = config
        return this.__getPool()
    },
    query(sql, value) {
        if (this.debug) {
            console.log(`${debugTip(' H-ORM ')} ${debugMsg(sql)}`)
        }
        return new Promise((resolve, reject) => {
            if (!this.pool) {
                reject(new Error('[h-rom]: you should init db before use query'))
            } else {
                this.pool.getConnection(function(err, connection) {
                    if (err) {
                        reject(err)
                    } else {
                        connection.query(sql, value, function (error, results, fields) {
                            connection.release() // 释放
                            // connection.destroy() // 关闭
                            if (error) {
                                reject(error)
                            }
                            resolve({
                                results,
                                fields
                            })
                        })
                    }
                })
            }
        })
    },
    escape(data) {
        return mysql.escape(data)
    },
    close() {
        this.pool.getConnection(function(err, connection) {
            connection.destroy()
        })
    }
}