X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fworker%2FWorkerAbstract.ts;h=3fbd4a0b2da68f7adaafbcfe8cb7a940c04c833a;hb=85c603dabe1b0415dfa956baec73f34267093cf0;hp=7c662d9dead4aa46e519b2519d1e560c774615b8;hpb=c3ee95af4edd541595e3873c8aa4c93d38e59474;p=e-mobility-charging-stations-simulator.git

diff --git a/src/worker/WorkerAbstract.ts b/src/worker/WorkerAbstract.ts
index 7c662d9d..3fbd4a0b 100644
--- a/src/worker/WorkerAbstract.ts
+++ b/src/worker/WorkerAbstract.ts
@@ -1,9 +1,11 @@
-import Constants from '../utils/Constants';
-import { WorkerData } from '../types/Worker';
+import fs from 'fs';
+
+import { WorkerData, WorkerOptions } from '../types/Worker';
+import WorkerConstants from './WorkerConstants';
 
 export default abstract class WorkerAbstract<T extends WorkerData> {
   protected readonly workerScript: string;
-  protected readonly workerStartDelay: number;
+  protected readonly workerOptions: WorkerOptions;
   public abstract readonly size: number;
   public abstract readonly maxElementsPerWorker: number | null;
 
@@ -11,11 +13,30 @@ export default abstract class WorkerAbstract<T extends WorkerData> {
    * `WorkerAbstract` constructor.
    *
    * @param workerScript
-   * @param workerStartDelay
+   * @param workerOptions
    */
-  constructor(workerScript: string, workerStartDelay: number = Constants.WORKER_START_DELAY) {
+  constructor(
+    workerScript: string,
+    workerOptions: WorkerOptions = {
+      workerStartDelay: WorkerConstants.DEFAULT_WORKER_START_DELAY,
+      elementStartDelay: WorkerConstants.DEFAULT_ELEMENT_START_DELAY,
+      poolMinSize: WorkerConstants.DEFAULT_POOL_MIN_SIZE,
+      poolMaxSize: WorkerConstants.DEFAULT_POOL_MAX_SIZE,
+      elementsPerWorker: WorkerConstants.DEFAULT_ELEMENTS_PER_WORKER,
+      poolOptions: {},
+      messageHandler: () => {
+        /* This is intentional */
+      },
+    }
+  ) {
+    if (!workerScript) {
+      throw new Error('Worker script is not defined');
+    }
+    if (!fs.existsSync(workerScript)) {
+      throw new Error('Worker script file does not exist');
+    }
     this.workerScript = workerScript;
-    this.workerStartDelay = workerStartDelay;
+    this.workerOptions = workerOptions;
   }
 
   public abstract start(): Promise<void>;