src/worker/WorkerFactory.ts

   1 import { Worker, isMainThread } from 'worker_threads';
   2 import { WorkerData, WorkerOptions, WorkerProcessType } from '../types/Worker';
   3
   4 import { PoolOptions } from 'poolifier';
   5 import type WorkerAbstract from './WorkerAbstract';
   6 import WorkerConstants from './WorkerConstants';
   7 import WorkerDynamicPool from './WorkerDynamicPool';
   8 import WorkerSet from './WorkerSet';
   9 import WorkerStaticPool from './WorkerStaticPool';
  10
  11 export default class WorkerFactory {
  12   private constructor() {
  13     // This is intentional
  14   }
  15
  16   public static getWorkerImplementation<T extends WorkerData>(
  17     workerScript: string,
  18     workerProcessType: WorkerProcessType,
  19     workerOptions?: WorkerOptions
  20   ): WorkerAbstract<T> | null {
  21     if (!isMainThread) {
  22       throw new Error('Trying to get a worker implementation outside the main thread');
  23     }
  24     workerOptions = workerOptions ?? ({} as WorkerOptions);
  25     workerOptions.workerStartDelay =
  26       workerOptions?.workerStartDelay ?? WorkerConstants.DEFAULT_WORKER_START_DELAY;
  27     workerOptions.elementStartDelay =
  28       workerOptions?.elementStartDelay ?? WorkerConstants.DEFAULT_ELEMENT_START_DELAY;
  29     workerOptions.poolOptions = workerOptions?.poolOptions ?? ({} as PoolOptions<Worker>);
  30     workerOptions?.messageHandler &&
  31       // eslint-disable-next-line @typescript-eslint/no-misused-promises
  32       (workerOptions.poolOptions.messageHandler = workerOptions.messageHandler);
  33     console.log('before');
  34     let workerImplementation: WorkerAbstract<T> = null;
  35     console.log(workerImplementation);
  36     switch (workerProcessType) {
  37       case WorkerProcessType.WORKER_SET:
  38         workerOptions.elementsPerWorker =
  39           workerOptions?.elementsPerWorker ?? WorkerConstants.DEFAULT_ELEMENTS_PER_WORKER;
  40         workerImplementation = new WorkerSet(workerScript, workerOptions);
  41         break;
  42       case WorkerProcessType.STATIC_POOL:
  43         workerOptions.poolMaxSize =
  44           workerOptions?.poolMaxSize ?? WorkerConstants.DEFAULT_POOL_MAX_SIZE;
  45         workerImplementation = new WorkerStaticPool(workerScript, workerOptions);
  46         break;
  47       case WorkerProcessType.DYNAMIC_POOL:
  48         workerOptions.poolMinSize =
  49           workerOptions?.poolMinSize ?? WorkerConstants.DEFAULT_POOL_MIN_SIZE;
  50         workerOptions.poolMaxSize =
  51           workerOptions?.poolMaxSize ?? WorkerConstants.DEFAULT_POOL_MAX_SIZE;
  52         workerImplementation = new WorkerDynamicPool(workerScript, workerOptions);
  53         break;
  54       default:
  55         // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
  56         throw new Error(`Worker implementation type '${workerProcessType}' not found`);
  57     }
  58     return workerImplementation;
  59   }
  60 }