src/worker/WorkerStaticPool.ts

   1 import { FixedThreadPool, PoolOptions } from 'poolifier';
   2
   3 import Utils from '../utils/Utils';
   4 import { Worker } from 'worker_threads';
   5 import WorkerAbstract from './WorkerAbstract';
   6 import { WorkerData } from '../types/Worker';
   7 import { WorkerUtils } from './WorkerUtils';
   8
   9 export default class WorkerStaticPool<T> extends WorkerAbstract {
  10   private pool: FixedThreadPool<WorkerData>;
  11
  12   /**
  13    * Create a new `WorkerStaticPool`.
  14    *
  15    * @param workerScript
  16    * @param numberOfThreads
  17    * @param startWorkerDelay
  18    * @param opts
  19    */
  20   constructor(workerScript: string, numberOfThreads: number, startWorkerDelay?: number, opts?: PoolOptions<Worker>) {
  21     super(workerScript, startWorkerDelay);
  22     opts.exitHandler = opts?.exitHandler ?? WorkerUtils.defaultExitHandler;
  23     this.pool = new FixedThreadPool(numberOfThreads, this.workerScript, opts);
  24   }
  25
  26   get size(): number {
  27     return this.pool.workers.length;
  28   }
  29
  30   get maxElementsPerWorker(): number | null {
  31     return null;
  32   }
  33
  34   /**
  35    *
  36    * @returns
  37    * @public
  38    */
  39   // eslint-disable-next-line @typescript-eslint/no-empty-function
  40   public async start(): Promise<void> {
  41     // This is intentional
  42   }
  43
  44   /**
  45    *
  46    * @returns
  47    * @public
  48    */
  49   public async stop(): Promise<void> {
  50     return this.pool.destroy();
  51   }
  52
  53   /**
  54    *
  55    * @param elementData
  56    * @returns
  57    * @public
  58    */
  59   public async addElement(elementData: T): Promise<void> {
  60     await this.pool.execute(elementData);
  61     // Start worker sequentially to optimize memory at startup
  62     await Utils.sleep(this.workerStartDelay);
  63   }
  64 }