68 lines
1.3 KiB
JavaScript
68 lines
1.3 KiB
JavaScript
/**
|
|
* mysql操作类
|
|
* @authors yutent (yutent@doui.cc)
|
|
* @date 2015-11-24 11:31:55
|
|
*
|
|
*/
|
|
'use strict'
|
|
require('es.shim')
|
|
const mysql = require('mysql')
|
|
const Method = require('./lib/method')
|
|
|
|
if (!Promise.defer) {
|
|
Promise.defer = function() {
|
|
let obj = {}
|
|
obj.promise = new this((yes, no) => {
|
|
obj.resolve = yes
|
|
obj.reject = no
|
|
})
|
|
return obj
|
|
}
|
|
}
|
|
class Mysqli {
|
|
/**
|
|
* [constructor 构造数据库连接池]
|
|
*/
|
|
constructor(config) {
|
|
if (!Array.isArray(config)) {
|
|
config = [config]
|
|
}
|
|
|
|
//是否有从库
|
|
this.useSlaveDB = config.length > 1
|
|
this.pool = mysql.createPoolCluster()
|
|
|
|
config.forEach((item, i) => {
|
|
let { host, port, user, charset, passwd: password, db: database } = item
|
|
let name = i < 1 ? 'MASTER' : 'SLAVE' + i
|
|
|
|
charset = charset || 'utf8'
|
|
let collate =
|
|
charset + (charset === 'utf8mb4' ? '_unicode_ci' : '_general_ci')
|
|
|
|
this.pool.add(name, {
|
|
host,
|
|
port,
|
|
user,
|
|
charset,
|
|
collate,
|
|
password,
|
|
database
|
|
})
|
|
})
|
|
return this
|
|
}
|
|
|
|
//对外的escape方法
|
|
static escape(val) {
|
|
return mysql.escape(val)
|
|
}
|
|
|
|
emit(fromSlave = false, db) {
|
|
const slave = fromSlave && this.useSlaveDB ? 'SLAVE*' : 'MASTER'
|
|
return new Method(this.pool, slave, db)
|
|
}
|
|
}
|
|
|
|
module.exports = Mysqli
|