X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fcharging-station%2FBootstrap.ts;h=6620764559974d12e6ca8a89cf7ccae497937e22;hb=ee7c1da0e12c70134f31537a1c1e7040d309af5a;hp=0dbf95106afbd57d7ae2609b40d5084033f12437;hpb=f6591eb9f063490a9b4950ff782c1026281e81bc;p=e-mobility-charging-stations-simulator.git diff --git a/src/charging-station/Bootstrap.ts b/src/charging-station/Bootstrap.ts index 0dbf9510..66207645 100644 --- a/src/charging-station/Bootstrap.ts +++ b/src/charging-station/Bootstrap.ts @@ -2,7 +2,7 @@ import { EventEmitter } from 'node:events'; import { dirname, extname, join } from 'node:path'; -import { exit } from 'node:process'; +import process, { exit } from 'node:process'; import { fileURLToPath } from 'node:url'; import chalk from 'chalk'; @@ -55,21 +55,20 @@ export class Bootstrap extends EventEmitter { private static instance: Bootstrap | null = null; public numberOfChargingStations!: number; public numberOfChargingStationTemplates!: number; - private workerImplementation: WorkerAbstract | null; - private readonly uiServer!: AbstractUIServer | null; - private readonly storage!: Storage; + private workerImplementation?: WorkerAbstract; + private readonly uiServer?: AbstractUIServer; + private storage?: Storage; private numberOfStartedChargingStations!: number; private readonly version: string = version; private initializedCounters: boolean; private started: boolean; private starting: boolean; private stopping: boolean; - private readonly workerScript: string; private constructor() { super(); for (const signal of ['SIGINT', 'SIGQUIT', 'SIGTERM']) { - process.on(signal, this.gracefulShutdown); + process.on(signal, this.gracefulShutdown.bind(this)); } // Enable unconditionally for now handleUnhandledRejection(); @@ -79,28 +78,10 @@ export class Bootstrap extends EventEmitter { this.stopping = false; this.initializedCounters = false; this.initializeCounters(); - this.workerImplementation = null; - this.workerScript = join( - dirname(fileURLToPath(import.meta.url)), - `ChargingStationWorker${extname(fileURLToPath(import.meta.url))}`, + this.uiServer = UIServerFactory.getUIServerImplementation( + Configuration.getConfigurationSection(ConfigurationSection.uiServer), ); - const uiServerConfiguration = Configuration.getConfigurationSection( - ConfigurationSection.uiServer, - ); - uiServerConfiguration.enabled === true && - (this.uiServer = UIServerFactory.getUIServerImplementation(uiServerConfiguration)); - const performanceStorageConfiguration = - Configuration.getConfigurationSection( - ConfigurationSection.performanceStorage, - ); - performanceStorageConfiguration.enabled === true && - (this.storage = StorageFactory.getStorage( - performanceStorageConfiguration.type!, - performanceStorageConfiguration.uri!, - this.logPrefix(), - )); - // FIXME: Disabled until the spurious configuration file change detection on MacOS is identified - // Configuration.configurationChangeCallback = async () => Bootstrap.getInstance().restart(false); + Configuration.configurationChangeCallback = async () => Bootstrap.getInstance().restart(false); } public static getInstance(): Bootstrap { @@ -114,14 +95,33 @@ export class Bootstrap extends EventEmitter { if (this.started === false) { if (this.starting === false) { this.starting = true; + this.on(ChargingStationWorkerMessageEvents.started, this.workerEventStarted); + this.on(ChargingStationWorkerMessageEvents.stopped, this.workerEventStopped); + this.on(ChargingStationWorkerMessageEvents.updated, this.workerEventUpdated); + this.on( + ChargingStationWorkerMessageEvents.performanceStatistics, + this.workerEventPerformanceStatistics, + ); this.initializeCounters(); const workerConfiguration = Configuration.getConfigurationSection( ConfigurationSection.worker, ); this.initializeWorkerImplementation(workerConfiguration); await this.workerImplementation?.start(); - await this.storage?.open(); - this.uiServer?.start(); + const performanceStorageConfiguration = + Configuration.getConfigurationSection( + ConfigurationSection.performanceStorage, + ); + if (performanceStorageConfiguration.enabled === true) { + this.storage = StorageFactory.getStorage( + performanceStorageConfiguration.type!, + performanceStorageConfiguration.uri!, + this.logPrefix(), + ); + await this.storage?.open(); + } + Configuration.getConfigurationSection(ConfigurationSection.uiServer) + .enabled === true && this.uiServer?.start(); // Start ChargingStation object instance in worker thread for (const stationTemplateUrl of Configuration.getStationTemplateUrls()!) { try { @@ -174,39 +174,30 @@ export class Bootstrap extends EventEmitter { } } - public async stop(waitChargingStationsStopped = true): Promise { + public async stop(stopChargingStations = true): Promise { if (this.started === true) { if (this.stopping === false) { this.stopping = true; - await this.uiServer?.sendInternalRequest( - this.uiServer.buildProtocolRequest( - generateUUID(), - ProcedureName.STOP_CHARGING_STATION, - Constants.EMPTY_FROZEN_OBJECT, - ), - ); - if (waitChargingStationsStopped === true) { - await Promise.race([ - waitChargingStationEvents( - this, - ChargingStationWorkerMessageEvents.stopped, - this.numberOfChargingStations, + if (stopChargingStations === true) { + await this.uiServer?.sendInternalRequest( + this.uiServer.buildProtocolRequest( + generateUUID(), + ProcedureName.STOP_CHARGING_STATION, + Constants.EMPTY_FROZEN_OBJECT, ), - new Promise((resolve) => { - setTimeout(() => { - const message = `Timeout ${formatDurationMilliSeconds( - Constants.STOP_SIMULATOR_TIMEOUT, - )} reached at stopping charging stations simulator`; - console.warn(chalk.yellow(message)); - resolve(message); - }, Constants.STOP_SIMULATOR_TIMEOUT); - }), - ]); + ); + try { + await this.waitChargingStationsStopped(); + } catch (error) { + console.error(chalk.red('Error while waiting for charging stations to stop: '), error); + } } await this.workerImplementation?.stop(); - this.workerImplementation = null; + delete this.workerImplementation; + this.removeAllListeners(); this.uiServer?.stop(); await this.storage?.close(); + delete this.storage; this.resetCounters(); this.initializedCounters = false; this.started = false; @@ -219,34 +210,66 @@ export class Bootstrap extends EventEmitter { } } - public async restart(waitChargingStationsStopped?: boolean): Promise { - await this.stop(waitChargingStationsStopped); + public async restart(stopChargingStations?: boolean): Promise { + await this.stop(stopChargingStations); await this.start(); } + private async waitChargingStationsStopped(): Promise { + return new Promise((resolve, reject) => { + const waitTimeout = setTimeout(() => { + const message = `Timeout ${formatDurationMilliSeconds( + Constants.STOP_CHARGING_STATIONS_TIMEOUT, + )} reached at stopping charging stations`; + console.warn(chalk.yellow(message)); + reject(new Error(message)); + }, Constants.STOP_CHARGING_STATIONS_TIMEOUT); + waitChargingStationEvents( + this, + ChargingStationWorkerMessageEvents.stopped, + this.numberOfChargingStations, + ) + .then(() => { + resolve('Charging stations stopped'); + }) + .catch(reject) + .finally(() => { + clearTimeout(waitTimeout); + }); + }); + } + private initializeWorkerImplementation(workerConfiguration: WorkerConfiguration): void { let elementsPerWorker: number | undefined; - if (workerConfiguration?.elementsPerWorker === 'auto') { - elementsPerWorker = - this.numberOfChargingStations > availableParallelism() - ? Math.round(this.numberOfChargingStations / (availableParallelism() * 1.5)) - : 1; + switch (workerConfiguration?.elementsPerWorker) { + case 'auto': + elementsPerWorker = + this.numberOfChargingStations > availableParallelism() + ? Math.round(this.numberOfChargingStations / (availableParallelism() * 1.5)) + : 1; + break; + case 'all': + elementsPerWorker = this.numberOfChargingStations; + break; } - this.workerImplementation === null && - (this.workerImplementation = WorkerFactory.getWorkerImplementation( - this.workerScript, - workerConfiguration.processType!, - { - workerStartDelay: workerConfiguration.startDelay, - elementStartDelay: workerConfiguration.elementStartDelay, - poolMaxSize: workerConfiguration.poolMaxSize!, - poolMinSize: workerConfiguration.poolMinSize!, - elementsPerWorker: elementsPerWorker ?? (workerConfiguration.elementsPerWorker as number), - poolOptions: { - messageHandler: this.messageHandler.bind(this) as (message: unknown) => void, - }, + this.workerImplementation = WorkerFactory.getWorkerImplementation( + join( + dirname(fileURLToPath(import.meta.url)), + `ChargingStationWorker${extname(fileURLToPath(import.meta.url))}`, + ), + workerConfiguration.processType!, + { + workerStartDelay: workerConfiguration.startDelay, + elementStartDelay: workerConfiguration.elementStartDelay, + poolMaxSize: workerConfiguration.poolMaxSize!, + poolMinSize: workerConfiguration.poolMinSize!, + elementsPerWorker: elementsPerWorker ?? (workerConfiguration.elementsPerWorker as number), + poolOptions: { + messageHandler: this.messageHandler.bind(this) as (message: unknown) => void, + workerOptions: { resourceLimits: workerConfiguration.resourceLimits }, }, - )); + }, + ); } private messageHandler( @@ -262,19 +285,15 @@ export class Bootstrap extends EventEmitter { try { switch (msg.event) { case ChargingStationWorkerMessageEvents.started: - this.workerEventStarted(msg.data as ChargingStationData); this.emit(ChargingStationWorkerMessageEvents.started, msg.data as ChargingStationData); break; case ChargingStationWorkerMessageEvents.stopped: - this.workerEventStopped(msg.data as ChargingStationData); this.emit(ChargingStationWorkerMessageEvents.stopped, msg.data as ChargingStationData); break; case ChargingStationWorkerMessageEvents.updated: - this.workerEventUpdated(msg.data as ChargingStationData); this.emit(ChargingStationWorkerMessageEvents.updated, msg.data as ChargingStationData); break; case ChargingStationWorkerMessageEvents.performanceStatistics: - this.workerEventPerformanceStatistics(msg.data as Statistics); this.emit( ChargingStationWorkerMessageEvents.performanceStatistics, msg.data as Statistics, @@ -335,7 +354,7 @@ export class Bootstrap extends EventEmitter { }; private workerEventPerformanceStatistics = (data: Statistics) => { - this.storage.storePerformanceStatistics(data) as void; + this.storage?.storePerformanceStatistics(data) as void; }; private initializeCounters() { @@ -384,17 +403,24 @@ export class Bootstrap extends EventEmitter { }); } - private gracefulShutdown = (): void => { + private gracefulShutdown(): void { this.stop() .then(() => { console.info(`${chalk.green('Graceful shutdown')}`); - exit(exitCodes.succeeded); + // stop() asks for charging stations to stop by default + this.waitChargingStationsStopped() + .then(() => { + exit(exitCodes.succeeded); + }) + .catch(() => { + exit(exitCodes.gracefulShutdownError); + }); }) .catch((error) => { console.error(chalk.red('Error while shutdowning charging stations simulator: '), error); exit(exitCodes.gracefulShutdownError); }); - }; + } private logPrefix = (): string => { return logPrefix(' Bootstrap |');