8a7e6e558f170c42e00ab22c257507370b50e96e
1 import type { Worker
} from
'node:worker_threads';
3 import { DynamicThreadPool
, type ErrorHandler
, type ExitHandler
} from
'poolifier';
5 import { WorkerAbstract
} from
'./WorkerAbstract';
6 import type { WorkerData
, WorkerOptions
} from
'./WorkerTypes';
7 import { WorkerUtils
} from
'./WorkerUtils';
9 export class WorkerDynamicPool
extends WorkerAbstract
<WorkerData
> {
10 private readonly pool
: DynamicThreadPool
<WorkerData
>;
13 * Create a new `WorkerDynamicPool`.
15 * @param workerScript -
16 * @param workerOptions -
18 constructor(workerScript
: string, workerOptions
?: WorkerOptions
) {
19 super(workerScript
, workerOptions
);
20 this.workerOptions
.poolOptions
.errorHandler
= (
21 this.workerOptions
?.poolOptions
?.errorHandler
?? WorkerUtils
.defaultErrorHandler
22 ).bind(this) as ErrorHandler
<Worker
>;
23 this.workerOptions
.poolOptions
.exitHandler
= (
24 this.workerOptions
?.poolOptions
?.exitHandler
?? WorkerUtils
.defaultExitHandler
25 ).bind(this) as ExitHandler
<Worker
>;
26 this.workerOptions
.poolOptions
.messageHandler
.bind(this);
27 this.pool
= new DynamicThreadPool
<WorkerData
>(
28 this.workerOptions
.poolMinSize
,
29 this.workerOptions
.poolMaxSize
,
31 this.workerOptions
.poolOptions
36 return this.pool
.workers
.length
;
39 get
maxElementsPerWorker(): number | undefined {
48 public async start(): Promise
<void> {
49 // This is intentional
57 public async stop(): Promise
<void> {
58 return this.pool
.destroy();
63 * @param elementData -
67 public async addElement(elementData
: WorkerData
): Promise
<void> {
68 await this.pool
.execute(elementData
);
69 // Start element sequentially to optimize memory at startup
70 this.workerOptions
.elementStartDelay
> 0 &&
71 (await WorkerUtils
.sleep(this.workerOptions
.elementStartDelay
));