X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fperformance%2Fstorage%2FMongoDBStorage.ts;h=c3c98022d4204c63aa6d940f7cac887d2424ed58;hb=a94f567f2d7cefba06e74fa0a957c096cf581b88;hp=d941d7c188ad920a036f804838056f26ae13b887;hpb=8114d10e3893e96bb725ce2fca9744429ee4b75b;p=e-mobility-charging-stations-simulator.git

diff --git a/src/performance/storage/MongoDBStorage.ts b/src/performance/storage/MongoDBStorage.ts
index d941d7c1..c3c98022 100644
--- a/src/performance/storage/MongoDBStorage.ts
+++ b/src/performance/storage/MongoDBStorage.ts
@@ -1,73 +1,73 @@
-// Copyright Jerome Benoit. 2021. All Rights Reserved.
+// Copyright Jerome Benoit. 2021-2024. All Rights Reserved.
 
-import { MongoClient } from 'mongodb';
+import { MongoClient } from 'mongodb'
 
-import Statistics from '../../types/Statistics';
-import { StorageType } from '../../types/Storage';
-import Constants from '../../utils/Constants';
-import { Storage } from './Storage';
+import { BaseError } from '../../exception/index.js'
+import { type Statistics, StorageType } from '../../types/index.js'
+import { Constants } from '../../utils/index.js'
+import { Storage } from './Storage.js'
 
 export class MongoDBStorage extends Storage {
-  private readonly client: MongoClient | null;
-  private connected: boolean;
+  private readonly client?: MongoClient
+  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;
+  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, '')
   }
 
-  public async storePerformanceStatistics(performanceStatistics: Statistics): Promise<void> {
+  public async storePerformanceStatistics (performanceStatistics: Statistics): Promise<void> {
     try {
-      this.checkDBConnection();
+      this.setPerformanceStatistics(performanceStatistics)
+      this.checkDBConnection()
       await this.client
-        .db(this.dbName)
+        ?.db(this.dbName)
         .collection<Statistics>(Constants.PERFORMANCE_RECORDS_TABLE)
-        .insertOne(performanceStatistics);
+        .replaceOne({ id: performanceStatistics.id }, performanceStatistics, { upsert: true })
     } catch (error) {
-      this.handleDBError(StorageType.MONGO_DB, error as Error, Constants.PERFORMANCE_RECORDS_TABLE);
+      this.handleDBError(StorageType.MONGO_DB, error as Error, Constants.PERFORMANCE_RECORDS_TABLE)
     }
   }
 
-  public async open(): Promise<void> {
+  public async open (): Promise<void> {
     try {
-      if (!this.connected && this?.client) {
-        await this.client.connect();
-        this.connected = true;
+      if (!this.connected && this.client != null) {
+        await this.client.connect()
+        this.connected = true
       }
     } catch (error) {
-      this.handleDBError(StorageType.MONGO_DB, error as Error);
+      this.handleDBError(StorageType.MONGO_DB, error as Error)
     }
   }
 
-  public async close(): Promise<void> {
+  public async close (): Promise<void> {
+    this.clearPerformanceStatistics()
     try {
-      if (this.connected && this?.client) {
-        await this.client.close();
-        this.connected = false;
+      if (this.connected && this.client != null) {
+        await this.client.close()
+        this.connected = false
       }
     } catch (error) {
-      this.handleDBError(StorageType.MONGO_DB, error as Error);
+      this.handleDBError(StorageType.MONGO_DB, error as Error)
     }
   }
 
-  private checkDBConnection() {
-    if (!this?.client) {
-      throw new Error(
+  private checkDBConnection (): void {
+    if (this.client == null) {
+      throw new BaseError(
         `${this.logPrefix} ${this.getDBNameFromStorageType(
           StorageType.MONGO_DB
         )} client initialization failed while trying to issue a request`
-      );
+      )
     }
     if (!this.connected) {
-      throw new Error(
+      throw new BaseError(
         `${this.logPrefix} ${this.getDBNameFromStorageType(
           StorageType.MONGO_DB
         )} connection not opened while trying to issue a request`
-      );
+      )
     }
   }
 }