src/pools/selection-strategies/abstract-worker-choice-strategy.ts

   1 import {
   2   DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
   3   buildWorkerChoiceStrategyOptions
   4 } from '../../utils.js'
   5 import type { IPool } from '../pool.js'
   6 import type { IWorker } from '../worker.js'
   7 import type {
   8   IWorkerChoiceStrategy,
   9   MeasurementStatisticsRequirements,
  10   StrategyPolicy,
  11   TaskStatisticsRequirements,
  12   WorkerChoiceStrategyOptions
  13 } from './selection-strategies-types.js'
  14
  15 /**
  16  * Worker choice strategy abstract base class.
  17  *
  18  * @typeParam Worker - Type of worker which manages the strategy.
  19  * @typeParam Data - Type of data sent to the worker. This can only be structured-cloneable data.
  20  * @typeParam Response - Type of execution response. This can only be structured-cloneable data.
  21  */
  22 export abstract class AbstractWorkerChoiceStrategy<
  23   Worker extends IWorker,
  24   Data = unknown,
  25   Response = unknown
  26 > implements IWorkerChoiceStrategy {
  27   /**
  28    * The next worker node key.
  29    */
  30   protected nextWorkerNodeKey: number | undefined = 0
  31
  32   /**
  33    * The previous worker node key.
  34    */
  35   protected previousWorkerNodeKey: number = 0
  36
  37   /** @inheritDoc */
  38   public readonly strategyPolicy: StrategyPolicy = {
  39     dynamicWorkerUsage: false,
  40     dynamicWorkerReady: true
  41   }
  42
  43   /** @inheritDoc */
  44   public readonly taskStatisticsRequirements: TaskStatisticsRequirements = {
  45     runTime: DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
  46     waitTime: DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
  47     elu: DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS
  48   }
  49
  50   /**
  51    * Constructs a worker choice strategy bound to the pool.
  52    *
  53    * @param pool - The pool instance.
  54    * @param opts - The worker choice strategy options.
  55    */
  56   public constructor (
  57     protected readonly pool: IPool<Worker, Data, Response>,
  58     protected opts?: WorkerChoiceStrategyOptions
  59   ) {
  60     this.setOptions(this.opts)
  61     this.choose = this.choose.bind(this)
  62   }
  63
  64   protected setTaskStatisticsRequirements (
  65     opts: WorkerChoiceStrategyOptions | undefined
  66   ): void {
  67     this.toggleMedianMeasurementStatisticsRequirements(
  68       this.taskStatisticsRequirements.runTime,
  69       // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
  70       opts!.runTime!.median
  71     )
  72     this.toggleMedianMeasurementStatisticsRequirements(
  73       this.taskStatisticsRequirements.waitTime,
  74       // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
  75       opts!.waitTime!.median
  76     )
  77     this.toggleMedianMeasurementStatisticsRequirements(
  78       this.taskStatisticsRequirements.elu,
  79       // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
  80       opts!.elu!.median
  81     )
  82   }
  83
  84   private toggleMedianMeasurementStatisticsRequirements (
  85     measurementStatisticsRequirements: MeasurementStatisticsRequirements,
  86     toggleMedian: boolean
  87   ): void {
  88     if (measurementStatisticsRequirements.average && toggleMedian) {
  89       measurementStatisticsRequirements.average = false
  90       measurementStatisticsRequirements.median = toggleMedian
  91     }
  92     if (measurementStatisticsRequirements.median && !toggleMedian) {
  93       measurementStatisticsRequirements.average = true
  94       measurementStatisticsRequirements.median = toggleMedian
  95     }
  96   }
  97
  98   protected resetWorkerNodeKeyProperties (): void {
  99     this.nextWorkerNodeKey = 0
 100     this.previousWorkerNodeKey = 0
 101   }
 102
 103   /** @inheritDoc */
 104   public abstract reset (): boolean
 105
 106   /** @inheritDoc */
 107   public abstract update (workerNodeKey: number): boolean
 108
 109   /** @inheritDoc */
 110   public abstract choose (): number | undefined
 111
 112   /** @inheritDoc */
 113   public abstract remove (workerNodeKey: number): boolean
 114
 115   /** @inheritDoc */
 116   public setOptions (opts: WorkerChoiceStrategyOptions | undefined): void {
 117     this.opts = buildWorkerChoiceStrategyOptions<Worker, Data, Response>(
 118       this.pool,
 119       opts
 120     )
 121     this.setTaskStatisticsRequirements(this.opts)
 122   }
 123
 124   /** @inheritDoc */
 125   public hasPoolWorkerNodesReady (): boolean {
 126     return this.pool.workerNodes.some(workerNode => workerNode.info.ready)
 127   }
 128
 129   /**
 130    * Whether the worker node is ready or not.
 131    *
 132    * @param workerNodeKey - The worker node key.
 133    * @returns Whether the worker node is ready or not.
 134    */
 135   protected isWorkerNodeReady (workerNodeKey: number): boolean {
 136     return this.pool.workerNodes[workerNodeKey]?.info?.ready ?? false
 137   }
 138
 139   /**
 140    * Check the next worker node readiness.
 141    */
 142   protected checkNextWorkerNodeReadiness (): void {
 143     // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
 144     if (!this.isWorkerNodeReady(this.nextWorkerNodeKey!)) {
 145       delete this.nextWorkerNodeKey
 146     }
 147   }
 148
 149   /**
 150    * Gets the worker node task runtime.
 151    * If the task statistics require the average runtime, the average runtime is returned.
 152    * If the task statistics require the median runtime , the median runtime is returned.
 153    *
 154    * @param workerNodeKey - The worker node key.
 155    * @returns The worker node task runtime.
 156    */
 157   protected getWorkerNodeTaskRunTime (workerNodeKey: number): number {
 158     return this.taskStatisticsRequirements.runTime.median
 159       ? this.pool.workerNodes[workerNodeKey].usage.runTime.median ?? 0
 160       : this.pool.workerNodes[workerNodeKey].usage.runTime.average ?? 0
 161   }
 162
 163   /**
 164    * Gets the worker node task wait time.
 165    * If the task statistics require the average wait time, the average wait time is returned.
 166    * If the task statistics require the median wait time, the median wait time is returned.
 167    *
 168    * @param workerNodeKey - The worker node key.
 169    * @returns The worker node task wait time.
 170    */
 171   protected getWorkerNodeTaskWaitTime (workerNodeKey: number): number {
 172     return this.taskStatisticsRequirements.waitTime.median
 173       ? this.pool.workerNodes[workerNodeKey].usage.waitTime.median ?? 0
 174       : this.pool.workerNodes[workerNodeKey].usage.waitTime.average ?? 0
 175   }
 176
 177   /**
 178    * Gets the worker node task ELU.
 179    * If the task statistics require the average ELU, the average ELU is returned.
 180    * If the task statistics require the median ELU, the median ELU is returned.
 181    *
 182    * @param workerNodeKey - The worker node key.
 183    * @returns The worker node task ELU.
 184    */
 185   protected getWorkerNodeTaskElu (workerNodeKey: number): number {
 186     return this.taskStatisticsRequirements.elu.median
 187       ? this.pool.workerNodes[workerNodeKey].usage.elu.active.median ?? 0
 188       : this.pool.workerNodes[workerNodeKey].usage.elu.active.average ?? 0
 189   }
 190
 191   /**
 192    * Sets safely the previous worker node key.
 193    *
 194    * @param workerNodeKey - The worker node key.
 195    */
 196   protected setPreviousWorkerNodeKey (workerNodeKey: number | undefined): void {
 197     this.previousWorkerNodeKey = workerNodeKey ?? this.previousWorkerNodeKey
 198   }
 199 }