app.js

var uuid = require('node-uuid');
var cookie = require('cookie');
var cookieParser = require('restify-cookies');
var nconf = require('nconf');
global._ROOT = process.env.PWD;
nconf.argv()
    .env()
    .file({file: global._ROOT + '/config/config.json'});
var nconfENV = nconf.get('NODE_ENV') || 'dev';
global.DB_CONFIG = nconf.get(nconfENV + ":db");
checkIfRequiredTablesExists();
var restify = require('restify');
var epsOptions = require('./controllers/eps_options/options');
var reward = require('./controllers/eps_options/reward');
var epsTest = require('./controllers/eps_options/eps_test');
var logger = require('./log/logger');
var server = restify.createServer({
    name: 'epsilon-ab-service',
    log: logger,
    versions: ['1.0.0']
});
server.use(restify.bodyParser());
server.use(restify.queryParser());
server.use(cookieParser.parse);
server.use(restify.requestLogger());
server.use(function (req, res, next) {
    req.connection.setTimeout(10000);
    next();
});
//Delete any timers added to logger
server.use(function (req, res, next) {
    res.on("header", function () {
        delete logger._times[req.log.fields.req_id];
    });
    next();
});
//Set unique id for each request, if the request have a cookie by name unique_id
//use it else generate a unique request id
server.use(function (req, res, next) {
    var uniqueId = uuid.v1();
    if (req.headers.cookie) {
        var cookies = cookie.parse(req.headers.cookie);
        uniqueId = cookies.unique_id || uniqueId;
    }
    req.req_id = uniqueId || req.req_id;
    res.setHeader('X-Request-Id', req.req_id);
    next();
});
//Accept only json content types
server.pre(function (req, res, next) {
    res.setHeader('content-type', 'application/json');
    if (!req.is('application/json')) {
        res.send(400, {"ERROR": 'Invalid content-type'});
        return;
    }
    if (!req.accepts('application/json')) {
        res.send(400, {"ERROR": 'Invalid accept-type'});
        return;
    }
    next();
});
var port = process.env.PORT || '5000';
server.use(function (req, res, next) {
    res.on("header", function () {
        delete logger._times[req.log.fields.req_id];
    });
    next();
});

server.get('/epsOption', epsOptions.get);
server.put('/epsTest/reward', reward.put);
server.put('/epsTest/conversion', epsTest.userConverted);
server.post('/epsTest', epsTest.create);
server.get('/epsTest/:id', epsTest.findById);
server.get('/epsTest', epsTest.findAll);
server.post('/epsTest/:id/options/:optionNumber', epsTest.update);
server.get('/epsTest/:name/stats', epsTest.stats);

server.listen(port, function () {
    logger.info('epsilon ab server listening on port ' + port);
});

function checkIfRequiredTablesExists() {
    if (global.DB_CONFIG['default']['name'] === 'mysql') {
        var mysql = require('mysql');
        global._MYSQL_CONNECTION_POOL = mysql.createPool(global.DB_CONFIG['default']);
        var mysqlClient = require(global._ROOT + '/db/pgsql/mysql_client');
        new Promise(function (resolve, reject) {
            mysqlClient.haveAllRequiredTables(global._MYSQL_CONNECTION_POOL, reject, resolve);
        }).then(function (res) {
                if (res['tableCount'] != 3) {
                    console.log("**ATTENTION SOME TABLES ARE MISSING IN THE SCHEMA. CREATE THE TABLES SPECIFIED UNDER schema/mysql/**");
                }
            });
    } else if (global.DB_CONFIG['default']['name'] === 'pgsql') {
        var pgClient = require(global._ROOT + '/db/pgsql/pgsql');
        pgClient.haveAllRequiredTables()
            .then(function (rows) {
                if (!rows || rows.length != 3) {
                    console.log("**ATTENTION SOME TABLES ARE MISSING IN THE SCHEMA. CREATE THE TABLES SPECIFIED UNDER schema/pgsql/**");
                }
            })
    }
}