X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fselection-strategies%2Fweighted-round-robin-worker-choice-strategy.ts;h=4b1e5382af333d2edb59eb5ae41d1c5132f6ccb3;hb=b953022b2e57106ca7c52e641da5e03c2f9d5e07;hp=2b9cfc74536851ee39f0616d92932323c730031c;hpb=11df35903da6f581c45e8b42e1d4fbd342bddc3c;p=poolifier.git diff --git a/src/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.ts b/src/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.ts index 2b9cfc74..4b1e5382 100644 --- a/src/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.ts +++ b/src/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.ts @@ -67,33 +67,35 @@ export class WeightedRoundRobinWorkerChoiceStrategy< /** @inheritDoc */ public choose (): Worker { - let chosenWorker = this.pool.workers[this.currentWorkerIndex] + const chosenWorker = this.pool.workers[this.currentWorkerIndex] if ( this.isDynamicPool === true && this.workersTaskRunTime.has(chosenWorker) === false ) { this.initWorkerTaskRunTime(chosenWorker) } + const workerTaskRunTime = + this.workersTaskRunTime.get(chosenWorker)?.runTime ?? 0 const workerTaskWeight = this.workersTaskRunTime.get(chosenWorker)?.weight ?? this.defaultWorkerWeight - if ( - (this.workersTaskRunTime.get(chosenWorker)?.runTime ?? 0) < - workerTaskWeight - ) { + if (workerTaskRunTime < workerTaskWeight) { this.setWorkerTaskRunTime( chosenWorker, workerTaskWeight, - (this.workersTaskRunTime.get(chosenWorker)?.runTime ?? 0) + + workerTaskRunTime + (this.getWorkerVirtualTaskRunTime(chosenWorker) ?? 0) ) } else { this.currentWorkerIndex = - this.pool.workers.length - 1 === this.currentWorkerIndex + this.currentWorkerIndex === this.pool.workers.length - 1 ? 0 : this.currentWorkerIndex + 1 - chosenWorker = this.pool.workers[this.currentWorkerIndex] - this.setWorkerTaskRunTime(chosenWorker, workerTaskWeight, 0) + this.setWorkerTaskRunTime( + this.pool.workers[this.currentWorkerIndex], + workerTaskWeight, + 0 + ) } return chosenWorker } @@ -127,9 +129,9 @@ export class WeightedRoundRobinWorkerChoiceStrategy< let cpusCycleTimeWeight = 0 for (const cpu of cpus()) { // CPU estimated cycle time - const numberOfDigit = cpu.speed.toString().length - 1 - const cpuCycleTime = 1 / (cpu.speed / Math.pow(10, numberOfDigit)) - cpusCycleTimeWeight += cpuCycleTime * Math.pow(10, numberOfDigit) + const numberOfDigits = cpu.speed.toString().length - 1 + const cpuCycleTime = 1 / (cpu.speed / Math.pow(10, numberOfDigits)) + cpusCycleTimeWeight += cpuCycleTime * Math.pow(10, numberOfDigits) } return Math.round(cpusCycleTimeWeight / cpus().length) }