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'; |
a4624c96 | 7 | |
fd1fdf1b | 8 | export default class WorkerDynamicPool<T> extends WorkerAbstract { |
a4624c96 JB |
9 | private pool: DynamicPool; |
10 | ||
11 | /** | |
12 | * Create a new `WorkerDynamicPool`. | |
13 | * | |
14 | * @param {string} workerScript | |
e71cccf3 JB |
15 | * @param {number} min |
16 | * @param {number} max | |
322c9192 | 17 | * @param {number} workerStartDelay |
a4624c96 | 18 | */ |
322c9192 JB |
19 | constructor(workerScript: string, min: number, max: number, workerStartDelay?: number) { |
20 | super(workerScript, workerStartDelay); | |
a4624c96 JB |
21 | this.pool = DynamicPool.getInstance(min, max, this.workerScript); |
22 | } | |
23 | ||
24 | get size(): number { | |
25 | return this.pool.workers.length; | |
26 | } | |
27 | ||
28 | get maxElementsPerWorker(): number { | |
85f78bc0 | 29 | return null; |
a4624c96 JB |
30 | } |
31 | ||
32 | /** | |
33 | * | |
3340259a | 34 | * @returns {Promise<void>} |
a4624c96 JB |
35 | * @public |
36 | */ | |
37 | // eslint-disable-next-line @typescript-eslint/no-empty-function | |
38 | public async start(): Promise<void> { } | |
39 | ||
ded13d97 JB |
40 | /** |
41 | * | |
3340259a | 42 | * @returns {Promise<void>} |
ded13d97 JB |
43 | * @public |
44 | */ | |
fd1fdf1b | 45 | // eslint-disable-next-line @typescript-eslint/require-await |
ded13d97 JB |
46 | public async stop(): Promise<void> { |
47 | return this.pool.destroy(); | |
48 | } | |
49 | ||
a4624c96 JB |
50 | /** |
51 | * | |
3340259a JB |
52 | * @param elementData |
53 | * @returns {Promise<void>} | |
a4624c96 JB |
54 | * @public |
55 | */ | |
8434025b | 56 | public async addElement(elementData: T): Promise<void> { |
a4624c96 JB |
57 | await this.pool.execute(elementData); |
58 | // Start worker sequentially to optimize memory at startup | |
322c9192 | 59 | await Utils.sleep(this.workerStartDelay); |
a4624c96 JB |
60 | } |
61 | } | |
62 | ||
63 | class DynamicPool extends DynamicThreadPool<WorkerData> { | |
64 | private static instance: DynamicPool; | |
65 | ||
63b19acd JB |
66 | private constructor(min: number, max: number, workerScript: string, opts?: PoolOptions<Worker>) { |
67 | super(min, max, workerScript, opts); | |
a4624c96 JB |
68 | } |
69 | ||
63b19acd | 70 | public static getInstance(min: number, max: number, workerScript: string): DynamicPool { |
a4624c96 | 71 | if (!DynamicPool.instance) { |
63b19acd | 72 | DynamicPool.instance = new DynamicPool(min, max, workerScript, |
a4624c96 JB |
73 | { |
74 | exitHandler: (code) => { | |
75 | if (code !== 0) { | |
1e924543 | 76 | console.error(`Worker stopped with exit code ${code}`); |
a4624c96 JB |
77 | } |
78 | } | |
79 | } | |
80 | ); | |
81 | } | |
82 | return DynamicPool.instance; | |
83 | } | |
84 | } |