mirror of https://github.com/requarks/wiki
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.3 KiB
64 lines
1.3 KiB
8 years ago
|
"use strict";
|
||
|
|
|
||
|
|
const modb = require('mongoose'),
|
||
|
|
fs = require("fs"),
|
||
|
|
path = require("path"),
|
||
|
|
_ = require('lodash');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* MongoDB module
|
||
|
|
*
|
||
|
|
* @param {Object} appconfig Application config
|
||
|
|
* @return {Object} MongoDB wrapper instance
|
||
|
|
*/
|
||
|
|
module.exports = {
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Initialize DB
|
||
|
|
*
|
||
|
|
* @param {Object} appconfig The application config
|
||
|
|
* @return {Object} DB instance
|
||
|
|
*/
|
||
|
|
init(appconfig) {
|
||
|
|
|
||
|
|
let self = this;
|
||
|
|
|
||
|
8 years ago
|
let dbModelsPath = path.resolve(ROOTPATH, 'models');
|
||
|
8 years ago
|
|
||
|
|
modb.Promise = require('bluebird');
|
||
|
|
|
||
|
|
// Event handlers
|
||
|
|
|
||
|
|
modb.connection.on('error', (err) => {
|
||
|
|
winston.error('[' + PROCNAME + '] Failed to connect to MongoDB instance.');
|
||
|
|
});
|
||
|
|
modb.connection.once('open', function() {
|
||
|
|
winston.log('[' + PROCNAME + '] Connected to MongoDB instance.');
|
||
|
|
});
|
||
|
|
|
||
|
|
// Store connection handle
|
||
|
|
|
||
|
|
self.connection = modb.connection;
|
||
|
|
self.ObjectId = modb.Types.ObjectId;
|
||
|
|
|
||
|
|
// Load DB Models
|
||
|
|
|
||
|
|
fs
|
||
|
|
.readdirSync(dbModelsPath)
|
||
|
|
.filter(function(file) {
|
||
|
|
return (file.indexOf(".") !== 0);
|
||
|
|
})
|
||
|
|
.forEach(function(file) {
|
||
|
|
let modelName = _.upperFirst(_.camelCase(_.split(file,'.')[0]));
|
||
|
|
self[modelName] = require(path.join(dbModelsPath, file));
|
||
|
|
});
|
||
|
|
|
||
|
|
// Connect
|
||
|
|
|
||
|
|
self.onReady = modb.connect(appconfig.db);
|
||
|
|
|
||
|
|
return self;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
};
|