kellner/lib/protocols/Mongoose.js

'use strict';

const mongoose = require('mongoose');

const getMongoUri = ({
	url, servers, host, port, username, password, authSource,
}) => {
	if (url)
		return url;

	const member = (servers || [{ host, port }]).map(m => `${m.host}:${m.port}`);
	const auth = (username && password) ? `${username}:${password}` : '';
	const options = authSource ? `?authSource=${authSource}` : '';

	return `mongodb://${auth}${member}/${options}`;
};

let connection = null;

module.exports = {
	type: 'mongoose',
	connect: async ({ mongodb }, dbName) => {
		if (!dbName)
			throw new Error('Mongoose connection requiers dbName');

		if (!connection) {
			mongoose.set('useNewUrlParser', true);
			mongoose.set('useFindAndModify', false);
			mongoose.set('useCreateIndex', true);
			mongoose.connect(getMongoUri(mongodb), { dbName });
			connection = mongoose.connection; // eslint-disable-line
			return connection;
		}

		return connection.useDb(dbName);
	},
	async close() {
		if (!connection)
			return;
		connection = null;
		const closingConnection = connection;
		await closingConnection.close();
	},
};