71 lines
1.4 KiB
JavaScript
71 lines
1.4 KiB
JavaScript
|
'use strict'
|
||
|
|
|
||
|
|
/* global ROOTPATH, appconfig, winston */
|
||
|
|
|
||
|
|
const modb = require('mongoose')
|
||
|
|
const fs = require('fs')
|
||
|
|
const path = require('path')
|
||
|
|
const _ = require('lodash')
|
||
|
|
|
||
|
|
/**
|
||
|
|
* MongoDB module
|
||
|
|
*
|
||
|
|
* @return {Object} MongoDB wrapper instance
|
||
|
|
*/
|
||
|
|
module.exports = {
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Initialize DB
|
||
|
|
*
|
||
|
|
* @return {Object} DB instance
|
||
|
|
*/
|
||
|
|
init () {
|
||
|
|
let self = this
|
||
|
|
global.Mongoose = modb
|
||
|
|
|
||
|
|
let dbModelsPath = path.resolve(ROOTPATH, 'models')
|
||
|
|
|
||
|
|
modb.Promise = require('bluebird')
|
||
|
|
|
||
|
|
// Event handlers
|
||
|
|
|
||
|
|
modb.connection.on('error', err => {
|
||
|
|
winston.error('Failed to connect to MongoDB instance.')
|
||
|
|
return err
|
||
|
|
})
|
||
|
|
modb.connection.once('open', function () {
|
||
|
|
winston.log('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))
|
||
|
|
})
|
||
|
|
|
||
|
|
// Using ENV variable?
|
||
|
|
|
||
|
|
if (_.startsWith(appconfig.db, '$')) {
|
||
|
|
appconfig.db = process.env[appconfig.db.slice(1)]
|
||
|
|
}
|
||
|
|
|
||
|
|
// Connect
|
||
|
|
|
||
|
|
self.onReady = modb.connect(appconfig.db)
|
||
|
|
|
||
|
|
return self
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|