Mongoose Transaction wrapper

transactionWrapper.js

/*
this needs replicaSet to work
*/
const mongoose = require('mongoose');

async function runTransaction(transactionFunction, ...args) {
    let session = null;

    try {
        session = await mongoose.startSession();
        session.startTransaction();

        const result = await transactionFunction(...args);

        await session.commitTransaction();
        session.endSession();

        return result;
    } catch (err) {
        if (session) {
            await session.abortTransaction();
            session.endSession();
        }
        throw err;
    }
}