90 lines
4.2 KiB
TypeScript
90 lines
4.2 KiB
TypeScript
import BaseModel from './BaseModel';
|
|
import { getModelForClass, prop, DocumentType, index } from '@typegoose/typegoose';
|
|
|
|
import { getZeroPoint, getZeroPointD } from '../pubUtils/timeUtil';
|
|
import { SHOP_REFRESH_TYPE } from '../consts';
|
|
|
|
@index({ roleId: 1, refTime: 1 })
|
|
@index({ refTime: 1, receiveNum: 1 })
|
|
export default class RougelikeScore extends BaseModel {
|
|
@prop({ required: true })
|
|
roleId: string;
|
|
|
|
@prop({ required: true, default: new Date() })
|
|
refTime: Date; // 每周刷新奖励
|
|
|
|
@prop({ required: true, default: 0 })
|
|
score: number; // 积分
|
|
|
|
@prop({ required: true, type: Number, default: [], _id: false })
|
|
received: number[]; // 已领取的id
|
|
|
|
@prop({ required: true, default: 0 })
|
|
receiveNum: number; // 已领取的id
|
|
|
|
@prop({ required: true, default: 0 })
|
|
takeoutRewardCnt: number; // 额外奖励已领取的次数（每日）
|
|
|
|
@prop({ required: true, default: () => new Date() })
|
|
refDaily: Date;
|
|
|
|
public static async findByRoleId(roleId: string) {
|
|
let refTime = getZeroPointD(SHOP_REFRESH_TYPE.WEEKLY);
|
|
let result: RougelikeScoreType = await RougelikeScoreModel.findOne({ roleId, refTime }).lean();
|
|
|
|
let today = getZeroPointD();
|
|
if (result && result.refDaily) {
|
|
let newResult = await RougelikeScoreModel.findOneAndUpdate({ roleId, refTime, refDaily: { $lt: today } }, { $set: { takeoutRewardCnt: 0, refDaily: new Date() }, $setOnInsert: { received: [], receiveNum: 0, score: 0 } }, { new: true }).lean();
|
|
if (newResult) return newResult;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
public static async incScore(roleId: string, score: number) {
|
|
let refTime = getZeroPointD(SHOP_REFRESH_TYPE.WEEKLY);
|
|
let result: RougelikeScoreType = await RougelikeScoreModel.findOneAndUpdate({ roleId, refTime }, { $inc: { score }, $setOnInsert: { received: [], receiveNum: 0, takeoutRewardCnt: 0, refDaily: new Date() } }, { new: true, upsert: true }).lean();
|
|
|
|
return result;
|
|
}
|
|
|
|
public static async receive(roleId: string, targetScore: number, index: number) {
|
|
let refTime = getZeroPointD(SHOP_REFRESH_TYPE.WEEKLY);
|
|
let result: RougelikeScoreType = await RougelikeScoreModel.findOneAndUpdate({ roleId, refTime, score: { $gte: targetScore } }, { $push: { received: index }, $inc: { receiveNum: 1 } }, { new: true }).lean();
|
|
return result;
|
|
}
|
|
|
|
public static async findByReceiveNum(refTime: Date, maxNum: number) {
|
|
let result: RougelikeScoreType[] = await RougelikeScoreModel.find({ refTime, receiveNum: { $lt: maxNum } }).lean();
|
|
return result;
|
|
}
|
|
|
|
public static async findByRefTime(refTime: Date) {
|
|
let result: RougelikeScoreType[] = await RougelikeScoreModel.find({ refTime }).lean();
|
|
return result;
|
|
}
|
|
|
|
public static async receiveAll(_ids: string[], maxNum: number) {
|
|
await RougelikeScoreModel.updateMany({ _id: { $in: _ids } }, { $set: { receiveNum: maxNum } });
|
|
}
|
|
|
|
public static async receiveTakeoutReward(roleId: string, maxNum: number, refDaily = new Date()) {
|
|
let refTime = getZeroPointD(SHOP_REFRESH_TYPE.WEEKLY);
|
|
// await RougelikeScoreModel.findOneAndUpdate({ roleId, refTime }, { $setOnInsert: { score: 0, received: [], receiveNum: 0, takeoutRewardCnt: 0, refDaily } }, { upsert: true });
|
|
await this.findByRoleId(roleId);
|
|
let result: RougelikeScoreType = await RougelikeScoreModel.findOneAndUpdate({ roleId, refTime, takeoutRewardCnt: { $lt: maxNum } }, { $set: { refDaily }, $inc: { takeoutRewardCnt: 1 } }, { new: true }).lean();
|
|
return result;
|
|
}
|
|
|
|
public static async bulkWriteUpdate(updateArr: RougelikeScorePara[]) {
|
|
if (updateArr.length == 0) return;
|
|
await RougelikeScoreModel.bulkWrite(updateArr.map((param) => {
|
|
delete param._id;
|
|
const { roleId, refTime } = param
|
|
return { updateOne: { filter: { roleId, refTime }, update: { $set: { ...param } }, upsert: true } }
|
|
}))
|
|
}
|
|
}
|
|
|
|
export const RougelikeScoreModel = getModelForClass(RougelikeScore);
|
|
export interface RougelikeScoreType extends Pick<DocumentType<RougelikeScore>, keyof RougelikeScore> { };
|
|
export type RougelikeScorePara = Partial<RougelikeScoreType>; // 将所有字段变成可选项
|