X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fselection-strategies%2Ffair-share-worker-choice-strategy.ts;h=2f2686e9bb0f5f6a6c18aa5ce51c913e9d3af7c5;hb=010f158c374ba75715130189cc1c53d8f6ca275f;hp=e8f9cc619f6a7f0d8ffc596332e27a9159c72e01;hpb=67f3f2d6cb8f915ec71f81c4533ab80a6c6a6f0f;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 e8f9cc61..2f2686e9 100644 --- a/src/pools/selection-strategies/fair-share-worker-choice-strategy.ts +++ b/src/pools/selection-strategies/fair-share-worker-choice-strategy.ts @@ -1,12 +1,11 @@ -import { DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS } from '../../utils.js' import type { IPool } from '../pool.js' import type { IWorker } from '../worker.js' import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy.js' import { type IWorkerChoiceStrategy, - type InternalWorkerChoiceStrategyOptions, Measurements, - type TaskStatisticsRequirements + type TaskStatisticsRequirements, + type WorkerChoiceStrategyOptions } from './selection-strategies-types.js' /** @@ -31,7 +30,11 @@ export class FairShareWorkerChoiceStrategy< average: true, median: false }, - waitTime: DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS, + waitTime: { + aggregate: true, + average: true, + median: false + }, elu: { aggregate: true, average: true, @@ -42,7 +45,7 @@ export class FairShareWorkerChoiceStrategy< /** @inheritDoc */ public constructor ( pool: IPool, - opts: InternalWorkerChoiceStrategyOptions + opts?: WorkerChoiceStrategyOptions ) { super(pool, opts) this.setTaskStatisticsRequirements(this.opts) @@ -78,6 +81,9 @@ export class FairShareWorkerChoiceStrategy< } private fairShareNextWorkerNodeKey (): number | undefined { + if (this.pool.workerNodes.length === 0) { + return undefined + } return this.pool.workerNodes.reduce( (minWorkerNodeKey, workerNode, workerNodeKey, workerNodes) => { if (workerNode.strategyData?.virtualTaskEndTimestamp == null) { @@ -117,11 +123,13 @@ export class FairShareWorkerChoiceStrategy< workerNodeKey: number, workerNodeVirtualTaskStartTimestamp: number ): number { - const workerNodeTaskRunTime = - this.opts.measurement === Measurements.elu + const workerNodeTaskExecutionTime = + this.getWorkerNodeTaskWaitTime(workerNodeKey) + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + (this.opts!.measurement === Measurements.elu ? this.getWorkerNodeTaskElu(workerNodeKey) - : this.getWorkerNodeTaskRunTime(workerNodeKey) - return workerNodeVirtualTaskStartTimestamp + workerNodeTaskRunTime + : this.getWorkerNodeTaskRunTime(workerNodeKey)) + return workerNodeVirtualTaskStartTimestamp + workerNodeTaskExecutionTime } private getWorkerNodeVirtualTaskStartTimestamp (