import type { IPool } from '../pool'
import type { IWorker } from '../worker'
import { FairShareWorkerChoiceStrategy } from './fair-share-worker-choice-strategy'
+import { InterleavedWeightedRoundRobinWorkerChoiceStrategy } from './interleaved-weighted-round-robin-worker-choice-strategy'
import { LessBusyWorkerChoiceStrategy } from './less-busy-worker-choice-strategy'
import { LessUsedWorkerChoiceStrategy } from './less-used-worker-choice-strategy'
import { RoundRobinWorkerChoiceStrategy } from './round-robin-worker-choice-strategy'
Data,
Response
>(pool, opts)
+ ],
+ [
+ WorkerChoiceStrategies.INTERLEAVED_WEIGHTED_ROUND_ROBIN,
+ new (InterleavedWeightedRoundRobinWorkerChoiceStrategy.bind(this))<
+ Worker,
+ Data,
+ Response
+ >(pool, opts)
]
])
}
}
/**
- * Updates the worker choice strategy internals in the context.
+ * Updates the worker node key in the worker choice strategy internals in the context.
*
* @returns `true` if the update is successful, `false` otherwise.
*/
- public update (): boolean {
+ public update (workerNodeKey: number): boolean {
return (
this.workerChoiceStrategies.get(
this.workerChoiceStrategy
) as IWorkerChoiceStrategy
- ).update()
+ ).update(workerNodeKey)
}
/**
* Executes the worker choice strategy algorithm in the context.
*
* @returns The key of the worker node.
+ * @throws {@link Error} If the worker node key is null or undefined.
*/
public execute (): number {
- return (
+ const workerNodeKey = (
this.workerChoiceStrategies.get(
this.workerChoiceStrategy
) as IWorkerChoiceStrategy
).choose()
+ if (workerNodeKey == null) {
+ throw new Error('Worker node key chosen is null or undefined')
+ }
+ return workerNodeKey
}
/**
- * Removes a worker node key from the worker choice strategy in the context.
+ * Removes the worker node key from the worker choice strategy in the context.
*
* @param workerNodeKey - The key of the worker node.
* @returns `true` if the removal is successful, `false` otherwise.