1 import { DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
} from
'../../utils'
2 import type { IPool
} from
'../pool'
3 import type { IWorker
} from
'../worker'
4 import { AbstractWorkerChoiceStrategy
} from
'./abstract-worker-choice-strategy'
7 WorkerChoiceStrategyOptions
8 } from
'./selection-strategies-types'
11 * Selects the next worker in a round robin fashion.
13 * @typeParam Worker - Type of worker which manages the strategy.
14 * @typeParam Data - Type of data sent to the worker. This can only be structured-cloneable data.
15 * @typeParam Response - Type of execution response. This can only be structured-cloneable data.
17 export class RoundRobinWorkerChoiceStrategy
<
18 Worker
extends IWorker
,
22 extends AbstractWorkerChoiceStrategy
<Worker
, Data
, Response
>
23 implements IWorkerChoiceStrategy
{
26 pool
: IPool
<Worker
, Data
, Response
>,
27 opts
: WorkerChoiceStrategyOptions
= DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
30 this.setTaskStatisticsRequirements(this.opts
)
34 public reset (): boolean {
35 this.resetWorkerNodeKeyProperties()
40 public update (): boolean {
45 public choose (): number | undefined {
46 const chosenWorkerNodeKey
= this.nextWorkerNodeKey
47 this.setPreviousWorkerNodeKey(chosenWorkerNodeKey
)
48 this.roundRobinNextWorkerNodeKey()
49 return chosenWorkerNodeKey
53 public remove (workerNodeKey
: number): boolean {
54 if (this.pool
.workerNodes
.length
=== 0) {
58 this.nextWorkerNodeKey
=== workerNodeKey
&&
59 this.nextWorkerNodeKey
> this.pool
.workerNodes
.length
- 1
61 this.nextWorkerNodeKey
= this.pool
.workerNodes
.length
- 1
64 this.previousWorkerNodeKey
=== workerNodeKey
&&
65 this.previousWorkerNodeKey
> this.pool
.workerNodes
.length
- 1
67 this.previousWorkerNodeKey
= this.pool
.workerNodes
.length
- 1
72 private roundRobinNextWorkerNodeKey (): number | undefined {
73 this.nextWorkerNodeKey
=
74 this.nextWorkerNodeKey
=== this.pool
.workerNodes
.length
- 1
76 : (this.nextWorkerNodeKey
?? this.previousWorkerNodeKey
) + 1
77 return this.nextWorkerNodeKey