This commit is contained in:
barisusakli
2015-12-21 12:02:10 +02:00
parent 4bff714947
commit b3f63e0a0d
4 changed files with 78 additions and 192 deletions

View File

@@ -1,56 +1,14 @@
"use strict";
var nconf = require('nconf'),
primaryDBName = nconf.get('database'),
secondaryDBName = nconf.get('secondary_database'),
secondaryModules = nconf.get('secondary_db_modules'),
winston = require('winston'),
async = require('async'),
var nconf = require('nconf');
var databaseName = nconf.get('database');
var winston = require('winston');
ALLOWED_MODULES = ['hash', 'list', 'sets', 'sorted'];
if(!primaryDBName) {
if (!databaseName) {
winston.info('Database type not set! Run ./nodebb setup');
process.exit();
}
function setupSecondaryDB() {
var secondaryDB = require('./database/' + secondaryDBName);
secondaryModules = secondaryModules.split(/,\s*/);
for (var module in secondaryModules) {
if (secondaryModules.hasOwnProperty(module) && ALLOWED_MODULES.indexOf(module) !== -1) {
primaryDB[module] = secondaryDB[module];
}
}
var primaryDBinit = primaryDB.init,
primaryDBclose = primaryDB.close,
primaryDBhelpers = primaryDB.helpers;
primaryDB.init = function(callback) {
async.parallel([primaryDBinit, secondaryDB.init], function(err, results) {
callback(err);
});
};
primaryDB.close = function(callback) {
async.parallel([primaryDBclose, secondaryDB.close], function(err, results) {
callback(err);
});
};
primaryDB.helpers = {};
primaryDB.helpers[primaryDBName] = primaryDBhelpers[primaryDBName];
primaryDB.helpers[secondaryDBName] = secondaryDB.helpers[secondaryDBName];
}
var primaryDB = require('./database/' + primaryDBName);
if (secondaryDBName && secondaryModules) {
setupSecondaryDB();
}
var primaryDB = require('./database/' + databaseName);
module.exports = primaryDB;