Apply prettier formating

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

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

import Constants from '../../utils/Constants';
import { MongoClient } from 'mongodb';
import Statistics from '../../types/Statistics';
import { Storage } from './Storage';
import { StorageType } from '../../types/Storage';

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 Error(
        `${this.logPrefix} ${this.getDBNameFromStorageType(
          StorageType.MONGO_DB
        )} client initialization failed while trying to issue a request`
      );
    }
    if (!this.connected) {
      throw new Error(
        `${this.logPrefix} ${this.getDBNameFromStorageType(
          StorageType.MONGO_DB
        )} connection not opened while trying to issue a request`
      );
    }
  }
}