src/performance/storage/MikroOrmStorage.ts

   1 // Copyright Jerome Benoit. 2021-2023. All Rights Reserved.
   2
   3 import {
   4   type Configuration,
   5   type Connection,
   6   type IDatabaseDriver,
   7   MikroORM,
   8   type Options
   9 } from '@mikro-orm/core'
  10 import { TsMorphMetadataProvider } from '@mikro-orm/reflection'
  11
  12 import { Storage } from './Storage.js'
  13 import {
  14   type MikroOrmDbType,
  15   PerformanceData,
  16   PerformanceRecord,
  17   type Statistics,
  18   StorageType
  19 } from '../../types/index.js'
  20 import { Constants } from '../../utils/index.js'
  21
  22 export class MikroOrmStorage extends Storage {
  23   private readonly storageType: StorageType
  24   private orm?: MikroORM
  25
  26   constructor (storageUri: string, logPrefix: string, storageType: StorageType) {
  27     super(storageUri, logPrefix)
  28     this.storageType = storageType
  29     this.dbName = this.getDBName()
  30   }
  31
  32   public async storePerformanceStatistics (performanceStatistics: Statistics): Promise<void> {
  33     try {
  34       const performanceRecord = new PerformanceRecord()
  35       await this.orm?.em.persistAndFlush(performanceRecord)
  36     } catch (error) {
  37       this.handleDBError(this.storageType, error as Error, Constants.PERFORMANCE_RECORDS_TABLE)
  38     }
  39   }
  40
  41   public async open (): Promise<void> {
  42     try {
  43       if (this?.orm == null) {
  44         this.orm = await MikroORM.init(this.getOptions(), true)
  45       }
  46     } catch (error) {
  47       this.handleDBError(this.storageType, error as Error)
  48     }
  49   }
  50
  51   public async close (): Promise<void> {
  52     try {
  53       if (this?.orm != null) {
  54         await this.orm.close()
  55         delete this?.orm
  56       }
  57     } catch (error) {
  58       this.handleDBError(this.storageType, error as Error)
  59     }
  60   }
  61
  62   private getDBName (): string {
  63     if (this.storageType === StorageType.SQLITE) {
  64       return `${Constants.DEFAULT_PERFORMANCE_RECORDS_DB_NAME}.db`
  65     }
  66     return (
  67       this.storageUri.pathname.replace(/(?:^\/)|(?:\/$)/g, '') ??
  68       Constants.DEFAULT_PERFORMANCE_RECORDS_DB_NAME
  69     )
  70   }
  71
  72   private getOptions ():
  73   | Configuration<IDatabaseDriver<Connection>>
  74   | Options<IDatabaseDriver<Connection>> {
  75     return {
  76       metadataProvider: TsMorphMetadataProvider,
  77       entities: [PerformanceRecord, PerformanceData],
  78       type: this.storageType as MikroOrmDbType,
  79       clientUrl: this.getClientUrl()
  80     }
  81   }
  82
  83   private getClientUrl (): string | undefined {
  84     switch (this.storageType) {
  85       case StorageType.SQLITE:
  86       case StorageType.MARIA_DB:
  87       case StorageType.MYSQL:
  88         return this.storageUri.toString()
  89     }
  90   }
  91 }