dev add stats for search check

2026-02-05 18:55:54 +07:00
parent adc978bea8
commit f90b8a0bce
15 changed files with 261 additions and 53 deletions
@@ -331,8 +331,11 @@ public class FileUploadServiceImpl implements FileUploadService {

                completeFileProcessingAsync(session, status);

+
                if (status != FileStatus.TEMP) {
-                    tariffInfoService.writeOffTokens(session.getUserId(), TariffConstants.TOKEN_VALUE);
+                    String fileType = session.getFileType();
+
+                    tariffInfoService.writeOffTokens(session.getUserId(), protectCost(fileType));
                }
            } else {
                sessionRepository.save(session);
@@ -369,6 +372,23 @@ public class FileUploadServiceImpl implements FileUploadService {
        }
    }

+    private int protectCost(String fileType) {
+        int cost = 0;
+
+        switch (fileType) {
+            case "video":
+                cost = TariffConstants.VIDEO_TOKEN_VALUE;
+            case "image":
+                cost = TariffConstants.IMAGE_TOKEN_VALUE;
+            case "audio":
+                cost = TariffConstants.AUDIO_TOKEN_VALUE;
+            case "pdf":
+                cost = TariffConstants.PDF_TOKEN_VALUE;
+        }
+
+        return cost;
+    }
+
    private String assembleFileSynchronously(FileUploadSession session) throws IOException {
        Path finalFilePath = prepareFinalFile(session);
        log.info("file path: " + finalFilePath);