From 95a6ad7770de81b8dafce84275d7ab9c3358de1e Mon Sep 17 00:00:00 2001
From: Methapon Metanipat <methapon@frappet.com>
Date: Mon, 26 Aug 2024 11:24:29 +0700
Subject: [PATCH] feat: do not process file more than 100MB

---
 Services/server/src/rabbitmq/handler.ts | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/Services/server/src/rabbitmq/handler.ts b/Services/server/src/rabbitmq/handler.ts
index 811de12..74d8126 100644
--- a/Services/server/src/rabbitmq/handler.ts
+++ b/Services/server/src/rabbitmq/handler.ts
@@ -40,6 +40,12 @@ export async function handler(key: string, event: string): Promise<boolean> {
     return await ensureDelete(pathname);
   }
 
+  if (!cachedMetadata[key]) {
+    const stat = await minioClient.statObject(bucket, pathname);
+    cachedMetadata[key] = { size: stat.size, type: stat.metaData["content-type"] };
+    if (stat.size > 1024 * 1024 * 100) cachedBuffer[key] = Buffer.from([]);
+  }
+
   if (!cachedBuffer[key]) {
     try {
       const stream = await minioClient.getObject(bucket, pathname);
@@ -56,11 +62,6 @@ export async function handler(key: string, event: string): Promise<boolean> {
     }
   }
 
-  if (!cachedMetadata[key]) {
-    const stat = await minioClient.statObject(bucket, pathname);
-    cachedMetadata[key] = { size: stat.size, type: stat.metaData["content-type"] };
-  }
-
   const rec = await popInfo(pathname);
 
   if (rec) cachedRecord[key] = rec;