Add eslint jsdoc plugin and refine a bit the existing comments.

[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 c7bc2337b0e772c4d73a3712b0606b9981500096..20fb27ca04ae47a533f7109525f07a8e7bba1768 100644 (file)
--- a/src/charging-station/StationWorker.ts
+++ b/src/charging-station/StationWorker.ts
@@ -1,26 +1,38 @@
-import { isMainThread, parentPort, workerData } from 'worker_threads';
-import Constants from '../utils/Constants';
+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';
 
-if (!isMainThread) {
-  const station = new ChargingStation(workerData.index as number, workerData.templateFile as string);
-  station.start();
-
-  // Listener: start new 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() {
-  parentPort.setMaxListeners(1000);
-  parentPort.on("message", e => {
-    if (e.id === Constants.START_NEW_CHARGING_STATION) {
-        startChargingStation(e.workerData);
+/**
+ *
+ */
+function addMessageListener(): void {
+  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);
+/**
+ * @param data
+ */
+function startChargingStation(data: StationWorkerData): void {
+  const station = new ChargingStation(data.index, data.templateFile);
   station.start();
 }