mysqli/index.js

/**
 * 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({
      removeNodeErrorCount: 1, // 连续失败立即从节点中移除, 并在10秒后尝试恢复
      restoreNodeTimeout: 10000
    })

    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