Optimize WRR implementation

[poolifier.git] / src / pools / pool-internal.ts

import EventEmitter from 'events'
import type { IPool } from './pool'
import type { IPoolWorker } from './pool-worker'

/**
 * Pool types.
 */
export enum PoolType {
  FIXED = 'fixed',
  DYNAMIC = 'dynamic'
}

/**
 * Tasks usage statistics.
 */
export interface TasksUsage {
  run: number
  running: number
  runTime: number
  avgRunTime: number
}

/**
 * Internal poolifier pool emitter.
 */
export class PoolEmitter extends EventEmitter {}

/**
 * Internal contract definition for a poolifier pool.
 *
 * @template Worker Type of worker which manages this pool.
 * @template Data Type of data sent to the worker.
 * @template Response Type of response of execution.
 */
export interface IPoolInternal<
  Worker extends IPoolWorker,
  Data = unknown,
  Response = unknown
> extends IPool<Data, Response> {
  /**
   * List of currently available workers.
   */
  readonly workers: Worker[]

  /**
   * The workers tasks usage map.
   *
   *  `key`: The `Worker`
   *  `value`: Worker tasks usage statistics.
   */
  readonly workersTasksUsage: Map<Worker, TasksUsage>

  /**
   * Emitter on which events can be listened to.
   *
   * Events that can currently be listened to:
   *
   * - `'busy'`
   */
  readonly emitter?: PoolEmitter

  /**
   * Pool type.
   *
   * If it is `'dynamic'`, it provides the `max` property.
   */
  readonly type: PoolType

  /**
   * Maximum number of workers that can be created by this pool.
   */
  readonly max?: number

  /**
   * Whether the pool is busy or not.
   *
   * The pool busyness boolean status.
   */
  readonly busy: boolean

  /**
   * Number of tasks currently concurrently running.
   */
  readonly numberOfRunningTasks: number

  /**
   * Find a free worker based on the number of tasks the worker has applied.
   *
   * If a worker is found with `0` running tasks, it is detected as free and returned.
   *
   * If no free worker is found, `false` is returned.
   *
   * @returns A free worker if there is one, otherwise `false`.
   */
  findFreeWorker(): Worker | false

  /**
   * Get worker index.
   *
   * @param worker The worker.
   * @returns The worker index.
   */
  getWorkerIndex(worker: Worker): number

  /**
   * Get worker running tasks.
   *
   * @param worker The worker.
   * @returns The number of tasks currently running on the worker.
   */
  getWorkerRunningTasks(worker: Worker): number | undefined

  /**
   * Get worker average tasks runtime.
   *
   * @param worker The worker.
   * @returns The average tasks runtime on the worker.
   */
  getWorkerAverageTasksRunTime(worker: Worker): number | undefined
}