src/charging-station/StationWorker.ts: renaming and typing

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

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';

// 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 addMessageListener(): void {
  parentPort.on('message', (message) => {
    if (message.id === WorkerEvents.START_WORKER_ELEMENT) {
      startChargingStation(message.workerData);
    }
  });
}

function startChargingStation(data: StationWorkerData): void {
  const station = new ChargingStation(data.index , data.templateFile);
  station.start();
}