1 import type { Worker
} from
'worker_threads';
3 import { type ErrorHandler
, type ExitHandler
, FixedThreadPool
} from
'poolifier';
5 import type { WorkerData
, WorkerOptions
} from
'../types/Worker';
6 import Utils from
'../utils/Utils';
7 import WorkerAbstract from
'./WorkerAbstract';
8 import { WorkerUtils
} from
'./WorkerUtils';
10 export default class WorkerStaticPool
extends WorkerAbstract
<WorkerData
> {
11 private readonly pool
: FixedThreadPool
<WorkerData
>;
14 * Create a new `WorkerStaticPool`.
16 * @param workerScript -
17 * @param workerOptions -
19 constructor(workerScript
: string, workerOptions
?: WorkerOptions
) {
20 super(workerScript
, workerOptions
);
21 this.workerOptions
.poolOptions
.errorHandler
= (
22 this.workerOptions
?.poolOptions
?.errorHandler
?? WorkerUtils
.defaultErrorHandler
23 ).bind(this) as ErrorHandler
<Worker
>;
24 this.workerOptions
.poolOptions
.exitHandler
= (
25 this.workerOptions
?.poolOptions
?.exitHandler
?? WorkerUtils
.defaultExitHandler
26 ).bind(this) as ExitHandler
<Worker
>;
27 this.workerOptions
.poolOptions
.messageHandler
.bind(this);
28 this.pool
= new FixedThreadPool(
29 this.workerOptions
.poolMaxSize
,
31 this.workerOptions
.poolOptions
36 return this.pool
.workers
.length
;
39 get
maxElementsPerWorker(): number | null {
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 Utils
.sleep(this.workerOptions
.elementStartDelay
));