X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fselection-strategies%2Ffair-share-worker-choice-strategy.ts;h=7b3ad3f7516f2b9522aca6acf3ec6b0356aabac3;hb=2c039e4373e86714cdf27e77440b12ee8eb2e4db;hp=4ea68d40cd127632de1fed3b94568aa531ebe8cb;hpb=4a59691c9f48b3a6602510aeece809e2a0fe14c1;p=poolifier.git diff --git a/src/pools/selection-strategies/fair-share-worker-choice-strategy.ts b/src/pools/selection-strategies/fair-share-worker-choice-strategy.ts index 4ea68d40..7b3ad3f7 100644 --- a/src/pools/selection-strategies/fair-share-worker-choice-strategy.ts +++ b/src/pools/selection-strategies/fair-share-worker-choice-strategy.ts @@ -2,10 +2,11 @@ import { DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS } from '../../utils' import type { IPool } from '../pool' import type { IWorker } from '../worker' import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy' -import type { - IWorkerChoiceStrategy, - TaskStatisticsRequirements, - WorkerChoiceStrategyOptions +import { + type IWorkerChoiceStrategy, + Measurements, + type TaskStatisticsRequirements, + type WorkerChoiceStrategyOptions } from './selection-strategies-types' /** @@ -35,7 +36,11 @@ export class FairShareWorkerChoiceStrategy< average: false, median: false }, - elu: false + elu: { + aggregate: true, + average: true, + median: false + } } /** @@ -67,7 +72,6 @@ export class FairShareWorkerChoiceStrategy< /** @inheritDoc */ public choose (): number { let minWorkerVirtualTaskEndTimestamp = Infinity - let chosenWorkerNodeKey!: number for (const [workerNodeKey] of this.pool.workerNodes.entries()) { if (this.workersVirtualTaskEndTimestamp[workerNodeKey] == null) { this.computeWorkerVirtualTaskEndTimestamp(workerNodeKey) @@ -76,10 +80,10 @@ export class FairShareWorkerChoiceStrategy< this.workersVirtualTaskEndTimestamp[workerNodeKey] if (workerVirtualTaskEndTimestamp < minWorkerVirtualTaskEndTimestamp) { minWorkerVirtualTaskEndTimestamp = workerVirtualTaskEndTimestamp - chosenWorkerNodeKey = workerNodeKey + this.nextWorkerNodeId = workerNodeKey } } - return chosenWorkerNodeKey + return this.nextWorkerNodeId } /** @inheritDoc */ @@ -105,9 +109,11 @@ export class FairShareWorkerChoiceStrategy< workerNodeKey: number, workerVirtualTaskStartTimestamp: number ): number { - return ( - workerVirtualTaskStartTimestamp + this.getWorkerTaskRunTime(workerNodeKey) - ) + const workerTaskRunTime = + this.opts.measurement === Measurements.elu + ? this.getWorkerTaskElu(workerNodeKey) + : this.getWorkerTaskRunTime(workerNodeKey) + return workerVirtualTaskStartTimestamp + workerTaskRunTime } private getWorkerVirtualTaskStartTimestamp (workerNodeKey: number): number {