Commit | Line | Data |
---|---|---|
d972af76 JB |
1 | import type { EventEmitter } from 'node:events'; |
2 | import { existsSync } from 'node:fs'; | |
4bfd80fa | 3 | |
a37fc6dc | 4 | import type { PoolEmitter, PoolInfo } from 'poolifier'; |
b779c0f8 | 5 | |
268a74bb | 6 | import { WorkerConstants } from './WorkerConstants'; |
b779c0f8 | 7 | import type { SetInfo, WorkerData, WorkerOptions } from './WorkerTypes'; |
be245fda | 8 | import { defaultErrorHandler, defaultExitHandler } from './WorkerUtils'; |
7dde0b73 | 9 | |
268a74bb | 10 | export abstract class WorkerAbstract<T extends WorkerData> { |
322c9192 | 11 | protected readonly workerScript: string; |
4d7227e6 | 12 | protected readonly workerOptions: WorkerOptions; |
b779c0f8 | 13 | public abstract readonly info: PoolInfo | SetInfo; |
f2bf9948 | 14 | public abstract readonly size: number; |
72092cfc | 15 | public abstract readonly maxElementsPerWorker: number | undefined; |
d972af76 | 16 | public abstract readonly emitter: EventEmitter | PoolEmitter | undefined; |
6af9012e | 17 | |
7dde0b73 | 18 | /** |
e71cccf3 | 19 | * `WorkerAbstract` constructor. |
7dde0b73 | 20 | * |
0e4fa348 JB |
21 | * @param workerScript - |
22 | * @param workerOptions - | |
7dde0b73 | 23 | */ |
e7aeea18 JB |
24 | constructor( |
25 | workerScript: string, | |
26 | workerOptions: WorkerOptions = { | |
3fa0f0ed JB |
27 | workerStartDelay: WorkerConstants.DEFAULT_WORKER_START_DELAY, |
28 | elementStartDelay: WorkerConstants.DEFAULT_ELEMENT_START_DELAY, | |
29 | poolMinSize: WorkerConstants.DEFAULT_POOL_MIN_SIZE, | |
30 | poolMaxSize: WorkerConstants.DEFAULT_POOL_MAX_SIZE, | |
31 | elementsPerWorker: WorkerConstants.DEFAULT_ELEMENTS_PER_WORKER, | |
abe9e9dd | 32 | poolOptions: {}, |
5edd8ba0 | 33 | }, |
e7aeea18 | 34 | ) { |
ba516f9c | 35 | if (workerScript === null || workerScript === undefined) { |
44a95b7f JB |
36 | throw new Error('Worker script is not defined'); |
37 | } | |
ba516f9c JB |
38 | if (typeof workerScript === 'string' && workerScript.trim().length === 0) { |
39 | throw new Error('Worker script is empty'); | |
40 | } | |
d972af76 | 41 | if (!existsSync(workerScript)) { |
44a95b7f JB |
42 | throw new Error('Worker script file does not exist'); |
43 | } | |
ad2f27c3 | 44 | this.workerScript = workerScript; |
4d7227e6 | 45 | this.workerOptions = workerOptions; |
a37fc6dc JB |
46 | this.workerOptions.poolOptions!.errorHandler = |
47 | this.workerOptions.poolOptions?.errorHandler ?? defaultErrorHandler; | |
48 | this.workerOptions.poolOptions!.exitHandler = | |
49 | this.workerOptions.poolOptions?.exitHandler ?? defaultExitHandler; | |
7dde0b73 JB |
50 | } |
51 | ||
8baf3f8f | 52 | /** |
361c98f5 | 53 | * Starts the worker pool/set. |
8baf3f8f | 54 | */ |
418106c8 | 55 | public abstract start(): Promise<void>; |
8baf3f8f | 56 | /** |
361c98f5 | 57 | * Stops the worker pool/set. |
8baf3f8f | 58 | */ |
ded13d97 | 59 | public abstract stop(): Promise<void>; |
8baf3f8f | 60 | /** |
361c98f5 | 61 | * Adds a task element to the worker pool/set. |
8baf3f8f JB |
62 | * |
63 | * @param elementData - | |
64 | */ | |
c3ee95af | 65 | public abstract addElement(elementData: T): Promise<void>; |
7dde0b73 | 66 | } |