Mongo db utils

mongo-utils.js

const MongoClient = require('mongodb').MongoClient;
const assert = require('assert');

exports.findInCollection = ({ db, collectionName }) => query => {
	assert.notEqual(db, undefined);

	const collection = db.collection(collectionName);
	return collection.find(query).toArray();
};

exports.insertManyInCollection = ({ db, collectionName }) => (items = []) => {
	assert.notEqual(db, undefined);

	if (!items.length) {
		return;
	}

	const collection = db.collection(collectionName);
	return collection.insertMany(items).then(result => {
		const { insertedCount, insertedIds } = result;
		return { insertedCount, insertedIds };
	});
};

exports.deleteManyInCollection = ({ db, collectionName }) => query => {
	assert.notEqual(db, undefined);

	const collection = db.collection(collectionName);
	return collection.deleteMany(query).then(result => {
		const { deletedCount } = result;
		return { deletedCount };
	});
};

exports.connectToMongo = ({ mongoUrl, options = {} }) => {
	return MongoClient.connect(mongoUrl, options).then(client => {
		const db = client.db(); // Use db from mongoUrl
		const dbCleanup = () => client.close();
		return { db, dbCleanup };
	});
};