feat: optimize worker choice strategies implementation
[poolifier.git] / src / pools / selection-strategies / least-used-worker-choice-strategy.ts
CommitLineData
2fc5cae3
JB
1import { DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS } from '../../utils'
2import type { IPool } from '../pool'
f06e48d8 3import type { IWorker } from '../worker'
bdaf31cd 4import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy'
2fc5cae3
JB
5import type {
6 IWorkerChoiceStrategy,
7 WorkerChoiceStrategyOptions
8} from './selection-strategies-types'
bdaf31cd
JB
9
10/**
e4543b14 11 * Selects the least used worker.
bdaf31cd 12 *
38e795c1 13 * @typeParam Worker - Type of worker which manages the strategy.
e102732c
JB
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.
bdaf31cd 16 */
e4543b14 17export class LeastUsedWorkerChoiceStrategy<
f06e48d8 18 Worker extends IWorker,
b2b1d84e
JB
19 Data = unknown,
20 Response = unknown
bf90656c
JB
21 >
22 extends AbstractWorkerChoiceStrategy<Worker, Data, Response>
17393ac8 23 implements IWorkerChoiceStrategy {
2fc5cae3
JB
24 /** @inheritDoc */
25 public constructor (
26 pool: IPool<Worker, Data, Response>,
27 opts: WorkerChoiceStrategyOptions = DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
28 ) {
29 super(pool, opts)
932fc8be 30 this.setTaskStatisticsRequirements(this.opts)
2fc5cae3
JB
31 }
32
afc003b2 33 /** @inheritDoc */
a6f7f1b4 34 public reset (): boolean {
ea7a90d3
JB
35 return true
36 }
37
138d29a8
JB
38 /** @inheritDoc */
39 public update (): boolean {
db703c75
JB
40 return true
41 }
42
43 /** @inheritDoc */
b1aae695 44 public choose (): number | undefined {
baca80f7 45 this.setPreviousWorkerNodeKey(this.nextWorkerNodeKey)
fce028d6 46 this.nextWorkerNodeKey = this.leastUsedNextWorkerNodeKey()
b1aae695 47 return this.nextWorkerNodeKey
9b106837
JB
48 }
49
50 /** @inheritDoc */
51 public remove (): boolean {
52 return true
53 }
54
b1aae695 55 private leastUsedNextWorkerNodeKey (): number | undefined {
f3a91bac
JB
56 return this.pool.workerNodes.reduce(
57 (minWorkerNodeKey, workerNode, workerNodeKey, workerNodes) => {
58 return workerNode.usage.tasks.executed +
59 workerNode.usage.tasks.executing +
60 workerNode.usage.tasks.queued <
61 workerNodes[minWorkerNodeKey].usage.tasks.executed +
62 workerNodes[minWorkerNodeKey].usage.tasks.executing +
63 workerNodes[minWorkerNodeKey].usage.tasks.queued
64 ? workerNodeKey
65 : minWorkerNodeKey
66 },
67 0
68 )
97a2abc3 69 }
bdaf31cd 70}