X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fcharging-station%2FBootstrap.ts;h=88b4b4c42c973193231aa5d9d4af582dd2f61b18;hb=8b7072dca602c80ec6ef9a3463b6d53b43aa8fa5;hp=2bc650cc32ec9116f638b33da3eb1d4e4a361d21;hpb=53e5fd67dea2dd83ec9c34d963dd6e502ec46f5c;p=e-mobility-charging-stations-simulator.git diff --git a/src/charging-station/Bootstrap.ts b/src/charging-station/Bootstrap.ts index 2bc650cc..88b4b4c4 100644 --- a/src/charging-station/Bootstrap.ts +++ b/src/charging-station/Bootstrap.ts @@ -1,43 +1,47 @@ -// Partial Copyright Jerome Benoit. 2021. All Rights Reserved. +// Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved. -import path from 'path'; -import { fileURLToPath } from 'url'; -import { isMainThread } from 'worker_threads'; +import path from 'node:path'; +import { fileURLToPath } from 'node:url'; +import { type Worker, isMainThread } from 'worker_threads'; import chalk from 'chalk'; +import { ChargingStationUtils } from './ChargingStationUtils'; +import type { AbstractUIServer } from './ui-server/AbstractUIServer'; +import UIServerFactory from './ui-server/UIServerFactory'; import { version } from '../../package.json'; import BaseError from '../exception/BaseError'; -import { Storage } from '../performance/storage/Storage'; +import type { Storage } from '../performance/storage/Storage'; import { StorageFactory } from '../performance/storage/StorageFactory'; import { - ChargingStationData, - ChargingStationWorkerData, - ChargingStationWorkerMessage, - ChargingStationWorkerMessageData, + type ChargingStationData, + type ChargingStationWorkerData, + type ChargingStationWorkerMessage, + type ChargingStationWorkerMessageData, ChargingStationWorkerMessageEvents, } from '../types/ChargingStationWorker'; -import { StationTemplateUrl } from '../types/ConfigurationData'; -import Statistics from '../types/Statistics'; -import { ApplicationProtocol } from '../types/UIProtocol'; +import type { StationTemplateUrl } from '../types/ConfigurationData'; +import type { Statistics } from '../types/Statistics'; +import type { MessageHandler } from '../types/Worker'; import Configuration from '../utils/Configuration'; import logger from '../utils/Logger'; import Utils from '../utils/Utils'; -import WorkerAbstract from '../worker/WorkerAbstract'; +import type WorkerAbstract from '../worker/WorkerAbstract'; import WorkerFactory from '../worker/WorkerFactory'; -import { ChargingStationUtils } from './ChargingStationUtils'; -import { AbstractUIServer } from './ui-server/AbstractUIServer'; -import { UIServiceUtils } from './ui-server/ui-services/UIServiceUtils'; -import UIServerFactory from './ui-server/UIServerFactory'; const moduleName = 'Bootstrap'; -export default class Bootstrap { +enum exitCodes { + missingChargingStationsConfiguration = 1, + noChargingStationTemplates = 2, +} + +export class Bootstrap { private static instance: Bootstrap | null = null; - private workerImplementation: WorkerAbstract | null = null; - private readonly uiServer!: AbstractUIServer; + private workerImplementation: WorkerAbstract | null; + private readonly uiServer!: AbstractUIServer | null; private readonly storage!: Storage; - private numberOfChargingStationTemplates!: number; + private numberOfChargingStationTemplates!: number | undefined; private numberOfChargingStations!: number; private numberOfStartedChargingStations!: number; private readonly version: string = version; @@ -46,18 +50,16 @@ export default class Bootstrap { private constructor() { this.started = false; + this.workerImplementation = null; this.workerScript = path.join( path.resolve(path.dirname(fileURLToPath(import.meta.url)), '../'), 'charging-station', - 'ChargingStationWorker' + path.extname(fileURLToPath(import.meta.url)) + `ChargingStationWorker${path.extname(fileURLToPath(import.meta.url))}` ); this.initialize(); - Configuration.getUIServer().enabled && - (this.uiServer = UIServerFactory.getUIServerImplementation(ApplicationProtocol.WS, { - ...Configuration.getUIServer().options, - handleProtocols: UIServiceUtils.handleProtocols, - })); - Configuration.getPerformanceStorage().enabled && + Configuration.getUIServer().enabled === true && + (this.uiServer = UIServerFactory.getUIServerImplementation(Configuration.getUIServer())); + Configuration.getPerformanceStorage().enabled === true && (this.storage = StorageFactory.getStorage( Configuration.getPerformanceStorage().type, Configuration.getPerformanceStorage().uri, @@ -74,14 +76,17 @@ export default class Bootstrap { } public async start(): Promise { - if (isMainThread && !this.started) { + if (isMainThread && this.started === false) { try { + // Enable unconditionally for now + this.logUnhandledRejection(); + this.logUncaughtException(); this.initialize(); await this.storage?.open(); - await this.workerImplementation.start(); + await this.workerImplementation?.start(); this.uiServer?.start(); const stationTemplateUrls = Configuration.getStationTemplateUrls(); - this.numberOfChargingStationTemplates = stationTemplateUrls.length; + this.numberOfChargingStationTemplates = stationTemplateUrls?.length; // Start ChargingStation object in worker thread if (!Utils.isEmptyArray(stationTemplateUrls)) { for (const stationTemplateUrl of stationTemplateUrls) { @@ -93,9 +98,7 @@ export default class Bootstrap { } catch (error) { console.error( chalk.red( - 'Error at starting charging station with template file ' + - stationTemplateUrl.file + - ': ' + `Error at starting charging station with template file ${stationTemplateUrl.file}: ` ), error ); @@ -105,12 +108,13 @@ export default class Bootstrap { console.warn( chalk.yellow("'stationTemplateUrls' not defined or empty in configuration, exiting") ); + process.exit(exitCodes.missingChargingStationsConfiguration); } if (this.numberOfChargingStations === 0) { console.warn( chalk.yellow('No charging station template enabled in configuration, exiting') ); - process.exit(); + process.exit(exitCodes.noChargingStationTemplates); } else { console.info( chalk.green( @@ -118,15 +122,15 @@ export default class Bootstrap { this.version } started with ${this.numberOfChargingStations.toString()} charging station(s) from ${this.numberOfChargingStationTemplates.toString()} configured charging station template(s) and ${ ChargingStationUtils.workerDynamicPoolInUse() - ? `${Configuration.getWorker().poolMinSize.toString()}/` + ? `${Configuration.getWorker().poolMinSize?.toString()}/` : '' - }${this.workerImplementation.size}${ + }${this.workerImplementation?.size}${ ChargingStationUtils.workerPoolInUse() - ? `/${Configuration.getWorker().poolMaxSize.toString()}` + ? `/${Configuration.getWorker().poolMaxSize?.toString()}` : '' } worker(s) concurrently running in '${Configuration.getWorker().processType}' mode${ - this.workerImplementation.maxElementsPerWorker - ? ` (${this.workerImplementation.maxElementsPerWorker} charging station(s) per worker)` + !Utils.isNullOrUndefined(this.workerImplementation?.maxElementsPerWorker) + ? ` (${this.workerImplementation?.maxElementsPerWorker} charging station(s) per worker)` : '' }` ) @@ -134,7 +138,7 @@ export default class Bootstrap { } this.started = true; } catch (error) { - console.error(chalk.red('Bootstrap start error '), error); + console.error(chalk.red('Bootstrap start error: '), error); } } else { console.error(chalk.red('Cannot start an already started charging stations simulator')); @@ -142,15 +146,15 @@ export default class Bootstrap { } public async stop(): Promise { - if (isMainThread && this.started) { - await this.workerImplementation.stop(); + if (isMainThread && this.started === true) { + await this.workerImplementation?.stop(); this.workerImplementation = null; this.uiServer?.stop(); await this.storage?.close(); + this.started = false; } else { - console.error(chalk.red('Trying to stop the charging stations simulator while not started')); + console.error(chalk.red('Cannot stop a not started charging stations simulator')); } - this.started = false; } public async restart(): Promise { @@ -160,7 +164,7 @@ export default class Bootstrap { } private initializeWorkerImplementation(): void { - !this.workerImplementation && + this.workerImplementation === null && (this.workerImplementation = WorkerFactory.getWorkerImplementation( this.workerScript, Configuration.getWorker().processType, @@ -173,9 +177,7 @@ export default class Bootstrap { poolOptions: { workerChoiceStrategy: Configuration.getWorker().poolStrategy, }, - messageHandler: this.messageHandler.bind(this) as ( - msg: ChargingStationWorkerMessage - ) => void, + messageHandler: this.messageHandler.bind(this) as MessageHandler, } )); } @@ -219,19 +221,33 @@ export default class Bootstrap { } } - private workerEventStarted(data: ChargingStationData) { - this.uiServer?.chargingStations.set(data.hashId, data); + private workerEventStarted = (data: ChargingStationData) => { + this.uiServer?.chargingStations.set(data.stationInfo.hashId, data); ++this.numberOfStartedChargingStations; - } + logger.info( + `${this.logPrefix()} ${moduleName}.workerEventStarted: Charging station ${ + data.stationInfo.chargingStationId + } (hashId: ${data.stationInfo.hashId}) started (${ + this.numberOfStartedChargingStations + } started from ${this.numberOfChargingStations})` + ); + }; - private workerEventStopped(data: ChargingStationData) { - this.uiServer?.chargingStations.set(data.hashId, data); + private workerEventStopped = (data: ChargingStationData) => { + this.uiServer?.chargingStations.set(data.stationInfo.hashId, data); --this.numberOfStartedChargingStations; - } + logger.info( + `${this.logPrefix()} ${moduleName}.workerEventStopped: Charging station ${ + data.stationInfo.chargingStationId + } (hashId: ${data.stationInfo.hashId}) stopped (${ + this.numberOfStartedChargingStations + } started from ${this.numberOfChargingStations})` + ); + }; - private workerEventUpdated(data: ChargingStationData) { - this.uiServer?.chargingStations.set(data.hashId, data); - } + private workerEventUpdated = (data: ChargingStationData) => { + this.uiServer?.chargingStations.set(data.stationInfo.hashId, data); + }; private workerEventPerformanceStatistics = (data: Statistics) => { this.storage.storePerformanceStatistics(data) as void; @@ -244,6 +260,18 @@ export default class Bootstrap { this.initializeWorkerImplementation(); } + private logUncaughtException(): void { + process.on('uncaughtException', (error: Error) => { + console.error(chalk.red('Uncaught exception: '), error); + }); + } + + private logUnhandledRejection(): void { + process.on('unhandledRejection', (reason: unknown) => { + console.error(chalk.red('Unhandled rejection: '), reason); + }); + } + private async startChargingStation( index: number, stationTemplateUrl: StationTemplateUrl @@ -257,11 +285,11 @@ export default class Bootstrap { stationTemplateUrl.file ), }; - await this.workerImplementation.addElement(workerData); + await this.workerImplementation?.addElement(workerData); ++this.numberOfChargingStations; } - private logPrefix(): string { + private logPrefix = (): string => { return Utils.logPrefix(' Bootstrap |'); - } + }; }