feat: move grid fs calls from db server to db helper[3966]

hashgraph · Aug 19, 2024 · 924763a · 924763a
1 parent fbaa21b
commit 924763a
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 23 deletions.
diff --git a/common/src/database-modules/database-server.ts b/common/src/database-modules/database-server.ts
@@ -3543,40 +3543,21 @@ export class DatabaseServer {
      * Save file
      * @param uuid
      * @param buffer
+     *
      * @returns file ID
      */
     public static async saveFile(uuid: string, buffer: Buffer): Promise<ObjectId> {
-        return new Promise<ObjectId>((resolve, reject) => {
-            try {
-                const fileStream = DataBaseHelper.gridFS.openUploadStream(uuid);
-                fileStream.write(buffer);
-                fileStream.end(() => {
-                    resolve(fileStream.id);
-                });
-            } catch (error) {
-                reject(error);
-            }
-        });
+        return DataBaseHelper.saveFile(uuid, buffer)
     }
 
     /**
-     * Save file
+     * Load file
      * @param id
      *
      * @returns file ID
      */
     public static async loadFile(id: ObjectId): Promise<Buffer> {
-        const files = await DataBaseHelper.gridFS.find(id).toArray();
-        if (files.length === 0) {
-            return null;
-        }
-        const file = files[0];
-        const fileStream = DataBaseHelper.gridFS.openDownloadStream(file._id);
-        const bufferArray = [];
-        for await (const data of fileStream) {
-            bufferArray.push(data);
-        }
-        return Buffer.concat(bufferArray);
+        return DataBaseHelper.loadFile(id)
     }
 
     /**

diff --git a/common/src/helpers/db-helper.ts b/common/src/helpers/db-helper.ts
@@ -138,6 +138,46 @@ export class DataBaseHelper<T extends BaseEntity> {
         return DataBaseHelper._gridFS;
     }
 
+    /**
+     * Save file
+     * @param uuid
+     * @param buffer
+     * @returns file ID
+     */
+    public static async saveFile(uuid: string, buffer: Buffer): Promise<ObjectId> {
+        return new Promise<ObjectId>((resolve, reject) => {
+            try {
+                const fileStream = DataBaseHelper.gridFS.openUploadStream(uuid);
+                fileStream.write(buffer);
+                fileStream.end(() => {
+                    resolve(fileStream.id);
+                });
+            } catch (error) {
+                reject(error);
+            }
+        });
+    }
+
+    /**
+     * Load file
+     * @param id
+     *
+     * @returns file ID
+     */
+    public static async loadFile(id: ObjectId): Promise<Buffer> {
+        const files = await DataBaseHelper.gridFS.find(id).toArray();
+        if (files.length === 0) {
+            return null;
+        }
+        const file = files[0];
+        const fileStream = DataBaseHelper.gridFS.openDownloadStream(file._id);
+        const bufferArray = [];
+        for await (const data of fileStream) {
+            bufferArray.push(data);
+        }
+        return Buffer.concat(bufferArray);
+    }
+
     /**
      * Delete entities by filters
      * @param filters filters