1 import { Worker
, isMainThread
} from
'worker_threads';
2 import { WorkerOptions
, WorkerProcessType
} from
'../types/Worker';
4 import Constants from
'../utils/Constants';
5 import { PoolOptions
} from
'poolifier';
6 import WorkerAbstract from
'./WorkerAbstract';
7 import WorkerDynamicPool from
'./WorkerDynamicPool';
8 import WorkerSet from
'./WorkerSet';
9 import WorkerStaticPool from
'./WorkerStaticPool';
11 export default class WorkerFactory
{
12 private constructor() {
13 // This is intentional
16 public static getWorkerImplementation
<T
>(workerScript
: string, workerProcessType
: WorkerProcessType
, options
?: WorkerOptions
): WorkerAbstract
| null {
18 throw new Error('Trying to get a worker implementation outside the main thread');
20 options
= options
?? {} as WorkerOptions
;
21 options
.startDelay
= options
?.startDelay
?? Constants
.WORKER_START_DELAY
;
22 options
.poolOptions
= options
?.poolOptions
?? {} as PoolOptions
<Worker
>;
23 options
?.messageHandler
&& (options
.poolOptions
.messageHandler
= options
.messageHandler
);
24 let workerImplementation
: WorkerAbstract
= null;
25 switch (workerProcessType
) {
26 case WorkerProcessType
.WORKER_SET
:
27 options
.elementsPerWorker
= options
.elementsPerWorker
?? Constants
.DEFAULT_CHARGING_STATIONS_PER_WORKER
;
28 workerImplementation
= new WorkerSet
<T
>(workerScript
, options
.elementsPerWorker
, options
.startDelay
, options
);
30 case WorkerProcessType
.STATIC_POOL
:
31 options
.poolMaxSize
= options
.poolMaxSize
?? Constants
.DEFAULT_WORKER_POOL_MAX_SIZE
;
32 workerImplementation
= new WorkerStaticPool
<T
>(workerScript
, options
.poolMaxSize
, options
.startDelay
, options
.poolOptions
);
34 case WorkerProcessType
.DYNAMIC_POOL
:
35 options
.poolMinSize
= options
.poolMinSize
?? Constants
.DEFAULT_WORKER_POOL_MIN_SIZE
;
36 options
.poolMaxSize
= options
.poolMaxSize
?? Constants
.DEFAULT_WORKER_POOL_MAX_SIZE
;
37 workerImplementation
= new WorkerDynamicPool
<T
>(workerScript
, options
.poolMinSize
, options
.poolMaxSize
, options
.startDelay
, options
.poolOptions
);
40 throw new Error(`Worker implementation type '${workerProcessType}' not found`);
42 return workerImplementation
;