X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fselection-strategies%2Fround-robin-worker-choice-strategy.ts;h=80e958a406fa2209ce1c20eded5b8e9b79e9ca0b;hb=56960396878dec81e3ebeea5b76387efca8cc2dc;hp=9cc966c47956a9a1689f82a435168127bbfbd272;hpb=afc003b20097712625ffd053e256ef5336e27b6e;p=poolifier.git diff --git a/src/pools/selection-strategies/round-robin-worker-choice-strategy.ts b/src/pools/selection-strategies/round-robin-worker-choice-strategy.ts index 9cc966c4..80e958a4 100644 --- a/src/pools/selection-strategies/round-robin-worker-choice-strategy.ts +++ b/src/pools/selection-strategies/round-robin-worker-choice-strategy.ts @@ -1,4 +1,4 @@ -import type { IPoolWorker } from '../pool-worker' +import type { IWorker } from '../worker' import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy' import type { IWorkerChoiceStrategy } from './selection-strategies-types' @@ -7,46 +7,46 @@ import type { IWorkerChoiceStrategy } from './selection-strategies-types' * * @typeParam Worker - Type of worker which manages the strategy. * @typeParam Data - Type of data sent to the worker. This can only be serializable data. - * @typeParam Response - Type of response of execution. This can only be serializable data. + * @typeParam Response - Type of execution response. This can only be serializable data. */ export class RoundRobinWorkerChoiceStrategy< - Worker extends IPoolWorker, + Worker extends IWorker, Data = unknown, Response = unknown > extends AbstractWorkerChoiceStrategy - implements IWorkerChoiceStrategy { + implements IWorkerChoiceStrategy { /** - * Id of the next worker. + * Id of the next worker node. */ - private nextWorkerId: number = 0 + private nextWorkerNodeId: number = 0 /** @inheritDoc */ public reset (): boolean { - this.nextWorkerId = 0 + this.nextWorkerNodeId = 0 return true } /** @inheritDoc */ public choose (): number { - const chosenWorkerKey = this.nextWorkerId - this.nextWorkerId = - this.nextWorkerId === this.pool.workers.length - 1 + const chosenWorkerNodeKey = this.nextWorkerNodeId + this.nextWorkerNodeId = + this.nextWorkerNodeId === this.pool.workerNodes.length - 1 ? 0 - : this.nextWorkerId + 1 - return chosenWorkerKey + : this.nextWorkerNodeId + 1 + return chosenWorkerNodeKey } /** @inheritDoc */ - public remove (workerKey: number): boolean { - if (this.nextWorkerId === workerKey) { - if (this.pool.workers.length === 0) { - this.nextWorkerId = 0 + public remove (workerNodeKey: number): boolean { + if (this.nextWorkerNodeId === workerNodeKey) { + if (this.pool.workerNodes.length === 0) { + this.nextWorkerNodeId = 0 } else { - this.nextWorkerId = - this.nextWorkerId > this.pool.workers.length - 1 - ? this.pool.workers.length - 1 - : this.nextWorkerId + this.nextWorkerNodeId = + this.nextWorkerNodeId > this.pool.workerNodes.length - 1 + ? this.pool.workerNodes.length - 1 + : this.nextWorkerNodeId } } return true