av/travian
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix train troop recurrent tasks

Browse Source
This commit is contained in:
Anton Vakhrushev 2020-10-12 13:52:02 +03:00
parent 7f3642d626
commit 33964dc99b
3 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/Handler/Action/TrainTrooperAction.ts
View File

@ -30,13 +30,20 @@ export class TrainTrooperAction extends BaseAction {
}
if (nextToTrainCount > 0) {
this.scheduler.scheduleTask(TrainTroopTask.name, {
...task.args,
count: nextToTrainCount,
});
this.scheduleNextTask(task, nextToTrainCount);
}
fillTrainCount(troopId, readyToTrainCount);
clickTrainButton();
}
private scheduleNextTask(prevTask: Task, nextToTrainCount: number) {
const villageId = prevTask.args.villageId || taskError('Empty village id');
const taskName = prevTask.name;
const taskCollection = this.villageFactory.getById(villageId).taskCollection();
taskCollection.addTaskAsFirst(taskName, {
...prevTask.args,
count: nextToTrainCount,
});
}
}

6
src/Scheduler.ts
View File

@ -119,8 +119,10 @@ export class Scheduler {
scheduleTask(name: string, args: Args, ts?: number | undefined): void {
if (isProductionTask(name) && args.villageId) {
const controller = this.villageControllerFactory.getById(args.villageId).controller();
controller.addTask(name, args);
const taskCollection = this.villageControllerFactory
.getById(args.villageId)
.taskCollection();
taskCollection.addTask(name, args);
} else {
this.logger.info('Schedule task', name, args, ts);
this.taskQueue.add(new Task(uniqTaskId(), ts || timestamp(), name, args));

19
src/Village/VillageState.ts
View File

@ -145,8 +145,8 @@ function createProductionQueueState(
const taskEndingTimestamp = taskQueueState.finishTs;
const resources = warehouseState.resources;
const performance = warehouseState.performance;
const firstTaskResources = tasks.slice(0, 1).reduce(taskResourceReducer, Resources.zero());
const allTaskResources = tasks.reduce(taskResourceReducer, Resources.zero());
const firstTaskResources = getTotalTaskCollectionResources(tasks.slice(0, 1));
const allTaskResources = getTotalTaskCollectionResources(tasks);
const currentTimestamp = timestamp();
@ -220,10 +220,14 @@ function getTotalTaskResources(task: TaskCore | undefined): Resources {
return Resources.zero();
}
function taskResourceReducer(resources: Resources, task: TaskCore) {
return task.args.resources
? resources.add(Resources.fromObject(task.args.resources))
: resources;
function getTotalTaskCollectionResources(tasks: ReadonlyArray<TaskState>): Resources {
const reducer = (memo: Resources, task: TaskState): Resources => {
const count = task.args.count || 1;
return task.args.resources
? memo.add(Resources.fromObject(task.args.resources).scale(count))
: memo;
};
return tasks.reduce(reducer, Resources.zero());
}
function makeTaskState(task: TaskCore, maxResourcesForTask: Resources): TaskState {
@ -234,7 +238,6 @@ function makeTaskState(task: TaskCore, maxResourcesForTask: Resources): TaskStat
const isEnoughGranaryCapacity = maxResourcesForTask.allGreaterOrEqual(
new Resources(0, 0, 0, taskResources.crop)
);
const canBeBuilt = isEnoughWarehouseCapacity && isEnoughGranaryCapacity;
return {
@ -253,7 +256,7 @@ function createVillageState(village: Village, storage: VillageStorage): VillageS
const capacity = storage.getWarehouseCapacity();
const performance = storage.getResourcesPerformance();
const warehouse = makeWarehouseState(resources, capacity, performance);
const tasks = storage.getTasks().map((t) => makeTaskState(t, warehouse.optimumFullness));
const tasks = storage.getTasks().map((task) => makeTaskState(task, warehouse.optimumFullness));
const queues = createTaskQueueStates(warehouse, tasks, storage);
const firstReadyTask = getReadyForProductionTask(queues);
const firstTaskResources = getTotalTaskResources(firstReadyTask);