✨ feat(gvg): 农庄

2023-02-17 10:58:05 +08:00
parent 54837f24b6
commit b2b9404684
30 changed files with 1192 additions and 104 deletions
@@ -0,0 +1,126 @@
+import BaseModel from './BaseModel';
+import { index, getModelForClass, prop, DocumentType, } from '@typegoose/typegoose';
+import { getFutureTime, nowSeconds } from '../pubUtils/timeUtil';
+import { genCode } from '../pubUtils/util';
+import { GVG } from '../pubUtils/dicParam';
+
+@index({ leagueCode: 1, configId: 1 })
+@index({ fieldId: 1 })
+@index({ batchCode: 1 })
+export default class GVGLeagueFarm extends BaseModel {
+
+    @prop({ required: true })
+    leagueCode: string;          // 联军唯一code
+
+    @prop({ required: true })
+    configId: number;
+
+    @prop({ required: true })
+    farmId: number;              // 农庄id
+
+    @prop({ required: true })
+    type: number;              // 农庄类型
+
+    @prop({ required: true })
+    fieldId: number;              // 农庄id
+
+    @prop({ required: true })
+    unlockTime: number;          // 解锁时间，打开界面时预锁定一批填，种下去之后锁定时间延长到收获时间
+
+    @prop({ required: true })
+    lockRoleId: string;          // 锁定的人
+
+    @prop({ required: true })
+    harvestTime: number;         // 收获时间，种植下去后将时间更新，收获之后设为0
+
+    @prop({ required: true })
+    seedType: number;            // 实际种的种子类型
+
+    @prop({ required: true })
+    addType: number;             // 种子加成
+
+    @prop({ required: true })
+    batchCode: string;             // 批量号
+
+    public static async findByType(configId: number, leagueCode: string, type: number) {
+        const result: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ configId, leagueCode, type }).select('-_id').lean();
+        return result;
+    }
+
+    public static async findByFarmId(configId: number, leagueCode: string, farmId: number) {
+        const result: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ configId, leagueCode, farmId }).select('-_id').lean();
+        return result;
+    }
+
+    public static async findLockedByFarmId(configId: number, leagueCode: string, farmId: number) {
+        const result: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ configId, leagueCode, farmId, unlockTime: { $gte: nowSeconds() }}).select('-_id').lean();
+        return result;
+    }
+
+
+    public static async findPlantedByFarmId(configId: number, leagueCode: string, farmId: number) {
+        const result: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ configId, leagueCode, farmId, unlockTime: { $gte: nowSeconds() }, harvestTime: { $gte: nowSeconds() }}).select('-_id').lean();
+        return result;
+    }
+
+    /**
+     * 首次打开页面时锁定田
+     * @param configId 
+     * @param leagueCode 
+     * @param lands [{fieldId: number, time: number, addType: number}]  fieldId:田地id， time: 锁定时间 addType: 种子加成
+     * @returns 
+     */
+    public static async lockFields(configId: number, leagueCode: string, farmId: number, roleId: string, lands: { fieldId: number, addType: number }[]) {
+        // 先创建
+        await GVGLeagueFarmModel.bulkWrite(lands.map(({ fieldId }) => {
+            return { updateOne: { filter: { configId, leagueCode, farmId, fieldId }, update: { $setOnInsert: { unlockTime: 0, harvestTime: 0, seedType: 0, addType: 0 } }, upsert: true} }
+        }));
+        const result = await GVGLeagueFarmModel.bulkWrite(lands.map(({ fieldId, addType }) => {
+            return { updateOne: { filter: { configId, leagueCode, farmId, fieldId, unlockTime: { $lt: nowSeconds() } }, update: { $set: { addType, unlockTime: nowSeconds()  + GVG.GVG_FARM_LOCK_TIME, lockRoleId: roleId } } } }
+        }));
+        return result;
+    }
+
+    public static async findByFarmIdAndRoleId(configId: number, leagueCode: string, farmId: number, roleId: string) {
+        const result: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ configId, leagueCode, farmId, lockRoleId: roleId, unlockTime: { $gte: nowSeconds() } }).sort({ fieldId: 1 }).select('-_id -createdAt -updatedAt -__v -leagueCode -configId -roleId').lean();
+        return result;
+    }
+
+    public static async plant(configId: number, leagueCode: string, farmId: number, fields: { fieldId: number, seedType: number, time: number }[], roleId: string) {
+        const batchCode = genCode(10);
+        const result = await GVGLeagueFarmModel.bulkWrite(fields.map(({ fieldId, seedType, time }) => {
+            return { updateOne: { filter: { configId, leagueCode, farmId, fieldId, lockRoleId: roleId }, update: { $set: { seedType, unlockTime: getFutureTime(), harvestTime: nowSeconds() + time, batchCode }} } }
+        }));
+        if(result.modifiedCount == 0) return [];
+        const fieldResult: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ batchCode }).sort({ fieldId: 1 }).select('-_id -createdAt -updatedAt -__v -configId -leagueCode').lean();
+        return fieldResult;
+    }
+
+    public static async harvest(configId: number, leagueCode: string, farmId: number, fieldIds: number[], roleId: string) {
+        const batchCode = genCode(10);
+        const result = await GVGLeagueFarmModel.bulkWrite(fieldIds.map(fieldId => {
+            return { updateOne: { filter: { configId, leagueCode, farmId, fieldId, lockRoleId: roleId, harvestTime: { $lt: nowSeconds() } }, update: { $set: { unlockTime: 0, harvestTime: 0, lockRoleId: '', batchCode }} } }
+        }));
+        if(result.modifiedCount == 0) return [];
+        const fieldResult: GVGLeagueFarmType[] = await GVGLeagueFarmModel.find({ batchCode }).sort({ fieldId: 1 }).select('-_id -createdAt -updatedAt -__v -configId -leagueCode').lean();
+        return fieldResult;
+    }
+
+    public static async helpHarvest(configId: number, leagueCode: string, farmId: number, fieldId: number) {
+        const result: GVGLeagueFarmType = await GVGLeagueFarmModel.findOneAndUpdate(
+            { configId, leagueCode, farmId, fieldId, harvestTime: { $lt: nowSeconds() }, unlockTime: { $gt: nowSeconds() } },
+            { $set: { unlockTime: 0, harvestTime: 0, lockRoleId: '' }}, { new: true }).lean();
+        return result
+    }
+
+    public static async releaseLock(configId: number, leagueCode: string, farmId: number, roleId: string) {
+        await GVGLeagueFarmModel.updateMany({ configId, leagueCode, farmId, lockRoleId: roleId }, { $set: { unlockTime: 0, lockRoleId: '' } });
+    }
+}
+
+export const GVGLeagueFarmModel = getModelForClass(GVGLeagueFarm);
+
+export interface GVGLeagueFarmType extends Pick<DocumentType<GVGLeagueFarm>, keyof GVGLeagueFarm> {
+    id: number;
+};
+export type GVGLeagueFarmUpdate = Partial<GVGLeagueFarmType>; // 将所有字段变成可选项