feat: restart worker on uncaught exception

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

// Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved.

import { AsyncResource } from 'node:async_hooks';
import { parentPort } from 'node:worker_threads';

import { ThreadWorker } from 'poolifier';

import { ChargingStation } from './ChargingStation';
import type { ChargingStationWorkerData } from '../types';
import { Configuration } from '../utils';
import { WorkerConstants, type WorkerMessage, WorkerMessageEvents } from '../worker';

/**
 * Create and start a charging station instance
 *
 * @param data - workerData
 */
const startChargingStation = (data: ChargingStationWorkerData): void => {
  new ChargingStation(data.index, data.templateFile).start();
};

class ChargingStationWorker extends AsyncResource {
  constructor() {
    super('ChargingStationWorker');
    // Add message listener to create and start charging station from the main thread
    parentPort?.on('message', (message: WorkerMessage<ChargingStationWorkerData>) => {
      if (message.id === WorkerMessageEvents.startWorkerElement) {
        this.run(message.data);
      }
    });
  }

  private run(data: ChargingStationWorkerData): void {
    this.runInAsyncScope(
      startChargingStation.bind(this) as (data: ChargingStationWorkerData) => void,
      this,
      data
    );
  }
}

export let chargingStationWorker: ChargingStationWorker;
// Conditionally export ThreadWorker instance for pool usage
export let threadWorker: ThreadWorker;
if (Configuration.workerPoolInUse()) {
  threadWorker = new ThreadWorker<ChargingStationWorkerData>(startChargingStation, {
    maxInactiveTime: WorkerConstants.POOL_MAX_INACTIVE_TIME,
  });
} else {
  chargingStationWorker = new ChargingStationWorker();
}