pool_cluster.js

var _ = require('underscore');
var configuration = require('../configuration');
var mysql = require('mysql');
var connection = mysql.createConnection({
    host: configuration.database.host,
    user: configuration.database.user,
    password: configuration.database.password,
    database: configuration.database.name
});

var poolCluster = mysql.createPoolCluster({
    canRetry: true,
    removeNodeErrorCount: 1,
    restoreNodeTimeout: 20000,
    defaultSelector: 'RR'
});

poolCluster.add('MASTER', {
    connectionLimit: 200,
    host: configuration.database.host,
    port: 3306,
    user: configuration.database.user,
    password: configuration.database.password,
    database: configuration.database.name
});

var createNodeMySQL = require('../mysql-wrap');

exports.setUp = function (done) {
    this.sql = createNodeMySQL(poolCluster, {
        replication: {
            write: 'MASTER',
            read: '*'
        }
    });

    poolCluster.getConnection('*', function (err, connection) {
        connection.query('TRUNCATE TABLE `table`', function (err, res) {
            connection.query(
                'INSERT INTO `table` (`unique`, `field`) ' +
                'VALUES ("a", "foo"), ("b", "bar"), ("c", "foo")',
                function (err, res) {
                    connection.query('TRUNCATE TABLE `table2`', function (err, res) {
                        connection.query(
                            'INSERT INTO `table2` (`field`) ' +
                            'VALUES ("bar")',
                            function () {
                                done();
                            }
                        );
                    });
                }
            );
        });
    });
};

_.each(require('./shared_tests.js')(connection, createNodeMySQL), function (test, name) {
    exports[name] = test;
});