mirror of
https://github.com/HabitRPG/habitica.git
synced 2025-12-15 05:37:22 +01:00
15f104ddd0
* Add option to set a socket timeout for mongodb requests * Handle mongodb timeouts better * add default to config
31 lines
1.0 KiB
JavaScript
31 lines
1.0 KiB
JavaScript
import nconf from 'nconf';
|
|
import mongoose from 'mongoose';
|
|
import logger from './logger';
|
|
import {
|
|
getDevelopmentConnectionUrl,
|
|
getDefaultConnectionOptions,
|
|
} from './mongodb';
|
|
|
|
const IS_PROD = nconf.get('IS_PROD');
|
|
const MAINTENANCE_MODE = nconf.get('MAINTENANCE_MODE');
|
|
const POOL_SIZE = nconf.get('MONGODB_POOL_SIZE');
|
|
const SOCKET_TIMEOUT = nconf.get('MONGODB_SOCKET_TIMEOUT');
|
|
|
|
// Do not connect to MongoDB when in maintenance mode
|
|
if (MAINTENANCE_MODE !== 'true') {
|
|
const mongooseOptions = getDefaultConnectionOptions();
|
|
|
|
if (POOL_SIZE) mongooseOptions.maxPoolSize = Number(POOL_SIZE);
|
|
if (SOCKET_TIMEOUT) mongooseOptions.socketTimeoutMS = Number(SOCKET_TIMEOUT);
|
|
|
|
const DB_URI = nconf.get('IS_TEST') ? nconf.get('TEST_DB_URI') : nconf.get('NODE_DB_URI');
|
|
const connectionUrl = IS_PROD ? DB_URI : getDevelopmentConnectionUrl(DB_URI);
|
|
|
|
mongoose.connect(connectionUrl, mongooseOptions).then(() => {
|
|
logger.info('Connected with Mongoose.');
|
|
}).catch(err => {
|
|
logger.error(err);
|
|
throw err;
|
|
});
|
|
}
|