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'
8 WorkerChoiceStrategyOptions
9 } from
'./selection-strategies-types'
12 * Selects the next worker in a round robin fashion.
14 * @typeParam Worker - Type of worker which manages the strategy.
15 * @typeParam Data - Type of data sent to the worker. This can only be structured-cloneable data.
16 * @typeParam Response - Type of execution response. This can only be structured-cloneable data.
18 export class RoundRobinWorkerChoiceStrategy
<
19 Worker
extends IWorker
,
23 extends AbstractWorkerChoiceStrategy
<Worker
, Data
, Response
>
24 implements IWorkerChoiceStrategy
{
26 public readonly strategyPolicy
: StrategyPolicy
= {
27 useDynamicWorker
: true
32 pool
: IPool
<Worker
, Data
, Response
>,
33 opts
: WorkerChoiceStrategyOptions
= DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
36 this.setTaskStatisticsRequirements(this.opts
)
40 public reset (): boolean {
41 this.nextWorkerNodeId
= 0
46 public update (): boolean {
51 public choose (): number {
52 const chosenWorkerNodeKey
= this.nextWorkerNodeId
53 this.nextWorkerNodeId
=
54 this.nextWorkerNodeId
=== this.pool
.workerNodes
.length
- 1
56 : this.nextWorkerNodeId
+ 1
57 return chosenWorkerNodeKey
61 public remove (workerNodeKey
: number): boolean {
62 if (this.nextWorkerNodeId
=== workerNodeKey
) {
63 if (this.pool
.workerNodes
.length
=== 0) {
64 this.nextWorkerNodeId
= 0
65 } else if (this.nextWorkerNodeId
> this.pool
.workerNodes
.length
- 1) {
66 this.nextWorkerNodeId
= this.pool
.workerNodes
.length
- 1