-// Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved.
+// Partial Copyright Jerome Benoit. 2021-2024. All Rights Reserved.
-import { parentPort, workerData } from 'worker_threads';
+import { parentPort } from 'node:worker_threads'
-import { ThreadWorker } from 'poolifier';
+import { ThreadWorker } from 'poolifier'
-import type { ChargingStationWorkerData } from '../types/ChargingStationWorker';
-import { type WorkerMessage, WorkerMessageEvents } from '../types/Worker';
-import Utils from '../utils/Utils';
-import WorkerConstants from '../worker/WorkerConstants';
-import ChargingStation from './ChargingStation';
-import { ChargingStationUtils } from './ChargingStationUtils';
+import { BaseError } from '../exception/index.js'
+import type { ChargingStationInfo, ChargingStationWorkerData } from '../types/index.js'
+import { Configuration } from '../utils/index.js'
+import { type WorkerDataError, type WorkerMessage, WorkerMessageEvents } from '../worker/index.js'
+import { ChargingStation } from './ChargingStation.js'
-// Conditionally export ThreadWorker instance for pool usage
-export let threadWorker: ThreadWorker;
-if (ChargingStationUtils.workerPoolInUse()) {
- threadWorker = new ThreadWorker<ChargingStationWorkerData>(startChargingStation, {
- maxInactiveTime: WorkerConstants.POOL_MAX_INACTIVE_TIME,
- async: false,
- });
+export let chargingStationWorker: object
+if (Configuration.workerPoolInUse()) {
+ chargingStationWorker = new ThreadWorker<
+ ChargingStationWorkerData,
+ ChargingStationInfo | undefined
+ >((data?: ChargingStationWorkerData): ChargingStationInfo | undefined => {
+ // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+ const { index, templateFile, options } = data!
+ return new ChargingStation(index, templateFile, options).stationInfo
+ })
} else {
- // Add message listener to start charging station from main thread
- addMessageListener();
- if (Utils.isUndefined(workerData) === false) {
- startChargingStation(workerData as ChargingStationWorkerData);
- }
-}
-
-/**
- * Listen messages send by the main thread
- */
-function addMessageListener(): void {
- parentPort?.on('message', (message: WorkerMessage<ChargingStationWorkerData>) => {
- if (message.id === WorkerMessageEvents.START_WORKER_ELEMENT) {
- startChargingStation(message.data);
+ // eslint-disable-next-line @typescript-eslint/no-extraneous-class
+ class ChargingStationWorker<Data extends ChargingStationWorkerData> {
+ constructor () {
+ parentPort?.on('message', (message: WorkerMessage<Data>) => {
+ const { uuid, event, data } = message
+ // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
+ if (uuid != null) {
+ switch (event) {
+ case WorkerMessageEvents.addWorkerElement:
+ try {
+ const chargingStation = new ChargingStation(
+ data.index,
+ data.templateFile,
+ data.options
+ )
+ parentPort?.postMessage({
+ uuid,
+ event: WorkerMessageEvents.addedWorkerElement,
+ // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+ data: chargingStation.stationInfo!
+ } satisfies WorkerMessage<ChargingStationInfo>)
+ } catch (error) {
+ parentPort?.postMessage({
+ uuid,
+ event: WorkerMessageEvents.workerElementError,
+ data: {
+ event,
+ name: (error as Error).name,
+ message: (error as Error).message,
+ stack: (error as Error).stack
+ }
+ } satisfies WorkerMessage<WorkerDataError>)
+ }
+ break
+ default:
+ throw new BaseError(
+ `Unknown worker message event: '${event}' received with data: '${JSON.stringify(
+ data,
+ undefined,
+ 2
+ )}'`
+ )
+ }
+ }
+ })
}
- });
-}
-
-/**
- * Create and start a charging station instance
- *
- * @param data - workerData
- */
-function startChargingStation(data: ChargingStationWorkerData): void {
- const station = new ChargingStation(data.index, data.templateFile);
- station.start();
+ }
+ chargingStationWorker = new ChargingStationWorker<ChargingStationWorkerData>()
}