// 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 { Configuration } from '../utils';
import { type WorkerMessage, WorkerMessageEvents } from '../worker';
-const moduleName = 'ChargingStationWorker';
-
/**
* Creates and starts a charging station instance
*
- * @param data - workerData
+ * @param data - data sent to worker
*/
const startChargingStation = (data?: ChargingStationWorkerData): void => {
new ChargingStation(data!.index, data!.templateFile).start();
};
-class ChargingStationWorker extends AsyncResource {
+class ChargingStationWorker<Data extends ChargingStationWorkerData> {
constructor() {
- super(moduleName);
// Add message listener to create and start charging station from the main thread
- parentPort?.on('message', (message: WorkerMessage<ChargingStationWorkerData>) => {
+ parentPort?.on('message', (message: WorkerMessage<Data>) => {
switch (message.event) {
case WorkerMessageEvents.startWorkerElement:
try {
- this.runInAsyncScope(
- startChargingStation.bind(this) as (data?: ChargingStationWorkerData) => void,
- this,
- message.data,
- );
+ startChargingStation(message.data);
parentPort?.postMessage({
event: WorkerMessageEvents.startedWorkerElement,
});
parentPort?.postMessage({
event: WorkerMessageEvents.startWorkerElementError,
data: {
+ name: (error as Error).name,
message: (error as Error).message,
stack: (error as Error).stack,
},
}
}
-export let chargingStationWorker: ChargingStationWorker | ThreadWorker<ChargingStationWorkerData>;
+export let chargingStationWorker:
+ | ChargingStationWorker<ChargingStationWorkerData>
+ | ThreadWorker<ChargingStationWorkerData>;
if (Configuration.workerPoolInUse()) {
chargingStationWorker = new ThreadWorker<ChargingStationWorkerData>(startChargingStation);
} else {
- chargingStationWorker = new ChargingStationWorker();
+ chargingStationWorker = new ChargingStationWorker<ChargingStationWorkerData>();
}