89 lines
3.2 KiB
TypeScript
89 lines
3.2 KiB
TypeScript
import BaseModel from './BaseModel';
|
||
import { index, getModelForClass, prop } from '@typegoose/typegoose';
|
||
|
||
class TowerTask {
|
||
@prop({ required: true })
|
||
id: number; // 任务唯一 Id，来自任务表
|
||
@prop({ required: true})
|
||
taskCode: string; // 服务器生成的任务唯一编号
|
||
@prop({ required: true, type: Number, default: []})
|
||
heroes: Array<number>; // 派遣武将 hid
|
||
@prop({ required: true })
|
||
status: boolean; // 派遣任务当前状态，0-可派遣，1-已派遣，2-已完成，3-已领取
|
||
}
|
||
|
||
function genCodeTmp(len) {
|
||
const chars = '123456789ABCDEFGHJKLMNPQRSTWXYZabcdefghijklmnopqrstuvwxyz';
|
||
const charArr = chars.split('');
|
||
let code = '';
|
||
for (let i = 0; i < len; i++) {
|
||
code += charArr[Math.floor(Math.random() * charArr.length)];
|
||
}
|
||
return code;
|
||
}
|
||
|
||
/**
|
||
* 天梯派遣记录表
|
||
*/
|
||
@index({ roleId: 1, status: 1 })
|
||
@index({ batchCode: 1 })
|
||
export default class TowerTaskRec extends BaseModel {
|
||
@prop({ required: true })
|
||
roleId: string; // 角色 id
|
||
@prop({ required: true })
|
||
roleName: string; // 角色名
|
||
|
||
@prop({ required: true})
|
||
batchCode: string; // 本批派遣任务唯一标识
|
||
@prop({ required: true, default: 0})
|
||
status: number; // 本批任务是否完成，0-未完成，1-已完成，2-已领取
|
||
@prop({ required: true, type: TowerTask, default: [] })
|
||
tasks: Array<TowerTask>; // 本批次任务列表
|
||
@prop({ required: true, type: Number, default: [] })
|
||
heroes: Array<number>; // 此批派遣使用的全部武将
|
||
|
||
public static async getCurTask(roleId: string, lean = true) {
|
||
const curTime = new Date();
|
||
curTime.setHours(5, 0, 0, 0);
|
||
const rec = await TowerTaskRecModel.findOne({roleId, createdAt: {$gte: curTime}}).lean(lean);
|
||
return rec;
|
||
}
|
||
|
||
public static async createTask(roleId: string, roleName: string, taskIds: Array<number>, lean = true) {
|
||
const batchCode = genCodeTmp(8);
|
||
const tasksData = taskIds.map(id => {
|
||
const taskCode = genCodeTmp(6);
|
||
const taskData = Object.assign(new TowerTask(), {id, taskCode});
|
||
return taskData;
|
||
});
|
||
const recDoc = new TowerTaskRecModel();
|
||
const update = Object.assign(recDoc.toJSON(), {tasks: tasksData, roleId, roleName});
|
||
const rec = await TowerTaskRecModel.findOneAndUpdate({roleId, batchCode}, update, {upsert: true, new: true}).lean(lean);
|
||
return rec;
|
||
}
|
||
|
||
public static async updateTask(roleId: string, batchCode: string, tasks: Array<{taskCode:string, heroes: Array<number>}>, lean = true) {
|
||
let allHeroes = [];
|
||
tasks.forEach(task => {
|
||
allHeroes.concat(task.heroes);
|
||
})
|
||
const curRec = await TowerTaskRecModel.findOne({batchCode, heroes: {$nin: allHeroes}}).lean(lean);
|
||
if (!curRec) return null;
|
||
|
||
let {status, tasks: recTasks, heroes: recHeroes} = curRec;
|
||
if (status == 2 || status == 1) return curRec;
|
||
for (let task of tasks) {
|
||
for (let recTask of recTasks) {
|
||
if (task.taskCode == recTask.taskCode) {
|
||
recTask.heroes = task.heroes;
|
||
recHeroes.concat(task.heroes);
|
||
}
|
||
}
|
||
}
|
||
const newRec = await TowerTaskRecModel.findOneAndUpdate({batchCode}, {heroes: recHeroes, tasks: recTasks}, {new: true}).lean(lean);
|
||
return newRec;
|
||
}
|
||
}
|
||
|
||
export const TowerTaskRecModel = getModelForClass(TowerTaskRec);
|