fix: ensure worker removal impact is propated to worker choice strategy

[poolifier.git] / src / pools / selection-strategies / abstract-worker-choice-strategy.ts

import type { IPoolInternal } from '../pool-internal'
import { PoolType } from '../pool-internal'
import type { IPoolWorker } from '../pool-worker'
import type {
  IWorkerChoiceStrategy,
  RequiredStatistics
} from './selection-strategies-types'

/**
 * Abstract worker choice strategy class.
 *
 * @typeParam Worker - Type of worker which manages the strategy.
 * @typeParam Data - Type of data sent to the worker. This can only be serializable data.
 * @typeParam Response - Type of response of execution. This can only be serializable data.
 */
export abstract class AbstractWorkerChoiceStrategy<
  Worker extends IPoolWorker,
  Data,
  Response
> implements IWorkerChoiceStrategy {
  /** {@inheritDoc} */
  public readonly isDynamicPool: boolean
  /** {@inheritDoc} */
  public requiredStatistics: RequiredStatistics = {
    runTime: false
  }

  /**
   * Constructs a worker choice strategy attached to the pool.
   *
   * @param pool - The pool instance.
   */
  public constructor (
    protected readonly pool: IPoolInternal<Worker, Data, Response>
  ) {
    this.isDynamicPool = this.pool.type === PoolType.DYNAMIC
  }

  /** {@inheritDoc} */
  public abstract reset (): boolean

  /** {@inheritDoc} */
  public abstract choose (): number

  /** {@inheritDoc} */
  public abstract remove (workerKey: number): boolean
}