-import type { WorkerChoiceStrategy } from './selection-strategies'
+import EventEmitter from 'events'
+import type {
+ ErrorHandler,
+ ExitHandler,
+ MessageHandler,
+ OnlineHandler
+} from './pool-worker'
+import type { WorkerChoiceStrategy } from './selection-strategies/selection-strategies-types'
+
+/**
+ * Pool events emitter.
+ */
+export class PoolEmitter extends EventEmitter {}
+
+/**
+ * Options for a poolifier pool.
+ */
+export interface PoolOptions<Worker> {
+ /**
+ * A function that will listen for message event on each worker.
+ */
+ messageHandler?: MessageHandler<Worker>
+ /**
+ * A function that will listen for error event on each worker.
+ */
+ errorHandler?: ErrorHandler<Worker>
+ /**
+ * A function that will listen for online event on each worker.
+ */
+ onlineHandler?: OnlineHandler<Worker>
+ /**
+ * A function that will listen for exit event on each worker.
+ */
+ exitHandler?: ExitHandler<Worker>
+ /**
+ * The worker choice strategy to use in this pool.
+ */
+ workerChoiceStrategy?: WorkerChoiceStrategy
+ /**
+ * Pool events emission.
+ *
+ * @default true
+ */
+ enableEvents?: boolean
+}
/**
* Contract definition for a poolifier pool.
*/
export interface IPool<Data = unknown, Response = unknown> {
/**
- * Perform the task specified in the constructor with the data parameter.
+ * Emitter on which events can be listened to.
+ *
+ * Events that can currently be listened to:
+ *
+ * - `'busy'`
+ */
+ readonly emitter?: PoolEmitter
+ /**
+ * Performs the task specified in the constructor with the data parameter.
*
* @param data The input for the specified task. This can only be serializable data.
* @returns Promise that will be resolved when the task is successfully completed.
*/
- execute(data: Data): Promise<Response>
+ execute: (data: Data) => Promise<Response>
/**
- * Shut down every current worker in this pool.
+ * Shutdowns every current worker in this pool.
*/
- destroy(): Promise<void>
+ destroy: () => Promise<void>
/**
- * Set the worker choice strategy in this pool.
+ * Sets the worker choice strategy in this pool.
*
* @param workerChoiceStrategy The worker choice strategy.
*/
- setWorkerChoiceStrategy(workerChoiceStrategy: WorkerChoiceStrategy): void
+ setWorkerChoiceStrategy: (workerChoiceStrategy: WorkerChoiceStrategy) => void
}