-// Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved.
+// Partial Copyright Jerome Benoit. 2021-2024. All Rights Reserved.
-import { AsyncResource } from 'node:async_hooks';
-import { parentPort, workerData } from 'node:worker_threads';
+import { parentPort } from 'node:worker_threads'
-import { ThreadWorker } from 'poolifier';
+import { ThreadWorker } from 'poolifier'
-import { ChargingStation } from './ChargingStation';
-import type { ChargingStationWorkerData } from '../types';
-import { Configuration } from '../utils';
-import { WorkerConstants, type WorkerMessage, WorkerMessageEvents } from '../worker';
+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'
-/**
- * Create and start a charging station instance
- *
- * @param data - workerData
- */
-const startChargingStation = (data: ChargingStationWorkerData): void => {
- new ChargingStation(data.index, data.templateFile).start();
-};
-
-// Conditionally export ThreadWorker instance for pool usage
-export let threadWorker: ThreadWorker;
+export let chargingStationWorker: object
if (Configuration.workerPoolInUse()) {
- threadWorker = new ThreadWorker<ChargingStationWorkerData>(startChargingStation, {
- maxInactiveTime: WorkerConstants.POOL_MAX_INACTIVE_TIME,
- });
+ 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 {
- class ChargingStationWorker extends AsyncResource {
- constructor() {
- super('ChargingStationWorker');
- }
-
- public run(data: ChargingStationWorkerData): void {
- this.runInAsyncScope(
- startChargingStation.bind(this) as (data: ChargingStationWorkerData) => void,
- this,
- data
- );
- }
- }
- // Add message listener to create and start charging station from the main thread
- parentPort?.on('message', (message: WorkerMessage<ChargingStationWorkerData>) => {
- if (message.id === WorkerMessageEvents.startWorkerElement) {
- new ChargingStationWorker().run(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
+ )}'`
+ )
+ }
+ }
+ })
}
- });
- if (workerData !== undefined) {
- new ChargingStationWorker().run(workerData as ChargingStationWorkerData);
}
+ chargingStationWorker = new ChargingStationWorker<ChargingStationWorkerData>()
}