X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=src%2Fperformance%2Fstorage%2FJsonFileStorage.ts;h=58680181f7499b858e485ea828e6f969910b78f9;hb=3f0734367ba34b15eb24c7d1e344ad8ae51a3d0d;hp=d5eb848c7c90d5dca20c09623496b71d44b36ef5;hpb=b40b5cb39eb44088100f2ce5efc2a6fdf4b84f26;p=e-mobility-charging-stations-simulator.git

diff --git a/src/performance/storage/JsonFileStorage.ts b/src/performance/storage/JsonFileStorage.ts
index d5eb848c..58680181 100644
--- a/src/performance/storage/JsonFileStorage.ts
+++ b/src/performance/storage/JsonFileStorage.ts
@@ -1,17 +1,23 @@
 // Copyright Jerome Benoit. 2021-2023. All Rights Reserved.
 
-import fs from 'fs';
-
-import lockfile from 'proper-lockfile';
+import { closeSync, existsSync, mkdirSync, openSync, writeSync } from 'node:fs';
+import { dirname } from 'node:path';
 
 import { Storage } from './Storage';
-import { FileType } from '../../types/FileType';
-import type { Statistics } from '../../types/Statistics';
-import FileUtils from '../../utils/FileUtils';
-import Utils from '../../utils/Utils';
+import { BaseError } from '../../exception';
+import { FileType, type Statistics } from '../../types';
+import {
+  AsyncLock,
+  AsyncLockType,
+  JSONStringifyWithMapSupport,
+  handleFileException,
+  isNullOrUndefined,
+} from '../../utils';
 
 export class JsonFileStorage extends Storage {
-  private fd: number | null = null;
+  private static performanceRecords: Map<string, Statistics>;
+
+  private fd?: number;
 
   constructor(storageUri: string, logPrefix: string) {
     super(storageUri, logPrefix);
@@ -20,70 +26,64 @@ export class JsonFileStorage extends Storage {
 
   public storePerformanceStatistics(performanceStatistics: Statistics): void {
     this.checkPerformanceRecordsFile();
-    lockfile
-      .lock(this.dbName, { stale: 5000, retries: 3 })
-      .then(async (release) => {
-        try {
-          const fileData = fs.readFileSync(this.dbName, 'utf8');
-          const performanceRecords: Statistics[] = fileData
-            ? (JSON.parse(fileData) as Statistics[])
-            : [];
-          performanceRecords.push(performanceStatistics);
-          fs.writeFileSync(
-            this.dbName,
-            Utils.JSONStringifyWithMapSupport(performanceRecords, 2),
-            'utf8'
-          );
-        } catch (error) {
-          FileUtils.handleFileException(
-            this.logPrefix,
-            FileType.PerformanceRecords,
-            this.dbName,
-            error as NodeJS.ErrnoException
-          );
-        }
-        await release();
-      })
-      .catch(() => {
-        /* This is intentional */
-      });
+    JsonFileStorage.performanceRecords.set(performanceStatistics.id, performanceStatistics);
+    AsyncLock.runExclusive(AsyncLockType.performance, () => {
+      writeSync(
+        this.fd!,
+        JSONStringifyWithMapSupport([...JsonFileStorage.performanceRecords.values()], 2),
+        0,
+        'utf8',
+      );
+    }).catch((error) => {
+      handleFileException(
+        this.dbName,
+        FileType.PerformanceRecords,
+        error as NodeJS.ErrnoException,
+        this.logPrefix,
+      );
+    });
   }
 
   public open(): void {
+    JsonFileStorage.performanceRecords = new Map<string, Statistics>();
     try {
-      if (this?.fd === undefined || this?.fd === null) {
-        this.fd = fs.openSync(this.dbName, 'a+');
+      if (isNullOrUndefined(this?.fd)) {
+        if (!existsSync(dirname(this.dbName))) {
+          mkdirSync(dirname(this.dbName), { recursive: true });
+        }
+        this.fd = openSync(this.dbName, 'w');
       }
     } catch (error) {
-      FileUtils.handleFileException(
-        this.logPrefix,
-        FileType.PerformanceRecords,
+      handleFileException(
         this.dbName,
-        error as NodeJS.ErrnoException
+        FileType.PerformanceRecords,
+        error as NodeJS.ErrnoException,
+        this.logPrefix,
       );
     }
   }
 
   public close(): void {
+    JsonFileStorage.performanceRecords.clear();
     try {
       if (this?.fd) {
-        fs.closeSync(this.fd);
-        this.fd = null;
+        closeSync(this.fd);
+        delete this?.fd;
       }
     } catch (error) {
-      FileUtils.handleFileException(
-        this.logPrefix,
-        FileType.PerformanceRecords,
+      handleFileException(
         this.dbName,
-        error as NodeJS.ErrnoException
+        FileType.PerformanceRecords,
+        error as NodeJS.ErrnoException,
+        this.logPrefix,
       );
     }
   }
 
   private checkPerformanceRecordsFile(): void {
     if (!this?.fd) {
-      throw new Error(
-        `${this.logPrefix} Performance records '${this.dbName}' file descriptor not found`
+      throw new BaseError(
+        `${this.logPrefix} Performance records '${this.dbName}' file descriptor not found`,
       );
     }
   }