Fix connectors status at reset.

[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 32c75ca79724e114daad68c9cfb08e8459a92d70..65e2fb8832883662aadb145798da9783f37a1ae2 100644 (file)
--- a/src/charging-station/StationWorker.ts
+++ b/src/charging-station/StationWorker.ts
@@ -1,9 +1,17 @@
+import { StationWorkerData, WorkerEvents } from '../types/Worker';
 import { isMainThread, parentPort, workerData } from 'worker_threads';
 
 import ChargingStation from './ChargingStation';
 import Constants from '../utils/Constants';
+import { ThreadWorker } from 'poolifier';
 import Utils from '../utils/Utils';
 
+// Conditionally export ThreadWorker instance for pool usage
+export let threadWorker;
+if (Utils.workerPoolInUse()) {
+  threadWorker = new ThreadWorker(startChargingStation, { maxInactiveTime: Constants.WORKER_POOL_MAX_INACTIVE_TIME, async: false });
+}
+
 if (!isMainThread) {
   // Add listener to start charging station from main thread
   addListener();
@@ -13,14 +21,14 @@ if (!isMainThread) {
 }
 
 function addListener() {
-  parentPort.on('message', (e) => {
-    if (e.id === Constants.START_WORKER_ELEMENT) {
-      startChargingStation(e.workerData);
+  parentPort.on('message', (message) => {
+    if (message.id === WorkerEvents.START_WORKER_ELEMENT) {
+      startChargingStation(message.workerData);
     }
   });
 }
 
-function startChargingStation(data: any) {
-  const station = new ChargingStation(data.index as number, data.templateFile as string);
+function startChargingStation(data: StationWorkerData) {
+  const station = new ChargingStation(data.index , data.templateFile);
   station.start();
 }