exception 추가 후 onErrorMap 을 통해 에러 맵핑

Daemawiki · Mar 1, 2024 · fbbda61 · fbbda61
1 parent 9737b34
commit fbbda61
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 2 deletions.
diff --git a/src/main/java/com/example/daemawiki/domain/mail/repository/AuthCodeRepository.java b/src/main/java/com/example/daemawiki/domain/mail/repository/AuthCodeRepository.java
@@ -1,7 +1,10 @@
 package com.example.daemawiki.domain.mail.repository;
 
 import com.example.daemawiki.domain.mail.model.AuthCode;
+import com.example.daemawiki.global.exception.h500.ExecuteFailedException;
+import com.example.daemawiki.global.exception.h500.RedisConnectFailedException;
 import com.example.daemawiki.global.type.RedisKey;
+import org.springframework.data.redis.RedisConnectionFailureException;
 import org.springframework.data.redis.core.ReactiveRedisOperations;
 import org.springframework.stereotype.Repository;
 import reactor.core.publisher.Mono;
@@ -22,11 +25,13 @@ public Mono<AuthCode> findByMail(String mail) {
         return redisOperations.opsForValue().get(AUTHCODE + mail)
                 .map(value -> AuthCode.builder()
                         .mail(mail)
-                        .code(value).build());
+                        .code(value).build())
+                .onErrorMap(e -> e instanceof RedisConnectionFailureException ? RedisConnectFailedException.EXCEPTION : ExecuteFailedException.EXCEPTION);
     }
 
     public Mono<Void> delete(AuthCode authCode) {
         return redisOperations.delete(AUTHCODE + authCode.getMail())
+                .onErrorMap(e -> e instanceof RedisConnectionFailureException ? RedisConnectFailedException.EXCEPTION : ExecuteFailedException.EXCEPTION)
                 .then();
     }
 
@@ -35,6 +40,7 @@ public Mono<Void> save(AuthCode authCode) {
                 .set(AUTHCODE + authCode.getMail(),
                         authCode.getCode(),
                         Duration.ofHours(3))
+                .onErrorMap(e -> e instanceof RedisConnectionFailureException ? RedisConnectFailedException.EXCEPTION : ExecuteFailedException.EXCEPTION)
                 .then();
     }
 

diff --git a/src/main/java/com/example/daemawiki/domain/mail/repository/AuthMailRepository.java b/src/main/java/com/example/daemawiki/domain/mail/repository/AuthMailRepository.java
@@ -1,7 +1,10 @@
 package com.example.daemawiki.domain.mail.repository;
 
 import com.example.daemawiki.domain.mail.model.AuthMail;
+import com.example.daemawiki.global.exception.h500.ExecuteFailedException;
+import com.example.daemawiki.global.exception.h500.RedisConnectFailedException;
 import com.example.daemawiki.global.type.RedisKey;
+import org.springframework.data.redis.RedisConnectionFailureException;
 import org.springframework.data.redis.core.ReactiveRedisOperations;
 import org.springframework.stereotype.Repository;
 import reactor.core.publisher.Mono;
@@ -24,17 +27,20 @@ public Mono<Void> save(AuthMail authMail) {
                 .set(AUTHMAIL + authMail.getMail(),
                         authMail.getMail(),
                         Duration.ofHours(3))
+                .onErrorMap(e -> e instanceof RedisConnectionFailureException ? RedisConnectFailedException.EXCEPTION : ExecuteFailedException.EXCEPTION)
                 .then();
     }
 
     public Mono<Boolean> findByMail(String mail) {
         return redisOperations.opsForValue()
                 .get(AUTHMAIL + mail)
-                .hasElement();
+                .hasElement()
+                .onErrorMap(e -> e instanceof RedisConnectionFailureException ? RedisConnectFailedException.EXCEPTION : ExecuteFailedException.EXCEPTION);
     }
 
     public Mono<Void> delete(String mail) {
         return redisOperations.delete(AUTHMAIL + mail)
+                .onErrorMap(e -> e instanceof RedisConnectionFailureException ? RedisConnectFailedException.EXCEPTION : ExecuteFailedException.EXCEPTION)
                 .then();
     }
 

diff --git a/src/main/java/com/example/daemawiki/global/error/ErrorCode.java b/src/main/java/com/example/daemawiki/global/error/ErrorCode.java
@@ -30,6 +30,7 @@ public enum ErrorCode {
     TOKEN_REISSUE_FAILED(500, "토큰 재발급 실패"),
     FILE_UPLOAD_FAILED(500, "파일 업로드 실패ㅋ"),
     EXECUTE_FAILED(500, "서버 에러"),
+    REDIS_CONNECT_FAILED(500, "redis 연결 실패"),
 
 
     TEST(400, "TEST");

diff --git a/src/main/java/com/example/daemawiki/global/exception/h500/RedisConnectFailedException.java b/src/main/java/com/example/daemawiki/global/exception/h500/RedisConnectFailedException.java
@@ -0,0 +1,13 @@
+package com.example.daemawiki.global.exception.h500;
+
+import com.example.daemawiki.global.error.ErrorCode;
+import com.example.daemawiki.global.error.exception.CustomException;
+
+public class RedisConnectFailedException extends CustomException {
+    public static final CustomException EXCEPTION = new RedisConnectFailedException();
+
+    private RedisConnectFailedException() {
+        super(ErrorCode.REDIS_CONNECT_FAILED);
+    }
+
+}