X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fcharging-station%2FBootstrap.ts;h=df0bb8e98cda7c85b30104ba9ed9fac37fceb610;hb=507a34c154d4877cc261df07b942bcef01b87b4b;hp=66a424193270ebc033f36dc49eda67fb8fc66c72;hpb=d1c99c59787e98b3954760fbc208a5dfbc3d8f57;p=e-mobility-charging-stations-simulator.git diff --git a/src/charging-station/Bootstrap.ts b/src/charging-station/Bootstrap.ts index 66a42419..df0bb8e9 100644 --- a/src/charging-station/Bootstrap.ts +++ b/src/charging-station/Bootstrap.ts @@ -1,33 +1,36 @@ // Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved. +import { EventEmitter } from 'node:events'; import path from 'node:path'; import { fileURLToPath } from 'node:url'; -import { type Worker, isMainThread } from 'worker_threads'; +import { type Worker, isMainThread } from 'node: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 type { Storage } from '../performance/storage/Storage'; -import { StorageFactory } from '../performance/storage/StorageFactory'; +import { UIServerFactory } from './ui-server/UIServerFactory'; +import packageJson from '../../package.json' assert { type: 'json' }; +import { BaseError } from '../exception'; +import { type Storage, StorageFactory } from '../performance'; import { type ChargingStationData, type ChargingStationWorkerData, type ChargingStationWorkerMessage, type ChargingStationWorkerMessageData, ChargingStationWorkerMessageEvents, -} from '../types/ChargingStationWorker'; -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 type WorkerAbstract from '../worker/WorkerAbstract'; -import WorkerFactory from '../worker/WorkerFactory'; + ProcedureName, + type StationTemplateUrl, + type Statistics, +} from '../types'; +import { + Configuration, + Constants, + Utils, + handleUncaughtException, + handleUnhandledRejection, + logger, +} from '../utils'; +import { type MessageHandler, type WorkerAbstract, WorkerFactory } from '../worker'; const moduleName = 'Bootstrap'; @@ -36,7 +39,7 @@ enum exitCodes { noChargingStationTemplates = 2, } -export class Bootstrap { +export class Bootstrap extends EventEmitter { private static instance: Bootstrap | null = null; public numberOfChargingStations!: number; public numberOfChargingStationTemplates!: number; @@ -44,19 +47,31 @@ export class Bootstrap { private readonly uiServer!: AbstractUIServer | null; private readonly storage!: Storage; private numberOfStartedChargingStations!: number; - private readonly version: string = version; + private readonly version: string = packageJson.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); + } + // Enable unconditionally for now + handleUnhandledRejection(); + handleUncaughtException(); this.started = false; + this.starting = false; + this.stopping = false; + this.initializedCounters = false; + this.initializeCounters(); this.workerImplementation = null; this.workerScript = path.join( - path.resolve(path.dirname(fileURLToPath(import.meta.url)), '../'), - 'charging-station', + path.dirname(fileURLToPath(import.meta.url)), `ChargingStationWorker${path.extname(fileURLToPath(import.meta.url))}` ); - this.initialize(); Configuration.getUIServer().enabled === true && (this.uiServer = UIServerFactory.getUIServerImplementation(Configuration.getUIServer())); Configuration.getPerformanceStorage().enabled === true && @@ -76,18 +91,19 @@ export class Bootstrap { } public async start(): Promise { - if (isMainThread && this.started === false) { - try { - // Enable unconditionally for now - this.logUnhandledRejection(); - this.logUncaughtException(); - this.initialize(); - await this.storage?.open(); + if (!isMainThread) { + throw new Error('Cannot start charging stations simulator from worker thread'); + } + if (this.started === false) { + if (this.starting === false) { + this.starting = true; + this.initializeCounters(); + this.initializeWorkerImplementation(); await this.workerImplementation?.start(); + await this.storage?.open(); this.uiServer?.start(); - const stationTemplateUrls = Configuration.getStationTemplateUrls(); - // Start ChargingStation object in worker thread - for (const stationTemplateUrl of stationTemplateUrls) { + // Start ChargingStation object instance in worker thread + for (const stationTemplateUrl of Configuration.getStationTemplateUrls()) { try { const nbStations = stationTemplateUrl.numberOfStations ?? 0; for (let index = 1; index <= nbStations; index++) { @@ -102,35 +118,29 @@ export class Bootstrap { ); } } - if (this.numberOfChargingStations === 0) { - console.warn( - chalk.yellow('No charging station template enabled in configuration, exiting') - ); - process.exit(exitCodes.noChargingStationTemplates); - } else { - console.info( - chalk.green( - `Charging stations simulator ${ - 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()}/` - : '' - }${this.workerImplementation?.size}${ - ChargingStationUtils.workerPoolInUse() - ? `/${Configuration.getWorker().poolMaxSize?.toString()}` - : '' - } worker(s) concurrently running in '${Configuration.getWorker().processType}' mode${ - !Utils.isNullOrUndefined(this.workerImplementation?.maxElementsPerWorker) - ? ` (${this.workerImplementation?.maxElementsPerWorker} charging station(s) per worker)` - : '' - }` - ) - ); - } + console.info( + chalk.green( + `Charging stations simulator ${ + this.version + } started with ${this.numberOfChargingStations.toString()} charging station(s) from ${this.numberOfChargingStationTemplates.toString()} configured charging station template(s) and ${ + Configuration.workerDynamicPoolInUse() + ? `${Configuration.getWorker().poolMinSize?.toString()}/` + : '' + }${this.workerImplementation?.size}${ + Configuration.workerPoolInUse() + ? `/${Configuration.getWorker().poolMaxSize?.toString()}` + : '' + } worker(s) concurrently running in '${Configuration.getWorker().processType}' mode${ + !Utils.isNullOrUndefined(this.workerImplementation?.maxElementsPerWorker) + ? ` (${this.workerImplementation?.maxElementsPerWorker} charging station(s) per worker)` + : '' + }` + ) + ); this.started = true; - } catch (error) { - console.error(chalk.red('Bootstrap start error: '), error); + this.starting = false; + } else { + console.error(chalk.red('Cannot start an already starting charging stations simulator')); } } else { console.error(chalk.red('Cannot start an already started charging stations simulator')); @@ -138,20 +148,38 @@ export class Bootstrap { } public async stop(): Promise { - if (isMainThread && this.started === true) { - await this.workerImplementation?.stop(); - this.workerImplementation = null; - this.uiServer?.stop(); - await this.storage?.close(); - this.started = false; + if (!isMainThread) { + throw new Error('Cannot stop charging stations simulator from worker thread'); + } + if (this.started === true) { + if (this.stopping === false) { + this.stopping = true; + await this.uiServer?.sendInternalRequest( + this.uiServer.buildProtocolRequest( + Utils.generateUUID(), + ProcedureName.STOP_CHARGING_STATION, + Constants.EMPTY_FREEZED_OBJECT + ) + ); + await this.waitForChargingStationsStopped(); + await this.workerImplementation?.stop(); + this.workerImplementation = null; + this.uiServer?.stop(); + await this.storage?.close(); + this.resetCounters(); + this.initializedCounters = false; + this.started = false; + this.stopping = false; + } else { + console.error(chalk.red('Cannot stop an already stopping charging stations simulator')); + } } else { - console.error(chalk.red('Cannot stop a not started charging stations simulator')); + console.error(chalk.red('Cannot stop an already stopped charging stations simulator')); } } public async restart(): Promise { await this.stop(); - this.initialize(); await this.start(); } @@ -186,17 +214,24 @@ export class Bootstrap { // ); try { switch (msg.id) { - case ChargingStationWorkerMessageEvents.STARTED: + case ChargingStationWorkerMessageEvents.started: this.workerEventStarted(msg.data as ChargingStationData); + this.emit(ChargingStationWorkerMessageEvents.started, msg.data as ChargingStationData); break; - case ChargingStationWorkerMessageEvents.STOPPED: + case ChargingStationWorkerMessageEvents.stopped: this.workerEventStopped(msg.data as ChargingStationData); + this.emit(ChargingStationWorkerMessageEvents.stopped, msg.data as ChargingStationData); break; - case ChargingStationWorkerMessageEvents.UPDATED: + case ChargingStationWorkerMessageEvents.updated: this.workerEventUpdated(msg.data as ChargingStationData); + this.emit(ChargingStationWorkerMessageEvents.updated, msg.data as ChargingStationData); break; - case ChargingStationWorkerMessageEvents.PERFORMANCE_STATISTICS: + case ChargingStationWorkerMessageEvents.performanceStatistics: this.workerEventPerformanceStatistics(msg.data as Statistics); + this.emit( + ChargingStationWorkerMessageEvents.performanceStatistics, + msg.data as Statistics + ); break; default: throw new BaseError( @@ -245,53 +280,81 @@ export class Bootstrap { this.storage.storePerformanceStatistics(data) as void; }; - private initialize() { - this.numberOfChargingStationTemplates = 0; - this.numberOfChargingStations = 0; - const stationTemplateUrls = Configuration.getStationTemplateUrls(); - if (!Utils.isEmptyArray(stationTemplateUrls)) { - this.numberOfChargingStationTemplates = stationTemplateUrls?.length; - stationTemplateUrls.forEach((stationTemplateUrl) => { - this.numberOfChargingStations += stationTemplateUrl.numberOfStations ?? 0; - }); - } else { - console.warn( - chalk.yellow("'stationTemplateUrls' not defined or empty in configuration, exiting") - ); - process.exit(exitCodes.missingChargingStationsConfiguration); + private initializeCounters() { + if (this.initializedCounters === false) { + this.resetCounters(); + const stationTemplateUrls = Configuration.getStationTemplateUrls(); + if (Utils.isNotEmptyArray(stationTemplateUrls)) { + this.numberOfChargingStationTemplates = stationTemplateUrls.length; + for (const stationTemplateUrl of stationTemplateUrls) { + this.numberOfChargingStations += stationTemplateUrl.numberOfStations ?? 0; + } + } else { + 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(exitCodes.noChargingStationTemplates); + } + this.initializedCounters = true; } - this.numberOfStartedChargingStations = 0; - 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 resetCounters(): void { + this.numberOfChargingStationTemplates = 0; + this.numberOfChargingStations = 0; + this.numberOfStartedChargingStations = 0; } private async startChargingStation( index: number, stationTemplateUrl: StationTemplateUrl ): Promise { - const workerData: ChargingStationWorkerData = { + await this.workerImplementation?.addElement({ index, templateFile: path.join( - path.resolve(path.dirname(fileURLToPath(import.meta.url)), '../'), + path.dirname(fileURLToPath(import.meta.url)), 'assets', 'station-templates', stationTemplateUrl.file ), - }; - await this.workerImplementation?.addElement(workerData); + }); } + private gracefulShutdown = (): void => { + console.info(`${chalk.green('Graceful shutdown')}`); + this.stop() + .then(() => { + process.exit(0); + }) + .catch((error) => { + console.error(chalk.red('Error while shutdowning charging stations simulator: '), error); + process.exit(1); + }); + }; + + private waitForChargingStationsStopped = async ( + stoppedEventsToWait = this.numberOfStartedChargingStations + ): Promise => { + return new Promise((resolve) => { + let stoppedEvents = 0; + if (stoppedEventsToWait === 0) { + resolve(stoppedEvents); + } + this.on(ChargingStationWorkerMessageEvents.stopped, () => { + ++stoppedEvents; + if (stoppedEvents === stoppedEventsToWait) { + resolve(stoppedEvents); + } + }); + }); + }; + private logPrefix = (): string => { return Utils.logPrefix(' Bootstrap |'); };