X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fcharging-station%2FStationWorker.ts;h=bb2db88a4c749acafee441291b05dc8ef4fdf36e;hb=ce4d00ee7295ccdb12f35154eb2eb72973b6e75b;hp=c0e85748ec0918b2f184d3561ee1f30100eddfb8;hpb=3e1416d80bd2640a5fe59cb4bc95c421dcd601d6;p=e-mobility-charging-stations-simulator.git

diff --git a/src/charging-station/StationWorker.ts b/src/charging-station/StationWorker.ts
index c0e85748..bb2db88a 100644
--- a/src/charging-station/StationWorker.ts
+++ b/src/charging-station/StationWorker.ts
@@ -1,18 +1,24 @@
-import { isMainThread, parentPort, workerData } from 'worker_threads';
+import { StationWorkerData, WorkerEvents } from '../types/Worker';
+import { parentPort, workerData } from 'worker_threads';
 
 import ChargingStation from './ChargingStation';
+import Constants from '../utils/Constants';
+import { ThreadWorker } from 'poolifier';
 import Utils from '../utils/Utils';
-import { WorkerEvents } from '../types/WorkerEvents';
 
-if (!isMainThread) {
-  // Add listener to start charging station from main thread
-  addListener();
+// Conditionally export ThreadWorker instance for pool usage
+export let threadWorker: ThreadWorker;
+if (Utils.workerPoolInUse()) {
+  threadWorker = new ThreadWorker<StationWorkerData>(startChargingStation, { maxInactiveTime: Constants.WORKER_POOL_MAX_INACTIVE_TIME, async: false });
+} else {
+  // Add message listener to start charging station from main thread
+  addMessageListener();
   if (!Utils.isUndefined(workerData)) {
     startChargingStation({ index: workerData.index as number, templateFile: workerData.templateFile as string });
   }
 }
 
-function addListener() {
+function addMessageListener(): void {
   parentPort.on('message', (message) => {
     if (message.id === WorkerEvents.START_WORKER_ELEMENT) {
       startChargingStation(message.workerData);
@@ -20,7 +26,7 @@ function addListener() {
   });
 }
 
-function startChargingStation(data: any) {
-  const station = new ChargingStation(data.index as number, data.templateFile as string);
+function startChargingStation(data: StationWorkerData): void {
+  const station = new ChargingStation(data.index, data.templateFile);
   station.start();
 }