1 import { DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
} from
'../../utils'
2 import type { IPool
} from
'../pool'
3 import type { IWorker
} from
'../worker'
7 WorkerChoiceStrategyOptions
8 } from
'./selection-strategies-types'
11 * Worker choice strategy abstract base class.
13 * @typeParam Worker - Type of worker which manages the strategy.
14 * @typeParam Data - Type of data sent to the worker. This can only be serializable data.
15 * @typeParam Response - Type of execution response. This can only be serializable data.
17 export abstract class AbstractWorkerChoiceStrategy
<
18 Worker
extends IWorker
,
21 > implements IWorkerChoiceStrategy
{
23 * Toggles finding the last free worker node key.
25 private toggleFindLastFreeWorkerNodeKey
: boolean = false
27 public readonly requiredStatistics
: RequiredStatistics
= {
34 * Constructs a worker choice strategy bound to the pool.
36 * @param pool - The pool instance.
37 * @param opts - The worker choice strategy options.
40 protected readonly pool
: IPool
<Worker
, Data
, Response
>,
41 protected opts
: WorkerChoiceStrategyOptions
= DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
43 this.choose
= this.choose
.bind(this)
46 protected checkOptions (opts
: WorkerChoiceStrategyOptions
): void {
47 if (this.requiredStatistics
.avgRunTime
&& opts
.medRunTime
=== true) {
48 this.requiredStatistics
.avgRunTime
= false
49 this.requiredStatistics
.medRunTime
= opts
.medRunTime
as boolean
51 if (this.requiredStatistics
.medRunTime
&& opts
.medRunTime
=== false) {
52 this.requiredStatistics
.avgRunTime
= true
53 this.requiredStatistics
.medRunTime
= opts
.medRunTime
as boolean
56 opts
.weights
!= null &&
57 Object.keys(opts
.weights
).length
< this.pool
.size
60 'Worker choice strategy options must have a weight for each worker node.'
66 public abstract reset (): boolean
69 public abstract choose (): number
72 public abstract remove (workerNodeKey
: number): boolean
75 public setOptions (opts
: WorkerChoiceStrategyOptions
): void {
76 opts
= opts
?? DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
77 this.checkOptions(opts
)
82 * Finds a free worker node key.
84 * @returns The free worker node key or `-1` if there is no free worker node.
86 protected findFreeWorkerNodeKey (): number {
87 if (this.toggleFindLastFreeWorkerNodeKey
) {
88 this.toggleFindLastFreeWorkerNodeKey
= false
89 return this.findLastFreeWorkerNodeKey()
91 this.toggleFindLastFreeWorkerNodeKey
= true
92 return this.findFirstFreeWorkerNodeKey()
96 * Finds the first free worker node key based on the number of tasks the worker has applied.
98 * If a worker is found with `0` running tasks, it is detected as free and its worker node key is returned.
100 * If no free worker is found, `-1` is returned.
102 * @returns A worker node key if there is one, `-1` otherwise.
104 private findFirstFreeWorkerNodeKey (): number {
105 return this.pool
.workerNodes
.findIndex(workerNode
=> {
106 return workerNode
.tasksUsage
?.running
=== 0
111 * Finds the last free worker node key based on the number of tasks the worker has applied.
113 * If a worker is found with `0` running tasks, it is detected as free and its worker node key is returned.
115 * If no free worker is found, `-1` is returned.
117 * @returns A worker node key if there is one, `-1` otherwise.
119 private findLastFreeWorkerNodeKey (): number {
120 // It requires node >= 18.0.0:
121 // return this.workerNodes.findLastIndex(workerNode => {
122 // return workerNode.tasksUsage?.running === 0
124 for (let i
= this.pool
.workerNodes
.length
- 1; i
>= 0; i
--) {
125 if (this.pool
.workerNodes
[i
].tasksUsage
?.running
=== 0) {