import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy'
import type {
IWorkerChoiceStrategy,
+ StrategyPolicy,
WorkerChoiceStrategyOptions
} from './selection-strategies-types'
>
extends AbstractWorkerChoiceStrategy<Worker, Data, Response>
implements IWorkerChoiceStrategy {
+ /** @inheritDoc */
+ public readonly strategyPolicy: StrategyPolicy = {
+ dynamicWorkerUsage: false,
+ dynamicWorkerReady: true
+ }
+
/** @inheritDoc */
public constructor (
pool: IPool<Worker, Data, Response>,
}
/** @inheritDoc */
- public choose (): number {
+ public choose (): number | undefined {
+ const chosenWorkerNodeKey = this.leastUsedNextWorkerNodeKey()
+ this.assignChosenWorkerNodeKey(chosenWorkerNodeKey)
+ return this.nextWorkerNodeKey
+ }
+
+ /** @inheritDoc */
+ public remove (): boolean {
+ return true
+ }
+
+ private leastUsedNextWorkerNodeKey (): number | undefined {
let minNumberOfTasks = Infinity
+ let chosenWorkerNodeKey: number | undefined
for (const [workerNodeKey, workerNode] of this.pool.workerNodes.entries()) {
const workerTaskStatistics = workerNode.usage.tasks
const workerTasks =
workerTaskStatistics.executed +
workerTaskStatistics.executing +
workerTaskStatistics.queued
- if (workerTasks === 0) {
- this.nextWorkerNodeId = workerNodeKey
+ if (this.isWorkerNodeEligible(workerNodeKey) && workerTasks === 0) {
+ chosenWorkerNodeKey = workerNodeKey
break
- } else if (workerTasks < minNumberOfTasks) {
+ } else if (
+ this.isWorkerNodeEligible(workerNodeKey) &&
+ workerTasks < minNumberOfTasks
+ ) {
minNumberOfTasks = workerTasks
- this.nextWorkerNodeId = workerNodeKey
+ chosenWorkerNodeKey = workerNodeKey
}
}
- return this.nextWorkerNodeId
- }
-
- /** @inheritDoc */
- public remove (): boolean {
- return true
+ return chosenWorkerNodeKey
}
}