import Configuration from './utils/Configuration';
-import { StationTemplateURL } from './types/ConfigurationData';
+import Constants from './utils/Constants';
import Utils from './utils/Utils';
-import Wrk from './charging-station/Worker';
import WorkerData from './types/WorkerData';
-import fs from 'fs';
+import WorkerGroup from './charging-station/WorkerGroup';
+import WorkerPool from './charging-station/WorkerPool';
class Bootstrap {
static async start() {
try {
let numStationsTotal = 0;
let numConcurrentWorkers = 0;
- let worker: Wrk;
- let chargingStationsPerWorker = Configuration.getChargingStationsPerWorker();
- let counter = 0;
+ const chargingStationsPerWorker = Configuration.getChargingStationsPerWorker();
+ let chargingStationsPerWorkerCounter = 0;
+ let workerImplementation: WorkerGroup | WorkerPool;
+ if (Configuration.useWorkerPool()) {
+ workerImplementation = new WorkerPool('./dist/charging-station/StationWorker.js');
+ void workerImplementation.start();
+ }
// Start each ChargingStation object in a worker thread
if (Configuration.getStationTemplateURLs()) {
- for await (const stationURL of Configuration.getStationTemplateURLs()) {
+ for (const stationURL of Configuration.getStationTemplateURLs()) {
try {
const nbStations = stationURL.numberOfStations ? stationURL.numberOfStations : 0;
- numStationsTotal += nbStations;
for (let index = 1; index <= nbStations; index++) {
- const workerData = {
+ const workerData: WorkerData = {
index,
templateFile: stationURL.file
- } as WorkerData;
- if(counter === 0 || counter === chargingStationsPerWorker) {
- // Start new worker with one charging station
- worker = await new Wrk('./dist/charging-station/StationWorker.js', workerData, numStationsTotal);
- worker.start().catch(() => {});
- counter = 0;
- // Start workers sequentially to optimize memory at start time
- await Utils.sleep(500);
+ };
+ if (Configuration.useWorkerPool()) {
+ void workerImplementation.addElement(workerData);
+ numConcurrentWorkers = workerImplementation.size;
+ numStationsTotal = workerImplementation.size;
+ // Start worker sequentially to optimize memory at start time
+ await Utils.sleep(Constants.START_WORKER_DELAY);
} else {
- // Add new charging station to existing Worker
- worker.startNewChargingStation(workerData, numStationsTotal)
+ // eslint-disable-next-line no-lonely-if
+ if (chargingStationsPerWorkerCounter === 0 || chargingStationsPerWorkerCounter >= chargingStationsPerWorker) {
+ // Start new WorkerGroup with one charging station
+ workerImplementation = new WorkerGroup('./dist/charging-station/StationWorker.js', workerData, chargingStationsPerWorker);
+ void workerImplementation.start();
+ numConcurrentWorkers++;
+ chargingStationsPerWorkerCounter = 1;
+ numStationsTotal++;
+ // Start worker sequentially to optimize memory at start time
+ await Utils.sleep(Constants.START_WORKER_DELAY);
+ } else {
+ // Add charging station to existing WorkerGroup
+ void workerImplementation.addElement(workerData);
+ chargingStationsPerWorkerCounter++;
+ numStationsTotal++;
+ }
}
- counter++;
- // Start charging station sequentially to optimize memory at start time
- numConcurrentWorkers = worker.concurrentWorkers;
}
} catch (error) {
// eslint-disable-next-line no-console
}
if (numStationsTotal === 0) {
console.log('No charging station template enabled in configuration, exiting');
+ } else if (Configuration.useWorkerPool()) {
+ console.log('Charging station simulator started with ' + numStationsTotal.toString() + ' charging station(s) and ' + numConcurrentWorkers.toString() + '/' + Configuration.getWorkerPoolMaxSize().toString() + ' worker(s) concurrently running');
} else {
- console.log('Charging station simulator started with ' + numStationsTotal.toString() + ' charging station(s) of ' + numConcurrentWorkers.toString() + ' concurrently running');
+ console.log('Charging station simulator started with ' + numStationsTotal.toString() + ' charging station(s) and ' + numConcurrentWorkers.toString() + ' worker(s) concurrently running');
}
} catch (error) {
// eslint-disable-next-line no-console
}
}
-Bootstrap.start();
+Bootstrap.start().catch(
+ (error) => {
+ console.error(error);
+ }
+);