Merge pull request #71 from YAPP-Github/feature/MAFOO-169

[MAFOO-169] feat: QR 사진 예외 관련 슬랙 알림을 추가했어요
YAPP-Github · Oct 26, 2024 · a4d047c · a4d047c
2 parents fc132fd + 20b06b7
commit a4d047c
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 1 deletion.
diff --git a/photo-service/src/main/java/kr/mafoo/photo/config/RequestBodyCachingFilter.java b/photo-service/src/main/java/kr/mafoo/photo/config/RequestBodyCachingFilter.java
@@ -0,0 +1,46 @@
+package kr.mafoo.photo.config;
+
+import org.springframework.core.io.buffer.DataBuffer;
+import org.springframework.core.io.buffer.DataBufferUtils;
+import org.springframework.http.server.reactive.ServerHttpRequest;
+import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
+import org.springframework.stereotype.Component;
+import org.springframework.web.server.ServerWebExchange;
+import org.springframework.web.server.ServerWebExchangeDecorator;
+import org.springframework.web.server.WebFilter;
+import org.springframework.web.server.WebFilterChain;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+@Component
+public class RequestBodyCachingFilter implements WebFilter {
+
+    // TODO: 추후 정리 필요
+
+    @Override
+    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
+        return DataBufferUtils.join(exchange.getRequest().getBody())
+                .flatMap(dataBuffer -> {
+                    byte[] bytes = new byte[dataBuffer.readableByteCount()];
+                    dataBuffer.read(bytes);
+                    DataBufferUtils.release(dataBuffer);
+
+                    ServerHttpRequestDecorator decoratedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {
+                        @Override
+                        public Flux<DataBuffer> getBody() {
+                            return Flux.just(exchange.getResponse().bufferFactory().wrap(bytes));
+                        }
+                    };
+
+                    ServerWebExchangeDecorator decoratedExchange = new ServerWebExchangeDecorator(exchange) {
+                        @Override
+                        public ServerHttpRequest getRequest() {
+                            return decoratedRequest;
+                        }
+                    };
+
+                    return chain.filter(decoratedExchange);
+                });
+    }
+}
+
diff --git a/photo-service/src/main/java/kr/mafoo/photo/config/WebExceptionHandler.java b/photo-service/src/main/java/kr/mafoo/photo/config/WebExceptionHandler.java
@@ -4,6 +4,7 @@
 import kr.mafoo.photo.controller.dto.response.ErrorResponse;
 import kr.mafoo.photo.exception.DomainException;
 import kr.mafoo.photo.exception.ErrorCode;
+import kr.mafoo.photo.exception.PhotoBrandNotExistsException;
 import kr.mafoo.photo.service.SlackService;
 import lombok.RequiredArgsConstructor;
 import org.slf4j.Logger;
@@ -61,6 +62,26 @@ public ResponseEntity<ErrorResponse> validException(Exception ex) {
                 .body(response);
     }
 
+    @ExceptionHandler(PhotoBrandNotExistsException.class)
+    public Mono<ResponseEntity<ErrorResponse>> handlePhotoBrandNotExistsException(ServerWebExchange exchange, PhotoBrandNotExistsException exception) {
+        String method = extractMethod(exchange);
+        String userAgent = extractUserAgent(exchange);
+        String fullPath = extractURI(exchange);
+        String originIp = extractOriginIp(exchange);
+
+        return extractRequestBody(exchange).flatMap(requestBody -> {
+            logException(method, fullPath, originIp, userAgent, exception);
+
+            return slackService.sendQrRelatedErrorNotification(
+                    method, fullPath, requestBody, originIp, userAgent, exception.getMessage()
+            ).then(Mono.just(
+                    ResponseEntity
+                            .badRequest()
+                            .body(ErrorResponse.fromErrorCode(exception.getErrorCode()))
+            ));
+        });
+    }
+
     @ExceptionHandler(ResponseStatusException.class)
     public Mono<ResponseEntity<String>> handleResponseStatusException(ServerWebExchange exchange, ResponseStatusException exception) {
         return handleExceptionInternal(exchange, exception, (HttpStatus) exception.getStatusCode());

diff --git a/photo-service/src/main/resources/application.yaml b/photo-service/src/main/resources/application.yaml
@@ -47,4 +47,5 @@ slack:
   webhook:
     token: ${SLACK_TOKEN}
     channel:
-      error: ${SLACK_ERROR_CHANNEL}
+      error: ${SLACK_ERROR_CHANNEL}
+      qr: ${SLACK_QR_ERROR_CHANNEL}