-import { isMainThread, parentPort, workerData } from 'worker_threads';
+import { StationWorkerData, WorkerEvents } from '../types/Worker';
+import { parentPort, workerData } from 'worker_threads';
import ChargingStation from './ChargingStation';
import Constants from '../utils/Constants';
+import { ThreadWorker } from 'poolifier';
+import Utils from '../utils/Utils';
-if (!isMainThread) {
- startChargingStation({ index: workerData.index as number, templateFile: workerData.templateFile as string });
-
- // Listener: start new charging station from main thread
- addListener();
+// Conditionally export ThreadWorker instance for pool usage
+export let threadWorker: ThreadWorker;
+if (Utils.workerPoolInUse()) {
+ threadWorker = new ThreadWorker<StationWorkerData>(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 });
+ }
}
-function addListener() {
- parentPort.on('message', (e) => {
- if (e.id === Constants.START_WORKER_ELEMENT) {
- startChargingStation(e.workerData);
+/**
+ * Listen messages send by the main thread
+ */
+function addMessageListener(): void {
+ parentPort.on('message', (message) => {
+ if (message.id === WorkerEvents.START_WORKER_ELEMENT) {
+ startChargingStation(message.workerData);
}
});
}
-function startChargingStation(data: any) {
- const station = new ChargingStation(data.index as number, data.templateFile as string);
+/**
+ * Create and start a charging station instance
+ *
+ * @param {StationWorkerData} data workerData
+ */
+function startChargingStation(data: StationWorkerData): void {
+ const station = new ChargingStation(data.index, data.templateFile);
station.start();
}