mysql handler

mysql.js

var mysql = require('mysql');
var inherits = require('util').inherits;
var EventEmitter = require('events').EventEmitter;

module.exports = MySql;

function MySql(config) {
    var clusterConfig = {
        removeNodeErrorCount: 1, // Remove the node immediately when connection fails.
        defaultSelector: 'RR',
        canRetry: true
    };

    this.poolCluster = mysql.createPoolCluster(clusterConfig);
    this.dirties = {};
    this.connections = {};
    this.config = config;

    this.connect(config);
};

inherits(MySql, EventEmitter);

MySql.prototype.connect = function(config) {
    var self = this;

    config.forEach(function(connectionGroup) {
        self.poolCluster.add('MASTER', connectionGroup.master);
        for( var i = 0; i < connectionGroup.slaves.length; i++) {
            self.poolCluster.add('SLAVE'+(i+1), connectionGroup.slaves[i]);
        }
    });
};


MySql.prototype.getConnection = function(key, callback) {
    var self = this;

    self.poolCluster.getConnection(key, callback);
};

MySql.prototype.query = function(statement, args, callback) {
    var self = this;
    self.getConnection('MASTER', function(error, connection) {
        connection.query(statement, args, function(err, result) {
            connection.release();
            callback(err, result);
        });
    });
};