feat: add performance statistics to UI protocol

[e-mobility-charging-stations-simulator.git] / src / performance / storage / MikroOrmStorage.ts

// Copyright Jerome Benoit. 2021-2024. All Rights Reserved.

import { MikroORM as MariaDbORM, type Options as MariaDbOptions } from '@mikro-orm/mariadb'
import { MikroORM as SqliteORM, type Options as SqliteOptions } from '@mikro-orm/sqlite'

import { Storage } from './Storage.js'
import { type PerformanceRecord, type Statistics, StorageType } from '../../types/index.js'
import { Constants } from '../../utils/index.js'

export class MikroOrmStorage extends Storage {
  private readonly storageType: StorageType
  private orm?: SqliteORM | MariaDbORM

  constructor (storageUri: string, logPrefix: string, storageType: StorageType) {
    super(storageUri, logPrefix)
    this.storageType = storageType
    this.dbName = this.getDBName()
  }

  public async storePerformanceStatistics (performanceStatistics: Statistics): Promise<void> {
    try {
      this.setPerformanceStatistics(performanceStatistics)
      await this.orm?.em.upsert({
        ...performanceStatistics,
        statisticsData: Array.from(performanceStatistics.statisticsData, ([name, value]) => ({
          name,
          ...value
        }))
      } satisfies PerformanceRecord)
    } catch (error) {
      this.handleDBError(this.storageType, error as Error, Constants.PERFORMANCE_RECORDS_TABLE)
    }
  }

  public async open (): Promise<void> {
    try {
      if (this.orm == null) {
        switch (this.storageType) {
          case StorageType.SQLITE:
            this.orm = await SqliteORM.init(this.getOptions() as SqliteOptions)
            break
          case StorageType.MARIA_DB:
          case StorageType.MYSQL:
            this.orm = await MariaDbORM.init(this.getOptions() as MariaDbOptions)
            break
        }
      }
    } catch (error) {
      this.handleDBError(this.storageType, error as Error)
    }
  }

  public async close (): Promise<void> {
    this.clearPerformanceStatistics()
    try {
      if (this.orm != null) {
        await this.orm.close()
        delete this.orm
      }
    } catch (error) {
      this.handleDBError(this.storageType, error as Error)
    }
  }

  private getDBName (): string {
    if (this.storageType === StorageType.SQLITE) {
      return `${Constants.DEFAULT_PERFORMANCE_DIRECTORY}/${Constants.DEFAULT_PERFORMANCE_RECORDS_DB_NAME}.db`
    }
    return this.storageUri.pathname.replace(/(?:^\/)|(?:\/$)/g, '')
  }

  private getOptions (): SqliteOptions | MariaDbOptions {
    return {
      dbName: this.dbName,
      entities: ['./dist/types/orm/entities/*.js'],
      entitiesTs: ['./src/types/orm/entities/*.ts'],
      clientUrl: this.getClientUrl()
    }
  }

  private getClientUrl (): string | undefined {
    switch (this.storageType) {
      case StorageType.SQLITE:
      case StorageType.MARIA_DB:
      case StorageType.MYSQL:
        return this.storageUri.toString()
    }
  }
}