fix: fix getDontLookUpFiles logic errors (#28)

YAPP-Github · Jul 13, 2024 · 8b309a6 · 8b309a6
1 parent 48adc1d
commit 8b309a6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/photo-service/src/main/java/kr/mafoo/photo/service/QrService.java b/photo-service/src/main/java/kr/mafoo/photo/service/QrService.java
@@ -91,7 +91,7 @@ private Mono<byte[]> getHaruFilmFiles(String qrUrl) {
     }
 
     private Mono<byte[]> getDontLookUpFiles(String qrUrl) {
-        String imageName = extractValueFromUrl(qrUrl, ".kr/");
+        String imageName = extractValueFromUrl(qrUrl, ".kr/image/");
 
         String baseUrl = "https://x.dontlxxkup.kr/uploads/";
         String imageUrl = baseUrl + imageName;
@@ -107,7 +107,10 @@ private Mono<byte[]> getDontLookUpFiles(String qrUrl) {
                     } else {
                         return getFileAsByte(imageUrl);
                     }
-                });
+                })
+                .onErrorResume(
+                        RedirectUriNotFoundException.class, e -> getFileAsByte(imageUrl)
+                );
     }
 
     private Mono<String> getRedirectUri(String url) {