Colorize console message ouputs
[e-mobility-charging-stations-simulator.git] / src / charging-station / Bootstrap.ts
CommitLineData
81797102
JB
1import { StationWorkerData, WorkerEvents, WorkerMessage } from '../types/Worker';
2
ded13d97 3import Configuration from '../utils/Configuration';
81797102
JB
4import { Storage } from '../utils/performance-storage/Storage';
5import { StorageFactory } from '../utils/performance-storage/StorageFactory';
ded13d97 6import Utils from '../utils/Utils';
fd1fdf1b 7import WorkerAbstract from '../worker/WorkerAbstract';
ded13d97 8import WorkerFactory from '../worker/WorkerFactory';
8eac9a09 9import chalk from 'chalk';
ded13d97 10import { isMainThread } from 'worker_threads';
bf1866b2 11import path from 'path';
84e52c2c 12import { version } from '../../package.json';
ded13d97
JB
13
14export default class Bootstrap {
535aaa27
JB
15 private static instance: Bootstrap | null = null;
16 private static workerImplementation: WorkerAbstract | null = null;
81797102 17 private static storage: Storage;
aef1b33a 18 private version: string = version;
eb87fe87 19 private started: boolean;
ded13d97 20 private workerScript: string;
ded13d97
JB
21
22 private constructor() {
eb87fe87 23 this.started = false;
bf1866b2 24 this.workerScript = path.join(path.resolve(__dirname, '../'), 'charging-station', 'StationWorker.js');
8df3f0a9 25 this.initWorkerImplementation();
81797102 26 Bootstrap.storage = StorageFactory.getStorage(Configuration.getPerformanceStorage().type, Configuration.getPerformanceStorage().URI, this.logPrefix());
7874b0b1 27 Configuration.setConfigurationChangeCallback(async () => Bootstrap.getInstance().restart());
ded13d97
JB
28 }
29
30 public static getInstance(): Bootstrap {
31 if (!Bootstrap.instance) {
32 Bootstrap.instance = new Bootstrap();
33 }
34 return Bootstrap.instance;
35 }
36
37 public async start(): Promise<void> {
eb87fe87 38 if (isMainThread && !this.started) {
ded13d97
JB
39 try {
40 let numStationsTotal = 0;
535aaa27 41 await Bootstrap.workerImplementation.start();
ded13d97
JB
42 // Start ChargingStation object in worker thread
43 if (Configuration.getStationTemplateURLs()) {
44 for (const stationURL of Configuration.getStationTemplateURLs()) {
45 try {
46 const nbStations = stationURL.numberOfStations ? stationURL.numberOfStations : 0;
47 for (let index = 1; index <= nbStations; index++) {
48 const workerData: StationWorkerData = {
49 index,
bf1866b2 50 templateFile: path.join(path.resolve(__dirname, '../'), 'assets', 'station-templates', path.basename(stationURL.file))
ded13d97 51 };
535aaa27 52 await Bootstrap.workerImplementation.addElement(workerData);
ded13d97
JB
53 numStationsTotal++;
54 }
55 } catch (error) {
8eac9a09 56 console.error(chalk.red('Charging station start with template file ' + stationURL.file + ' error '), error);
ded13d97
JB
57 }
58 }
59 } else {
8eac9a09 60 console.warn(chalk.yellow('No stationTemplateURLs defined in configuration, exiting'));
ded13d97
JB
61 }
62 if (numStationsTotal === 0) {
8eac9a09 63 console.warn(chalk.yellow('No charging station template enabled in configuration, exiting'));
ded13d97 64 } else {
8eac9a09 65 console.log(chalk.green(`Charging station simulator ${this.version} started with ${numStationsTotal.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)` : ''}`));
ded13d97 66 }
eb87fe87 67 this.started = true;
ded13d97 68 } catch (error) {
8eac9a09 69 console.error(chalk.red('Bootstrap start error '), error);
ded13d97
JB
70 }
71 }
72 }
73
74 public async stop(): Promise<void> {
eb87fe87 75 if (isMainThread && this.started) {
535aaa27 76 await Bootstrap.workerImplementation.stop();
ded13d97 77 }
eb87fe87 78 this.started = false;
ded13d97
JB
79 }
80
81 public async restart(): Promise<void> {
82 await this.stop();
535aaa27 83 this.initWorkerImplementation();
ded13d97
JB
84 await this.start();
85 }
86
535aaa27
JB
87 private initWorkerImplementation() {
88 Bootstrap.workerImplementation = WorkerFactory.getWorkerImplementation<StationWorkerData>(this.workerScript, Configuration.getWorkerProcess(),
8df3f0a9
JB
89 {
90 startDelay: Configuration.getWorkerStartDelay(),
91 poolMaxSize: Configuration.getWorkerPoolMaxSize(),
92 poolMinSize: Configuration.getWorkerPoolMinSize(),
93 elementsPerWorker: Configuration.getChargingStationsPerWorker(),
94 poolOptions: {
95 workerChoiceStrategy: Configuration.getWorkerPoolStrategy()
96 }
81797102
JB
97 }, (msg: WorkerMessage) => {
98 if (msg.id === WorkerEvents.PERFORMANCE_STATISTICS) {
99 Bootstrap.storage.storePerformanceStatistics(msg.data);
100 }
535aaa27
JB
101 });
102 if (!Bootstrap.workerImplementation) {
8df3f0a9 103 throw new Error('Worker implementation not found');
ded13d97 104 }
ded13d97 105 }
81797102
JB
106
107 private logPrefix(): string {
108 return Utils.logPrefix(' Bootstrap');
109 }
ded13d97 110}