Commit | Line | Data |
---|---|---|
c32882b0 | 1 | import { DynamicThreadPool, PoolOptions } from 'poolifier'; |
a4624c96 | 2 | |
a4624c96 | 3 | import Utils from '../utils/Utils'; |
c32882b0 | 4 | import { Worker } from 'worker_threads'; |
fd1fdf1b | 5 | import WorkerAbstract from './WorkerAbstract'; |
a4624c96 | 6 | import { WorkerData } from '../types/Worker'; |
7874b0b1 | 7 | import { WorkerUtils } from './WorkerUtils'; |
a4624c96 | 8 | |
fd1fdf1b | 9 | export default class WorkerDynamicPool<T> extends WorkerAbstract { |
8df3f0a9 | 10 | private pool: DynamicThreadPool<WorkerData>; |
a4624c96 JB |
11 | |
12 | /** | |
13 | * Create a new `WorkerDynamicPool`. | |
14 | * | |
81797102 JB |
15 | * @param workerScript |
16 | * @param min | |
17 | * @param max | |
18 | * @param workerStartDelay | |
19 | * @param opts | |
20 | * @param messageListenerCallback | |
a4624c96 | 21 | */ |
81797102 JB |
22 | constructor(workerScript: string, min: number, max: number, workerStartDelay?: number, opts?: PoolOptions<Worker>, |
23 | messageListenerCallback: (message: any) => void = () => { /* This is intentional */ }) { | |
24 | super(workerScript, workerStartDelay, messageListenerCallback); | |
8df3f0a9 JB |
25 | opts.exitHandler = opts?.exitHandler ?? WorkerUtils.defaultExitHandler; |
26 | this.pool = new DynamicThreadPool<WorkerData>(min, max, this.workerScript, opts); | |
a4624c96 JB |
27 | } |
28 | ||
29 | get size(): number { | |
30 | return this.pool.workers.length; | |
31 | } | |
32 | ||
6e0964c8 | 33 | get maxElementsPerWorker(): number | null { |
85f78bc0 | 34 | return null; |
a4624c96 JB |
35 | } |
36 | ||
37 | /** | |
38 | * | |
81797102 | 39 | * @returns |
a4624c96 JB |
40 | * @public |
41 | */ | |
42 | // eslint-disable-next-line @typescript-eslint/no-empty-function | |
6c3cfef8 JB |
43 | public async start(): Promise<void> { |
44 | // This is intentional | |
45 | } | |
a4624c96 | 46 | |
ded13d97 JB |
47 | /** |
48 | * | |
81797102 | 49 | * @returns |
ded13d97 JB |
50 | * @public |
51 | */ | |
fd1fdf1b | 52 | // eslint-disable-next-line @typescript-eslint/require-await |
ded13d97 JB |
53 | public async stop(): Promise<void> { |
54 | return this.pool.destroy(); | |
55 | } | |
56 | ||
a4624c96 JB |
57 | /** |
58 | * | |
81797102 JB |
59 | * @param elementData |
60 | * @returns | |
a4624c96 JB |
61 | * @public |
62 | */ | |
8434025b | 63 | public async addElement(elementData: T): Promise<void> { |
a4624c96 JB |
64 | await this.pool.execute(elementData); |
65 | // Start worker sequentially to optimize memory at startup | |
322c9192 | 66 | await Utils.sleep(this.workerStartDelay); |
a4624c96 JB |
67 | } |
68 | } |