UNPKG

@shahzaibalam231/mongoose-sequential-plugin

Version:
63 lines (61 loc) 1.68 kB
var __async = (__this, __arguments, generator) => { return new Promise((resolve, reject) => { var fulfilled = (value) => { try { step(generator.next(value)); } catch (e) { reject(e); } }; var rejected = (value) => { try { step(generator.throw(value)); } catch (e) { reject(e); } }; var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected); step((generator = generator.apply(__this, __arguments)).next()); }); }; // src/mongoose-plugin.ts import { Schema, model } from "mongoose"; var mongooseSequentialPlugin = (schema, options) => { const { modelName, field, startAt = 1, incrementBy = 1 } = options; if (!modelName || !field) { throw new Error( "modelName and field options are required for mongoose-sequential plugin." ); } const SequenceSchema = new Schema({ _id: String, sequenceValue: { type: Number, default: startAt } }); const SequenceModel = model( `${modelName}_sequence`, SequenceSchema ); schema.pre("save", function(next) { return __async(this, null, function* () { const doc = this; if (!doc[field]) { try { const sequenceDoc = yield SequenceModel.findByIdAndUpdate( modelName, { $inc: { sequenceValue: incrementBy || 1 } }, { new: true, upsert: true, setDefaultsOnInsert: true } ); doc[field] = sequenceDoc.sequenceValue; next(); } catch (err) { next(err); } } else { next(); } }); }); }; export { mongooseSequentialPlugin };