| 1 | import fs from 'fs'; |
| 2 | |
| 3 | import { WorkerData, WorkerOptions } from '../types/Worker'; |
| 4 | import WorkerConstants from './WorkerConstants'; |
| 5 | |
| 6 | export default abstract class WorkerAbstract<T extends WorkerData> { |
| 7 | protected readonly workerScript: string; |
| 8 | protected readonly workerOptions: WorkerOptions; |
| 9 | public abstract readonly size: number; |
| 10 | public abstract readonly maxElementsPerWorker: number | null; |
| 11 | |
| 12 | /** |
| 13 | * `WorkerAbstract` constructor. |
| 14 | * |
| 15 | * @param workerScript |
| 16 | * @param workerOptions |
| 17 | */ |
| 18 | constructor( |
| 19 | workerScript: string, |
| 20 | workerOptions: WorkerOptions = { |
| 21 | workerStartDelay: WorkerConstants.DEFAULT_WORKER_START_DELAY, |
| 22 | elementStartDelay: WorkerConstants.DEFAULT_ELEMENT_START_DELAY, |
| 23 | poolMinSize: WorkerConstants.DEFAULT_POOL_MIN_SIZE, |
| 24 | poolMaxSize: WorkerConstants.DEFAULT_POOL_MAX_SIZE, |
| 25 | elementsPerWorker: WorkerConstants.DEFAULT_ELEMENTS_PER_WORKER, |
| 26 | poolOptions: {}, |
| 27 | messageHandler: () => { |
| 28 | /* This is intentional */ |
| 29 | }, |
| 30 | } |
| 31 | ) { |
| 32 | if (!workerScript) { |
| 33 | throw new Error('Worker script is not defined'); |
| 34 | } |
| 35 | if (!fs.existsSync(workerScript)) { |
| 36 | throw new Error('Worker script file does not exist'); |
| 37 | } |
| 38 | this.workerScript = workerScript; |
| 39 | this.workerOptions = workerOptions; |
| 40 | } |
| 41 | |
| 42 | public abstract start(): Promise<void>; |
| 43 | public abstract stop(): Promise<void>; |
| 44 | public abstract addElement(elementData: T): Promise<void>; |
| 45 | } |