27 lines
849 B
TypeScript
27 lines
849 B
TypeScript
import BaseModel from './BaseModel';
|
|
import { index, getModelForClass, prop } from '@typegoose/typegoose';
|
|
|
|
/**
|
|
* 自增 ID
|
|
*/
|
|
@index({ name: 1 })
|
|
export default class Counter extends BaseModel {
|
|
|
|
@prop({ required: true })
|
|
name: string;
|
|
|
|
@prop({ required: true, default: 1 })
|
|
seq: number;
|
|
|
|
public static async getNewCounter(param:{name: string, def: number}, lean = true) {
|
|
let {name, def:defaultVal} = param;
|
|
let counter = await CounterModel.findOneAndUpdate({ name }, { $inc: { seq: 1 } }, { new: true, upsert: true }).lean(lean);
|
|
if(!counter || (counter&&counter.seq == 1) && defaultVal != 1) {
|
|
counter = await CounterModel.findOneAndUpdate({ name }, { $set: { seq: defaultVal } }, { new: true, upsert: true }).lean(lean);
|
|
}
|
|
return counter?.seq;
|
|
}
|
|
|
|
}
|
|
|
|
export const CounterModel = getModelForClass(Counter); |