1 // Partial Copyright Jerome Benoit. 2021. All Rights Reserved.
3 import { ChargingStationWorkerData
, WorkerMessage
, WorkerMessageEvents
} from
'../types/Worker';
5 import Configuration from
'../utils/Configuration';
6 import { Storage
} from
'../performance/storage/Storage';
7 import { StorageFactory
} from
'../performance/storage/StorageFactory';
8 import Utils from
'../utils/Utils';
9 import WorkerAbstract from
'../worker/WorkerAbstract';
10 import WorkerFactory from
'../worker/WorkerFactory';
11 import chalk from
'chalk';
12 import { isMainThread
} from
'worker_threads';
13 import path from
'path';
14 import { version
} from
'../../package.json';
16 export default class Bootstrap
{
17 private static instance
: Bootstrap
| null = null;
18 private static workerImplementation
: WorkerAbstract
| null = null;
19 private static storage
: Storage
;
20 private static numberOfChargingStations
: number;
21 private version
: string = version
;
22 private started
: boolean;
23 private workerScript
: string;
25 private constructor() {
27 this.workerScript
= path
.join(path
.resolve(__dirname
, '../'), 'charging-station', 'ChargingStationWorker.js');
28 this.initWorkerImplementation();
29 Bootstrap
.storage
= StorageFactory
.getStorage(Configuration
.getPerformanceStorage().type, Configuration
.getPerformanceStorage().URI
, this.logPrefix());
30 Configuration
.setConfigurationChangeCallback(async () => Bootstrap
.getInstance().restart());
33 public static getInstance(): Bootstrap
{
34 if (!Bootstrap
.instance
) {
35 Bootstrap
.instance
= new Bootstrap();
37 return Bootstrap
.instance
;
40 public async start(): Promise
<void> {
41 if (isMainThread
&& !this.started
) {
43 Bootstrap
.numberOfChargingStations
= 0;
44 await Bootstrap
.storage
.open();
45 await Bootstrap
.workerImplementation
.start();
46 // Start ChargingStation object in worker thread
47 if (Configuration
.getStationTemplateURLs()) {
48 for (const stationURL
of Configuration
.getStationTemplateURLs()) {
50 const nbStations
= stationURL
.numberOfStations
?? 0;
51 for (let index
= 1; index
<= nbStations
; index
++) {
52 const workerData
: ChargingStationWorkerData
= {
54 templateFile
: path
.join(path
.resolve(__dirname
, '../'), 'assets', 'station-templates', path
.basename(stationURL
.file
))
56 await Bootstrap
.workerImplementation
.addElement(workerData
);
57 Bootstrap
.numberOfChargingStations
++;
60 console
.error(chalk
.red('Charging station start with template file ' + stationURL
.file
+ ' error '), error
);
64 console
.warn(chalk
.yellow('No stationTemplateURLs defined in configuration, exiting'));
66 if (Bootstrap
.numberOfChargingStations
=== 0) {
67 console
.warn(chalk
.yellow('No charging station template enabled in configuration, exiting'));
69 console
.log(chalk
.green(`Charging stations simulator ${this.version} started with ${Bootstrap.numberOfChargingStations.toString()} charging station(s) and ${Utils.workerDynamicPoolInUse() ? `${Configuration.getWorkerPoolMinSize().toString()}/` : ''}${Bootstrap.workerImplementation.size}${Utils.workerPoolInUse() ? `/${Configuration.getWorkerPoolMaxSize().toString()}
` : ''} worker(s) concurrently running in '${Configuration.getWorkerProcess()}' mode${Bootstrap.workerImplementation.maxElementsPerWorker ? ` (${Bootstrap.workerImplementation.maxElementsPerWorker} charging station(s) per worker)` : ''}`));
73 console.error(chalk.red('Bootstrap start error '), error);
78 public async stop(): Promise<void> {
79 if (isMainThread && this.started) {
80 await Bootstrap.workerImplementation.stop();
81 await Bootstrap.storage.close();
86 public async restart(): Promise<void> {
88 this.initWorkerImplementation();
92 private initWorkerImplementation(): void {
93 Bootstrap.workerImplementation = WorkerFactory.getWorkerImplementation<ChargingStationWorkerData>(this.workerScript, Configuration.getWorkerProcess(),
95 startDelay: Configuration.getWorkerStartDelay(),
96 poolMaxSize: Configuration.getWorkerPoolMaxSize(),
97 poolMinSize: Configuration.getWorkerPoolMinSize(),
98 elementsPerWorker: Configuration.getChargingStationsPerWorker(),
100 workerChoiceStrategy: Configuration.getWorkerPoolStrategy()
102 messageHandler: async (msg: WorkerMessage) => {
103 if (msg.id === WorkerMessageEvents.PERFORMANCE_STATISTICS) {
104 await Bootstrap.storage.storePerformanceStatistics(msg.data);
110 private logPrefix(): string {
111 return Utils.logPrefix(' Bootstrap |');