@shahzaibalam231/mongoose-sequential-plugin
Version:
mongoose auto increment ~
63 lines (61 loc) • 1.68 kB
JavaScript
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
};