src/worker/WorkerFactory.ts

   1 import { WorkerOptions, WorkerProcessType } from '../types/Worker';
   2
   3 import Utils from '../utils/Utils';
   4 import WorkerDynamicPool from './WorkerDynamicPool';
   5 import WorkerSet from './WorkerSet';
   6 import WorkerStaticPool from './WorkerStaticPool';
   7 import Wrk from './Wrk';
   8
   9 export default class WorkerFactory {
  10   public static getWorkerImpl<T>(workerScript: string, workerProcessType: WorkerProcessType, options?: WorkerOptions): Wrk {
  11     if (Utils.isUndefined(options)) {
  12       options = {} as WorkerOptions;
  13     }
  14     switch (workerProcessType) {
  15       case WorkerProcessType.WORKER_SET:
  16         if (Utils.isUndefined(options.elementsPerWorker)) {
  17           options.elementsPerWorker = 1;
  18         }
  19         return new WorkerSet<T>(workerScript, options.elementsPerWorker);
  20       case WorkerProcessType.STATIC_POOL:
  21         if (Utils.isUndefined(options.poolMaxSize)) {
  22           options.elementsPerWorker = 16;
  23         }
  24         return new WorkerStaticPool<T>(workerScript, options.poolMaxSize);
  25       case WorkerProcessType.DYNAMIC_POOL:
  26         if (Utils.isUndefined(options.poolMinSize)) {
  27           options.elementsPerWorker = 4;
  28         }
  29         if (Utils.isUndefined(options.poolMaxSize)) {
  30           options.elementsPerWorker = 16;
  31         }
  32         return new WorkerDynamicPool<T>(workerScript, options.poolMinSize, options.poolMaxSize);
  33       default:
  34         return null;
  35     }
  36   }
  37 }