mirror of
https://github.com/NodeBB/NodeBB.git
synced 2025-11-07 22:45:46 +01:00
closes #3938
This commit is contained in:
@@ -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;
|
||||
Reference in New Issue
Block a user