feat: fix authentication

YAPP-Github · Sep 10, 2024 · e8d4f19 · e8d4f19
1 parent 4f76bd6
commit e8d4f19
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/api-gateway/src/main/java/kr/mafoo/gateway/filter/AdminAuthenticationFilter.java b/api-gateway/src/main/java/kr/mafoo/gateway/filter/AdminAuthenticationFilter.java
@@ -47,6 +47,7 @@ private Mono<Void> validateUserFromGoogleToken(ServerWebExchange exchange, Gatew
         return webClient.get()
                 .uri("https://www.googleapis.com/oauth2/v1/userinfo?alt=json&access_token=" + token)
                 .retrieve()
+                .onStatus(status -> status.value() == 401, res -> Mono.error(new RuntimeException("AUTH")))
                 .bodyToMono(Map.class)
                 .flatMap(userInfo -> {
                     if (userInfo.containsKey("id")) {
@@ -56,7 +57,9 @@ private Mono<Void> validateUserFromGoogleToken(ServerWebExchange exchange, Gatew
                         }
                     }
                     return createSimpleErrorResponse(exchange, "AU0004", "인증에 실패했습니다");
-                });
+                })
+                .onErrorResume(error -> error.getMessage().equals("AUTH"),
+                        error -> createSimpleErrorResponse(exchange, "AU0004", "인증에 실패했습니다"));
     }
 
     private Mono<Void> createSimpleErrorResponse(ServerWebExchange exchange, String code, String message) {