Eric Mc SweenandCopybot
e3485f01da
Do not mock Mongo in unit tests GitOrigin-RevId: 7a200a4ddc8f91b14e96cf02cb4873c51fc3489a
58 lines
1.4 KiB
JavaScript
58 lines
1.4 KiB
JavaScript
const mongoose = require('mongoose')
|
|
const Settings = require('@overleaf/settings')
|
|
const Metrics = require('@overleaf/metrics')
|
|
const logger = require('@overleaf/logger')
|
|
const { addConnectionDrainer } = require('./GracefulShutdown')
|
|
|
|
mongoose.set('autoIndex', false)
|
|
mongoose.set('strictQuery', false)
|
|
|
|
const connectionPromise = mongoose.connect(
|
|
Settings.mongo.url,
|
|
Settings.mongo.options
|
|
)
|
|
|
|
connectionPromise
|
|
.then(mongooseInstance => {
|
|
Metrics.mongodb.monitor(mongooseInstance.connection.client)
|
|
})
|
|
.catch(error => {
|
|
logger.error(
|
|
{ error },
|
|
'Failed to connect to MongoDB - cannot set up monitoring'
|
|
)
|
|
})
|
|
|
|
addConnectionDrainer('mongoose', async () => {
|
|
await connectionPromise
|
|
await mongoose.disconnect()
|
|
})
|
|
|
|
mongoose.connection.on('connected', () =>
|
|
logger.debug('mongoose default connection open')
|
|
)
|
|
|
|
mongoose.connection.on('error', err =>
|
|
logger.err({ err }, 'mongoose error on default connection')
|
|
)
|
|
|
|
mongoose.connection.on('disconnected', () =>
|
|
logger.debug('mongoose default connection disconnected')
|
|
)
|
|
|
|
if (process.env.MONGOOSE_DEBUG) {
|
|
mongoose.set('debug', (collectionName, method, query, doc) =>
|
|
logger.debug({ collectionName, method, query, doc }, 'mongoose debug')
|
|
)
|
|
}
|
|
|
|
mongoose.plugin(schema => {
|
|
schema.options.usePushEach = true
|
|
})
|
|
|
|
mongoose.Promise = global.Promise
|
|
|
|
mongoose.connectionPromise = connectionPromise
|
|
|
|
module.exports = mongoose
|