Task 20 : Revise CustomError and CustomResponse

Rapter1990 · Jun 30, 2024 · fe21ac7 · fe21ac7
1 parent 9fe2432
commit fe21ac7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/src/main/java/com/springboot/ratelimiter/common/exception/error/CustomError.java b/src/main/java/com/springboot/ratelimiter/common/exception/error/CustomError.java
@@ -59,7 +59,9 @@ public enum Header {
 
         DATABASE_ERROR("DATABASE ERROR"),
 
-        PROCESS_ERROR("PROCESS ERROR");
+        PROCESS_ERROR("PROCESS ERROR"),
+
+        RATE_LIMITER_EXCEEDED_ERROR("RATE LIMITER EXCEEDED ERROR");
 
         private final String name;
 

diff --git a/src/main/java/com/springboot/ratelimiter/common/model/dto/response/CustomResponse.java b/src/main/java/com/springboot/ratelimiter/common/model/dto/response/CustomResponse.java
@@ -4,6 +4,7 @@
 import lombok.Builder;
 import lombok.Getter;
 import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.ResponseStatus;
 
 import java.time.LocalDateTime;
 
@@ -27,11 +28,20 @@ public class CustomResponse<T> {
             .build();
 
 
-    public static <T> CustomResponse<T> successOf(final T response) {
-        return CustomResponse.<T>builder()
-                .httpStatus(HttpStatus.OK)
+    public static <E> CustomResponse<E> ok(E response) {
+        return CustomResponse.<E>builder()
+                .response(response)
                 .isSuccess(true)
+                .httpStatus(HttpStatus.OK)
+                .build();
+    }
+
+    @ResponseStatus(HttpStatus.CREATED)
+    public static <E> CustomResponse<E> created(E response) {
+        return CustomResponse.<E>builder()
                 .response(response)
+                .isSuccess(true)
+                .httpStatus(HttpStatus.CREATED)
                 .build();
     }