-// Partial Copyright Jerome Benoit. 2021. All Rights Reserved.
+// Partial Copyright Jerome Benoit. 2021-2024. All Rights Reserved.
-import {
- ChargingStationWorkerData,
- ChargingStationWorkerMessage,
- ChargingStationWorkerMessageEvents,
-} from '../types/ChargingStationWorker';
-import { parentPort, workerData } from 'worker_threads';
+import { parentPort } from 'node:worker_threads'
-import ChargingStation from './ChargingStation';
-import Constants from '../utils/Constants';
-import { ThreadWorker } from 'poolifier';
-import Utils from '../utils/Utils';
+import { ThreadWorker } from 'poolifier'
-// Conditionally export ThreadWorker instance for pool usage
-export let threadWorker: ThreadWorker;
-if (Utils.workerPoolInUse()) {
- threadWorker = new ThreadWorker<ChargingStationWorkerData>(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,
- });
- }
-}
+import { ChargingStation } from './ChargingStation.js'
+import { BaseError } from '../exception/index.js'
+import type { ChargingStationWorkerData } from '../types/index.js'
+import { Configuration } from '../utils/index.js'
+import { type WorkerMessage, WorkerMessageEvents } from '../worker/index.js'
/**
- * Listen messages send by the main thread
+ * Creates and starts a charging station instance
+ *
+ * @param data - data sent to worker
*/
-function addMessageListener(): void {
- parentPort?.on('message', (message: ChargingStationWorkerMessage) => {
- if (message.id === ChargingStationWorkerMessageEvents.START_WORKER_ELEMENT) {
- startChargingStation(message.data);
- }
- });
+const startChargingStation = (data?: ChargingStationWorkerData): void => {
+ // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+ new ChargingStation(data!.index, data!.templateFile).start()
}
-/**
- * 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();
+// eslint-disable-next-line @typescript-eslint/no-extraneous-class
+class ChargingStationWorker<Data extends ChargingStationWorkerData> {
+ constructor () {
+ // Add message listener to create and start charging station from the main thread
+ parentPort?.on('message', (message: WorkerMessage<Data>) => {
+ switch (message.event) {
+ case WorkerMessageEvents.startWorkerElement:
+ try {
+ startChargingStation(message.data)
+ parentPort?.postMessage({
+ event: WorkerMessageEvents.startedWorkerElement
+ })
+ } catch (error) {
+ parentPort?.postMessage({
+ event: WorkerMessageEvents.startWorkerElementError,
+ data: {
+ name: (error as Error).name,
+ message: (error as Error).message,
+ stack: (error as Error).stack
+ }
+ })
+ }
+ break
+ default:
+ throw new BaseError(
+ `Unknown worker event: '${message.event}' received with data: '${JSON.stringify(
+ message.data,
+ undefined,
+ 2
+ )}'`
+ )
+ }
+ })
+ }
+}
+
+export let chargingStationWorker:
+| ChargingStationWorker<ChargingStationWorkerData>
+| ThreadWorker<ChargingStationWorkerData>
+if (Configuration.workerPoolInUse()) {
+ chargingStationWorker = new ThreadWorker<ChargingStationWorkerData>(startChargingStation)
+} else {
+ chargingStationWorker = new ChargingStationWorker<ChargingStationWorkerData>()
}