build(deps-dev): apply updates

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

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

import { MongoClient } from 'mongodb';

import { Storage } from './Storage';
import { BaseError } from '../../exception';
import { type Statistics, StorageType } from '../../types';
import { Constants } from '../../utils';

export class MongoDBStorage extends Storage {
  private readonly client: MongoClient | null;
  private connected: boolean;

  constructor(storageUri: string, logPrefix: string) {
    super(storageUri, logPrefix);
    this.client = new MongoClient(this.storageUri.toString());
    this.connected = false;
    this.dbName =
      this.storageUri.pathname.replace(/(?:^\/)|(?:\/$)/g, '') ??
      Constants.DEFAULT_PERFORMANCE_RECORDS_DB_NAME;
  }

  public async storePerformanceStatistics(performanceStatistics: Statistics): Promise<void> {
    try {
      this.checkDBConnection();
      await this.client
        ?.db(this.dbName)
        .collection<Statistics>(Constants.PERFORMANCE_RECORDS_TABLE)
        .insertOne(performanceStatistics);
    } catch (error) {
      this.handleDBError(StorageType.MONGO_DB, error as Error, Constants.PERFORMANCE_RECORDS_TABLE);
    }
  }

  public async open(): Promise<void> {
    try {
      if (!this.connected && this?.client) {
        await this.client.connect();
        this.connected = true;
      }
    } catch (error) {
      this.handleDBError(StorageType.MONGO_DB, error as Error);
    }
  }

  public async close(): Promise<void> {
    try {
      if (this.connected && this?.client) {
        await this.client.close();
        this.connected = false;
      }
    } catch (error) {
      this.handleDBError(StorageType.MONGO_DB, error as Error);
    }
  }

  private checkDBConnection() {
    if (!this?.client) {
      throw new BaseError(
        `${this.logPrefix} ${this.getDBNameFromStorageType(
          StorageType.MONGO_DB,
        )} client initialization failed while trying to issue a request`,
      );
    }
    if (!this.connected) {
      throw new BaseError(
        `${this.logPrefix} ${this.getDBNameFromStorageType(
          StorageType.MONGO_DB,
        )} connection not opened while trying to issue a request`,
      );
    }
  }
}