Fix worker startup and cleanup some attributes in Wrk class.

[e-mobility-charging-stations-simulator.git] / src / charging-station / StationWorker.ts

diff --git a/src/charging-station/StationWorker.ts b/src/charging-station/StationWorker.ts
index f54334e952d4cb86e0f50557209e87757c68b2a1..a948a1b4bbb1bfc1572ffe3dbac8a47d6e6149bc 100644 (file)
--- a/src/charging-station/StationWorker.ts
+++ b/src/charging-station/StationWorker.ts
@@ -1,8 +1,25 @@
-import { isMainThread, workerData } from 'worker_threads';
+import { isMainThread, parentPort, workerData } from 'worker_threads';
 
 import ChargingStation from './ChargingStation';
+import Constants from '../utils/Constants';
 
 if (!isMainThread) {
-  const station = new ChargingStation(workerData.index, workerData.templateFile);
+  startChargingStation({ index: workerData.index as number, templateFile: workerData.templateFile as string });
+
+  // Listener: start new charging station from main thread
+  addListener();
+}
+
+function addListener() {
+  parentPort.setMaxListeners(Constants.MAX_LISTENERS);
+  parentPort.on('message', (e) => {
+    if (e.id === Constants.START_WORKER_ELEMENT) {
+      startChargingStation(e.workerData);
+    }
+  });
+}
+
+function startChargingStation(data: any) {
+  const station = new ChargingStation(data.index as number, data.templateFile as string);
   station.start();
 }