perf: improve node eligibility branching on worker choice strategies
[poolifier.git] / src / pools / selection-strategies / least-elu-worker-choice-strategy.ts
CommitLineData
3c93feb9
JB
1import {
2 DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
3 DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
4} from '../../utils'
058a9457
JB
5import type { IPool } from '../pool'
6import type { IWorker } from '../worker'
7import { AbstractWorkerChoiceStrategy } from './abstract-worker-choice-strategy'
8import type {
9 IWorkerChoiceStrategy,
05302647 10 TaskStatisticsRequirements,
058a9457
JB
11 WorkerChoiceStrategyOptions
12} from './selection-strategies-types'
13
14/**
15 * Selects the worker with the least ELU.
16 *
17 * @typeParam Worker - Type of worker which manages the strategy.
e102732c
JB
18 * @typeParam Data - Type of data sent to the worker. This can only be structured-cloneable data.
19 * @typeParam Response - Type of execution response. This can only be structured-cloneable data.
058a9457
JB
20 */
21export class LeastEluWorkerChoiceStrategy<
22 Worker extends IWorker,
23 Data = unknown,
24 Response = unknown
25 >
26 extends AbstractWorkerChoiceStrategy<Worker, Data, Response>
27 implements IWorkerChoiceStrategy {
28 /** @inheritDoc */
05302647 29 public readonly taskStatisticsRequirements: TaskStatisticsRequirements = {
3c93feb9
JB
30 runTime: DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
31 waitTime: DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
5df69fab
JB
32 elu: {
33 aggregate: true,
34 average: false,
35 median: false
36 }
058a9457
JB
37 }
38
39 /** @inheritDoc */
40 public constructor (
41 pool: IPool<Worker, Data, Response>,
42 opts: WorkerChoiceStrategyOptions = DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
43 ) {
44 super(pool, opts)
e460940e 45 this.setTaskStatisticsRequirements(this.opts)
058a9457
JB
46 }
47
48 /** @inheritDoc */
49 public reset (): boolean {
50 return true
51 }
52
53 /** @inheritDoc */
54 public update (): boolean {
db703c75
JB
55 return true
56 }
57
58 /** @inheritDoc */
b1aae695 59 public choose (): number | undefined {
fce028d6 60 this.nextWorkerNodeKey = this.leastEluNextWorkerNodeKey()
b1aae695 61 return this.nextWorkerNodeKey
9b106837
JB
62 }
63
64 /** @inheritDoc */
65 public remove (): boolean {
66 return true
67 }
68
b1aae695 69 private leastEluNextWorkerNodeKey (): number | undefined {
cdb517b3 70 let minWorkerElu = Infinity
b1aae695 71 let chosenWorkerNodeKey: number | undefined
058a9457 72 for (const [workerNodeKey, workerNode] of this.pool.workerNodes.entries()) {
9fe8fd69
JB
73 if (!this.isWorkerNodeEligible(workerNodeKey)) {
74 continue
75 }
465b2940 76 const workerUsage = workerNode.usage
98e72cda 77 const workerElu = workerUsage.elu?.active?.aggregate ?? 0
9fe8fd69 78 if (workerElu === 0) {
b1aae695 79 chosenWorkerNodeKey = workerNodeKey
5ea80606 80 break
9fe8fd69 81 } else if (workerElu < minWorkerElu) {
cdb517b3 82 minWorkerElu = workerElu
b1aae695 83 chosenWorkerNodeKey = workerNodeKey
058a9457
JB
84 }
85 }
b1aae695 86 return chosenWorkerNodeKey
058a9457
JB
87 }
88}