src/pools/selection-strategies/least-elu-worker-choice-strategy.ts

   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'
   5 import type {
   6   IWorkerChoiceStrategy,
   7   TaskStatisticsRequirements,
   8   WorkerChoiceStrategyOptions
   9 } from './selection-strategies-types'
  10
  11 /**
  12  * Selects the worker with the least ELU.
  13  *
  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.
  17  */
  18 export class LeastEluWorkerChoiceStrategy<
  19     Worker extends IWorker,
  20     Data = unknown,
  21     Response = unknown
  22   >
  23   extends AbstractWorkerChoiceStrategy<Worker, Data, Response>
  24   implements IWorkerChoiceStrategy {
  25   /** @inheritDoc */
  26   public readonly taskStatisticsRequirements: TaskStatisticsRequirements = {
  27     runTime: {
  28       aggregate: false,
  29       average: false,
  30       median: false
  31     },
  32     waitTime: {
  33       aggregate: false,
  34       average: false,
  35       median: false
  36     },
  37     elu: {
  38       aggregate: true,
  39       average: false,
  40       median: false
  41     }
  42   }
  43
  44   /** @inheritDoc */
  45   public constructor (
  46     pool: IPool<Worker, Data, Response>,
  47     opts: WorkerChoiceStrategyOptions = DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS
  48   ) {
  49     super(pool, opts)
  50     this.setTaskStatisticsRequirements(this.opts)
  51   }
  52
  53   /** @inheritDoc */
  54   public reset (): boolean {
  55     return true
  56   }
  57
  58   /** @inheritDoc */
  59   public update (): boolean {
  60     return true
  61   }
  62
  63   /** @inheritDoc */
  64   public choose (): number {
  65     let minWorkerElu = Infinity
  66     for (const [workerNodeKey, workerNode] of this.pool.workerNodes.entries()) {
  67       const workerUsage = workerNode.workerUsage
  68       const workerElu = workerUsage.elu?.active.aggregate ?? 0
  69       if (workerElu === 0) {
  70         this.nextWorkerNodeId = workerNodeKey
  71         break
  72       } else if (workerElu < minWorkerElu) {
  73         minWorkerElu = workerElu
  74         this.nextWorkerNodeId = workerNodeKey
  75       }
  76     }
  77     return this.nextWorkerNodeId
  78   }
  79
  80   /** @inheritDoc */
  81   public remove (): boolean {
  82     return true
  83   }
  84 }