✨ feat(gvg): 农庄

2023-01-12 20:50:36 +08:00
parent 54837f24b6
commit b2b9404684
30 changed files with 1192 additions and 104 deletions
--- a/shared/db/GVGLeagueFarm.ts
+++ b/shared/db/GVGLeagueFarm.ts
@@ -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>; // 将所有字段变成可选项
--- a/shared/db/GVGLeaguePrepare.ts
+++ b/shared/db/GVGLeaguePrepare.ts
@@ -109,19 +109,36 @@ export default class GVGLeaguePrepare extends BaseModel {

    public static async unlockTech(configId: number, leagueCode: string, techId: number, maxProgress: number) {
        const result: GVGLeaguePrepareType = await GVGLeaguePrepareModel.findOneAndUpdate({ 
-            configId, leagueCode, techQueueCnt: { $lt: GVG.GVG_LEAGUE_TECH_LIST } }, { $push: { techQueue: new Tech(techId, maxProgress) }, $inc: { techQueueCnt: 1 } }, { new: true }).lean();
+            configId, leagueCode, techQueueCnt: { $lt: GVG.GVG_LEAGUE_TECH_LIST } 
+        }, { $push: { techQueue: new Tech(techId, maxProgress) }, $inc: { techQueueCnt: 1 } }, { new: true }).lean();
        return result;
    }

    public static async addProgress(configId: number, leagueCode: string, techId: number, addProgress: number) {
        const result: GVGLeaguePrepareType = await GVGLeaguePrepareModel.findOneAndUpdate({ 
-            configId, leagueCode, 'techQueue.id': techId }, { $inc: { 'techQueue.$.progress': addProgress } }, { new: true }).lean();
+            configId, leagueCode, 'techQueue.id': techId 
+        }, { $inc: { 'techQueue.$.progress': addProgress } }, { new: true }).lean();
        return result;
    }

    public static async activate(configId: number, leagueCode: string, techId: number) {
        const result: GVGLeaguePrepareType = await GVGLeaguePrepareModel.findOneAndUpdate({ 
-            configId, leagueCode }, { $push: { activeTech: techId},  $pull: { techQueue: { id: techId } } }, { new: true }).lean();
+            configId, leagueCode 
+        }, { $push: { activeTech: techId},  $pull: { techQueue: { id: techId } }, $inc: { techQueueCnt: -1 } }, { new: true }).lean();
+        return result;
+    }
+
+    public static async addResource(configId: number, leagueCode: string, food: number, mineral: number, wood: number) {
+        const result: GVGLeaguePrepareType = await GVGLeaguePrepareModel.findOneAndUpdate({ 
+            configId, leagueCode
+        }, { 
+            $inc: { 'resources.food': food, 'resources.mineral': mineral, 'resources.wood': wood }
+        }, { new: true }).lean();
+        return result;
+    }
+
+    public static async setLv(configId: number, leagueCode: string, lv: number) {
+        const result: GVGLeaguePrepareType = await GVGLeaguePrepareModel.findOneAndUpdate({ configId, leagueCode }, { $set: { lv }}, { new: true }).lean();
        return result;
    }
 }
--- a/shared/db/GVGUserDailyData.ts
+++ b/shared/db/GVGUserDailyData.ts
@@ -61,6 +61,11 @@ export default class GVGUserDailyData extends BaseModel {
        const result: GVGUserDailyDataType = await GVGUserDailyDataModel.findOneAndUpdate({ configId, leagueCode, roleId, time: getZeroPoint() }, { hasLogin: true }, { upsert: true, new: true }).lean();
        return result;
    }
+
+    public static async addResource(configId: number, leagueCode: string, roleId: string, food: number, mineral: number, wood: number) {
+        const result: GVGUserDailyDataType = await GVGUserDailyDataModel.findOneAndUpdate({ configId, leagueCode, roleId, time: getZeroPoint()}, { $inc: { food, mineral, wood }}, { new: true, upsert: true }).lean();
+        return result;
+    }
 }

 export const GVGUserDailyDataModel = getModelForClass(GVGUserDailyData);
--- a/shared/db/GVGUserData.ts
+++ b/shared/db/GVGUserData.ts
@@ -103,6 +103,15 @@ export default class GVGUserData extends BaseModel {
        const result: GVGUserDataType = await GVGUserDataModel.findOneAndUpdate({ configId, leagueCode, roleId }, { $inc: { active }, $push: { activeRec: new ActiveRec(activeType, active) } }, { new: true, upsert: true}).select('active').lean();
        return result;
    }
+
+    public static async addResource(configId: number, leagueCode: string, roleId: string, food: number, mineral: number, wood: number) {
+        const result: GVGUserDataType = await GVGUserDataModel.findOneAndUpdate({ 
+            configId, leagueCode, roleId
+        }, { 
+            $inc: { 'distribute.food': food, 'distribute.mineral': mineral, 'distribute.wood': wood }
+        }, { new: true, upsert: true }).lean();
+        return result;
+    }
 }

 export const GVGUserDataModel = getModelForClass(GVGUserData);
--- a/shared/db/GVGUserItem.ts
+++ b/shared/db/GVGUserItem.ts
@@ -48,6 +48,11 @@ export default class GVGUserItem extends BaseModel {
        console.log(result.matchedCount, items.length)
        return result.matchedCount == items.length;
    }
+
+    public static async findByItem(configId: number, leagueCode: string, roleId: string, id: number) {
+        const result: GVGUserItemType = await GVGUserItemModel.findOne({ configId, leagueCode, roleId, id }).select('count').lean();
+        return result;
+    }
 }

 export const GVGUserItemModel = getModelForClass(GVGUserItem);