130 lines
5.6 KiB
TypeScript
130 lines
5.6 KiB
TypeScript
import BaseModel from './BaseModel';
|
||
import { index, getModelForClass, prop, DocumentType, Ref } from '@typegoose/typegoose';
|
||
import { nowSeconds } from '../pubUtils/timeUtil';
|
||
class Reward {
|
||
@prop({ required: true })
|
||
id: number;
|
||
@prop({ required: true })
|
||
count: number;
|
||
}
|
||
|
||
class TrainBox {
|
||
@prop({ required: true })
|
||
roleId: string;
|
||
@prop({ required: true, type: Reward })
|
||
good: Reward;
|
||
@prop({ required: true })
|
||
index: number;
|
||
}
|
||
|
||
export class TrainInstance {
|
||
@prop({ required: true })
|
||
hid: number;
|
||
@prop({ required: true })
|
||
progress: number;
|
||
@prop({ required: true })
|
||
endTime: number;
|
||
@prop({ required: true, default: [], type: TrainBox, _id: false })
|
||
trainBoxs: TrainBox[];
|
||
}
|
||
|
||
class Rank {
|
||
@prop({ required: true })
|
||
roleId: string;
|
||
@prop({ required: true })
|
||
score: number;
|
||
}
|
||
|
||
class Report {
|
||
@prop({ required: true })
|
||
roleId: string;
|
||
@prop({ required: true })
|
||
trainId: number;//挑战的试炼关
|
||
@prop({ required: true })
|
||
hid: number;//据点
|
||
@prop({ required: true })
|
||
score: number;
|
||
@prop({ required: true })
|
||
time: number;
|
||
@prop({ required: true })
|
||
isSuccessed: boolean;
|
||
@prop({ required: true })
|
||
type: number;//1表示普通战报， 2表示系统战报即：被成功压制
|
||
}
|
||
@index({guildCode:1, trainId:1})
|
||
export default class GuildTrain extends BaseModel {
|
||
@prop({ required: true })
|
||
guildCode: string;
|
||
@prop({ required: true })
|
||
trainId: number;
|
||
@prop({ required: true })
|
||
isComplete: boolean;
|
||
@prop({ required: true, default: [], type: TrainInstance, _id: false })
|
||
trainInstances: TrainInstance[];
|
||
@prop({ required: true, default: [], type: Rank, _id: false })
|
||
ranks: Rank[];
|
||
@prop({ required: true, default: [], type: Report, _id: false })
|
||
reports: Report[];
|
||
@prop({ required: true, default: false})
|
||
locked: boolean;
|
||
|
||
public static async findGuildTrain(guildCode: string, select?: string, locked = false, lean = true) {
|
||
const guildTrains: GuildTrainType[] = await GuildTrainModel.find({ guildCode, locked }).select(select).lean(lean);
|
||
return guildTrains;
|
||
}
|
||
|
||
public static async findGuildTrainByTrainIds(guildCode: string, trainIds:Array<number>, locked = false, lean = true) {
|
||
const guildTrains: GuildTrainType[] = await GuildTrainModel.find({ trainId:{ $in:trainIds}, guildCode, locked }).lean(lean);
|
||
return guildTrains;
|
||
}
|
||
|
||
public static async findTrainByTrainIdNotLock(guildCode: string, trainId: number, select?: string, locked = false, lean = true) {
|
||
const guildTrain: GuildTrainType = await GuildTrainModel.findOne({ guildCode, trainId, locked }).select(select).lean(lean);
|
||
return guildTrain;
|
||
}
|
||
|
||
public static async updateGuildTrain(guildCode: string, trainId: number, update: GuildTrainTypeParam, lean = true) {
|
||
const guildTrain: GuildTrainType = await GuildTrainModel.findOneAndUpdate({ guildCode, trainId}, { $set: update }, {new: true}).lean(lean);
|
||
return guildTrain;
|
||
}
|
||
|
||
public static async updateGuildTrainProgress(guildCode: string, trainId: number, hid:number, progress: number, ranks:Array<Rank>, reports: Array<Report>, isComplete: boolean,lean = true) {
|
||
const guildTrain: GuildTrainType = await GuildTrainModel.findOneAndUpdate({ guildCode, trainId, 'trainInstances.hid':hid},
|
||
{ $set: {'trainInstances.$.progress': progress, ranks, reports, isComplete,'trainInstances.$.endTime': nowSeconds() + 24*60*60} },{new: true}).lean(lean);
|
||
return guildTrain;
|
||
}
|
||
|
||
public static async openGuildTrain(guildCode: string, trainId: number, trainInstances: Array<TrainInstance>, lean = true) {
|
||
const guildTrain: GuildTrainType = await GuildTrainModel.findOneAndUpdate({ guildCode, trainId },{ trainId, reports: [], locked: false, ranks: [], trainInstances, isComplete: false, guildCode}, {new: true, upsert: true}).lean(lean);
|
||
return guildTrain;
|
||
}
|
||
|
||
public static async findTrainInstanceBoxByIndex(guildCode: string, roleId: string, trainId: number, hid: number, index: number, progress: number, time: number, locked = false, lean = true) {
|
||
const guildTrain: GuildTrainType = await GuildTrainModel.findOne({ guildCode, trainId, locked, 'trainInstances.hid': hid,
|
||
'trainInstances.trainBoxs.index':{$ne:index },'trainInstances.trainBoxs.roleId':{$ne:roleId }, 'trainInstances.progress': {$gte: progress}, 'trainInstances.endTime':{$gte: time}}).lean(lean);
|
||
return guildTrain;
|
||
}
|
||
|
||
public static async receiveBoxByIndex(guildCode: string, roleId: string, trainId: number, hid: number, index: number, good: Reward, locked = false, lean = true) {
|
||
const guildTrain: GuildTrainType = await GuildTrainModel.findOneAndUpdate({ guildCode, trainId, locked, 'trainInstances.hid': hid, },
|
||
{ $push:{ 'trainInstances.$.trainBoxs': {roleId, index, good} }}, {new: true}).lean(lean);
|
||
return guildTrain;
|
||
}
|
||
|
||
public static async resetGuildTrain(guildCode: string) {
|
||
const result = await GuildTrainModel.updateMany({guildCode}, {$set:{locked: false}});
|
||
return result;
|
||
}
|
||
|
||
public static async getGuildTrainBoxs(guildCode: string) {
|
||
const result: GuildTrainType[] = await GuildTrainModel.find({guildCode, 'trainInstances.endTime':{$gte: nowSeconds()}, locked: false});
|
||
return result;
|
||
}
|
||
}
|
||
|
||
|
||
export const GuildTrainModel = getModelForClass(GuildTrain);
|
||
|
||
export interface GuildTrainType extends Pick<DocumentType<GuildTrain>, keyof GuildTrain> { };
|
||
export type GuildTrainTypeParam = Partial<GuildTrainType>; // 将所有字段变成可选项
|