-import fs from 'fs';
+import type { EventEmitterAsyncResource } from 'node:events'
+import { existsSync } from 'node:fs'
-import WorkerConstants from './WorkerConstants';
-import type { WorkerData, WorkerOptions } from '../types/Worker';
+import type { PoolInfo } from 'poolifier'
-export default abstract class WorkerAbstract<T extends WorkerData> {
- protected readonly workerScript: string;
- protected readonly workerOptions: WorkerOptions;
- public abstract readonly size: number;
- public abstract readonly maxElementsPerWorker: number | null;
+import type { SetInfo, WorkerData, WorkerOptions } from './WorkerTypes.js'
+
+export abstract class WorkerAbstract<D extends WorkerData, R extends WorkerData> {
+ protected readonly workerScript: string
+ protected readonly workerOptions: WorkerOptions
+ public abstract readonly info: PoolInfo | SetInfo
+ public abstract readonly size: number
+ public abstract readonly maxElementsPerWorker: number | undefined
+ public abstract readonly emitter: EventEmitterAsyncResource | undefined
/**
* `WorkerAbstract` constructor.
* @param workerScript -
* @param workerOptions -
*/
- constructor(
- workerScript: string,
- workerOptions: WorkerOptions = {
- workerStartDelay: WorkerConstants.DEFAULT_WORKER_START_DELAY,
- elementStartDelay: WorkerConstants.DEFAULT_ELEMENT_START_DELAY,
- poolMinSize: WorkerConstants.DEFAULT_POOL_MIN_SIZE,
- poolMaxSize: WorkerConstants.DEFAULT_POOL_MAX_SIZE,
- elementsPerWorker: WorkerConstants.DEFAULT_ELEMENTS_PER_WORKER,
- poolOptions: {},
- messageHandler: () => {
- /* This is intentional */
- },
+ constructor (workerScript: string | undefined, workerOptions: WorkerOptions) {
+ if (workerScript == null) {
+ throw new TypeError('Worker script is not defined')
+ }
+ if (typeof workerScript !== 'string') {
+ throw new TypeError('Worker script must be a string')
}
- ) {
- if (!workerScript) {
- throw new Error('Worker script is not defined');
+ if (workerScript.trim().length === 0) {
+ throw new Error('Worker script is an empty string')
}
- if (!fs.existsSync(workerScript)) {
- throw new Error('Worker script file does not exist');
+ if (!existsSync(workerScript)) {
+ throw new Error('Worker script file does not exist')
}
- this.workerScript = workerScript;
- this.workerOptions = workerOptions;
+ this.workerScript = workerScript
+ this.workerOptions = workerOptions
}
- public abstract start(): Promise<void>;
- public abstract stop(): Promise<void>;
- public abstract addElement(elementData: T): Promise<void>;
+ /**
+ * Starts the worker pool/set.
+ */
+ public abstract start (): void | Promise<void>
+ /**
+ * Stops the worker pool/set.
+ */
+ public abstract stop (): Promise<void>
+ /**
+ * Adds a task element to the worker pool/set.
+ *
+ * @param elementData -
+ */
+ public abstract addElement (elementData: D): Promise<R>
}