X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fpools%2Fpool.ts;h=257b0d64b0844d664ba00763b1dcde79c4711ee7;hb=e8b3a5abad7d3ae4fa836ab0b76d54c1efd146e9;hp=992aaaf1099c7e341daf3bd1c6a937597018b5ea;hpb=70c7d7d3af9c38417416ca9096a6ab3ae835b5d0;p=poolifier.git diff --git a/src/pools/pool.ts b/src/pools/pool.ts index 992aaaf1..257b0d64 100644 --- a/src/pools/pool.ts +++ b/src/pools/pool.ts @@ -1,4 +1,4 @@ -import EventEmitter from 'node:events' +import EventEmitterAsyncResource from 'node:events' import type { ErrorHandler, ExitHandler, @@ -13,33 +13,50 @@ import type { } from './selection-strategies/selection-strategies-types' /** - * Pool types. - * - * @enum - * @internal + * Enumeration of pool types. */ -export enum PoolType { +export const PoolTypes = Object.freeze({ /** * Fixed pool type. */ - FIXED = 'fixed', + fixed: 'fixed', /** * Dynamic pool type. */ - DYNAMIC = 'dynamic' -} + dynamic: 'dynamic' +} as const) + +/** + * Pool type. + */ +export type PoolType = keyof typeof PoolTypes + +/** + * Enumeration of worker types. + */ +export const WorkerTypes = Object.freeze({ + cluster: 'cluster', + thread: 'thread' +} as const) + +/** + * Worker type. + */ +export type WorkerType = keyof typeof WorkerTypes /** * Pool events emitter. */ -export class PoolEmitter extends EventEmitter {} +export class PoolEmitter extends EventEmitterAsyncResource {} /** * Enumeration of pool events. */ export const PoolEvents = Object.freeze({ full: 'full', - busy: 'busy' + busy: 'busy', + error: 'error', + taskError: 'taskError' } as const) /** @@ -47,6 +64,24 @@ export const PoolEvents = Object.freeze({ */ export type PoolEvent = keyof typeof PoolEvents +/** + * Pool information. + */ +export interface PoolInfo { + type: PoolType + worker: WorkerType + minSize: number + maxSize: number + workerNodes: number + idleWorkerNodes: number + busyWorkerNodes: number + executedTasks: number + executingTasks: number + queuedTasks: number + maxQueuedTasks: number + failedTasks: number +} + /** * Worker tasks queue options. */ @@ -91,6 +126,10 @@ export interface PoolOptions { * The worker choice strategy options. */ workerChoiceStrategyOptions?: WorkerChoiceStrategyOptions + /** + * Restart worker on error. + */ + restartWorkerOnError?: boolean /** * Pool events emission. * @@ -122,11 +161,9 @@ export interface IPool< Response = unknown > { /** - * Pool type. - * - * If it is `'dynamic'`, it provides the `max` property. + * Pool information. */ - readonly type: PoolType + readonly info: PoolInfo /** * Pool worker nodes. */ @@ -138,10 +175,12 @@ export interface IPool< * * - `'full'`: Emitted when the pool is dynamic and full. * - `'busy'`: Emitted when the pool is busy. + * - `'error'`: Emitted when an uncaught error occurs. + * - `'taskError'`: Emitted when an error occurs while executing a task. */ readonly emitter?: PoolEmitter /** - * Executes the function specified in the worker constructor with the task data input parameter. + * Executes the specified function in the worker constructor with the task data input parameter. * * @param data - The task input data for the specified worker function. This can only be serializable data. * @param name - The name of the worker function to execute. If not specified, the default worker function will be executed. @@ -149,7 +188,7 @@ export interface IPool< */ execute: (data?: Data, name?: string) => Promise /** - * Shutdowns every current worker in this pool. + * Terminate every current worker in this pool. */ destroy: () => Promise /**