Benchmarks and performance enhancements (#209)

[poolifier.git] / src / utility-types.ts

import type { Worker } from 'cluster'
import type { MessagePort } from 'worker_threads'
import type { IWorker } from './pools/abstract-pool'
import type { KillBehavior } from './worker/worker-options'

/**
 * Make all properties in T non-readonly.
 */
export type Draft<T> = { -readonly [P in keyof T]?: T[P] }

/**
 * Message object that is passed between worker and main worker.
 */
export interface MessageValue<
  Data = unknown,
  MainWorker extends Worker | MessagePort | unknown = unknown
> {
  /**
   * Input data that will be passed to the worker.
   */
  readonly data?: Data
  /**
   * ID of the message.
   */
  readonly id?: number
  /**
   * Kill code.
   */
  readonly kill?: KillBehavior | 1
  /**
   * Error.
   */
  readonly error?: string
  /**
   * Reference to main worker.
   *
   * _Only for internal use_
   */
  readonly parent?: MainWorker
}

/**
 * An object holding the worker that will be used to resolve/rejects the promise later on.
 *
 * @template Worker Type of worker.
 * @template Response Type of response of execution. This can only be serializable data.
 */
export interface PromiseWorkerResponseWrapper<
  Worker extends IWorker,
  Response = unknown
> {
  /**
   * Resolve callback to fulfill the promise.
   */
  readonly resolve: (value: Response) => void
  /**
   * Reject callback to reject the promise.
   */
  readonly reject: (reason?: string) => void
  /**
   * The worker that has the assigned task.
   */
  readonly worker: Worker
}