Merge pull request #20 from YAPP-Github/feature/#18

feat: 유저 서비스 요청 Validation 작업
YAPP-Github · Jul 12, 2024 · 4602744 · 4602744
2 parents 35fc52f + e1fe762
commit 4602744
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 0 deletions.
diff --git a/user-service/build.gradle.kts b/user-service/build.gradle.kts
@@ -24,6 +24,7 @@ dependencies {
 	implementation("org.springframework.boot:spring-boot-starter-webflux")
 	implementation("org.flywaydb:flyway-core")
 	implementation("org.flywaydb:flyway-mysql")
+	implementation("org.springframework.boot:spring-boot-starter-validation")
 // https://mvnrepository.com/artifact/com.mysql/mysql-connector-j
 	implementation("com.mysql:mysql-connector-j:8.4.0")
 

diff --git a/user-service/src/main/java/kr/mafoo/user/api/AuthApi.java b/user-service/src/main/java/kr/mafoo/user/api/AuthApi.java
@@ -5,12 +5,14 @@
 import kr.mafoo.user.controller.dto.request.KakaoLoginRequest;
 import kr.mafoo.user.controller.dto.request.TokenRefreshRequest;
 import kr.mafoo.user.controller.dto.response.LoginResponse;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import reactor.core.publisher.Mono;
 
 @Tag(name = "인증(로그인) 관련 API", description = "토큰 발행, 로그인 등 API")
+@Validated
 @RequestMapping("/v1/auth")
 public interface AuthApi {
     @Operation(summary = "카카오 로그인", description = "카카오 인가 코드로 로그인(토큰 발행)합니다.")

diff --git a/user-service/src/main/java/kr/mafoo/user/api/MeApi.java b/user-service/src/main/java/kr/mafoo/user/api/MeApi.java
@@ -5,12 +5,14 @@
 import io.swagger.v3.oas.annotations.tags.Tag;
 import kr.mafoo.user.annotation.RequestMemberId;
 import kr.mafoo.user.controller.dto.response.MemberResponse;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import reactor.core.publisher.Mono;
 
 @Tag(name = "로그인 사용자 정보 API", description = "현재 토큰 주인의 정보를 다루는 API")
+@Validated
 @RequestMapping("/v1/me")
 public interface MeApi {
     @Operation(summary = "내 정보 조회", description = "현재 토큰 주인의 정보를 조회합니다.")

diff --git a/user-service/src/main/java/kr/mafoo/user/config/WebExceptionHandler.java b/user-service/src/main/java/kr/mafoo/user/config/WebExceptionHandler.java
@@ -1,10 +1,14 @@
 package kr.mafoo.user.config;
 
+import jakarta.validation.ConstraintViolationException;
 import kr.mafoo.user.controller.dto.response.ErrorResponse;
 import kr.mafoo.user.exception.DomainException;
+import kr.mafoo.user.exception.ErrorCode;
 import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.MethodArgumentNotValidException;
 import org.springframework.web.bind.annotation.ControllerAdvice;
 import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.support.WebExchangeBindException;
 
 @ControllerAdvice
 public class WebExceptionHandler {
@@ -14,4 +18,28 @@ public ResponseEntity<ErrorResponse> handleDomainException(DomainException excep
                 .badRequest()
                 .body(ErrorResponse.fromErrorCode(exception.getErrorCode()));
     }
+
+    @ExceptionHandler({MethodArgumentNotValidException.class,
+            ConstraintViolationException.class,
+            WebExchangeBindException.class})
+    public ResponseEntity<ErrorResponse> validException(Exception ex) {
+        String errorMessage = "입력값 검증 오류: ";
+        if (ex instanceof MethodArgumentNotValidException mex) {
+            errorMessage += mex.getBindingResult().getAllErrors().get(0).getDefaultMessage();
+        } else if (ex instanceof ConstraintViolationException cvex) {
+            errorMessage += cvex.getConstraintViolations().iterator().next().getMessage();
+        } else if (ex instanceof WebExchangeBindException wex) {
+            errorMessage += wex.getAllErrors().get(0).getDefaultMessage();
+        } else {
+            errorMessage += "알 수 없는 오류";
+        }
+        ErrorResponse response = new ErrorResponse(
+                ErrorCode.REQUEST_INPUT_NOT_VALID.getCode(),
+                errorMessage
+        );
+
+        return ResponseEntity
+                .badRequest()
+                .body(response);
+    }
 }
diff --git a/user-service/src/main/java/kr/mafoo/user/exception/ErrorCode.java b/user-service/src/main/java/kr/mafoo/user/exception/ErrorCode.java
@@ -8,6 +8,7 @@
 public enum ErrorCode {
     MEMBER_NOT_FOUND("ME0001", "사용자를 찾을 수 없습니다"),
     KAKAO_LOGIN_FAILED("EX0001", "카카오 로그인에 실패했습니다"),
+    REQUEST_INPUT_NOT_VALID("EX0002", "입력 값이 올바르지 않습니다."),
 
     TOKEN_TYPE_MISMATCH("AU0001", "토큰 타입이 일치하지 않습니다. (아마 AccessToken?)"),
     TOKEN_EXPIRED("AU0002", "토큰이 만료되었습니다"),