creating a common error handling for RestController with basic error codes (own error handling!)

Csaba79-coder · Csaba79-coder · commit 8e3fb8b6c88c · 2023-05-26T08:23:57.000+02:00
diff --git a/src/main/java/com/csaba79coder/littersnap/exception/ControllerExceptionHandler.java b/src/main/java/com/csaba79coder/littersnap/exception/ControllerExceptionHandler.java
@@ -0,0 +1,37 @@
+package com.csaba79coder.littersnap.exception;
+
+import com.csaba79coder.littersnap.value.ErrorCode;
+import org.modelmapper.ValidationException;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+
+import java.util.InputMismatchException;
+import java.util.Map;
+import java.util.NoSuchElementException;
+
+import static com.csaba79coder.littersnap.value.ErrorCode.LS_001;
+import static com.csaba79coder.littersnap.value.ErrorCode.LS_002;
+import static com.csaba79coder.littersnap.value.ErrorCode.LS_003;
+
+public class ControllerExceptionHandler {
+
+    @ExceptionHandler(value = {NoSuchElementException.class})
+    public ResponseEntity<Object> handleNoSuchElementException(NoSuchElementException ex) {
+        return new ResponseEntity<>(responseBodyWithMessage(LS_001, ex.getMessage()), HttpStatus.NOT_FOUND);
+    }
+
+    @ExceptionHandler(value = {InputMismatchException.class})
+    public ResponseEntity<Object> handleInvalidInputException(InputMismatchException ex) {
+        return new ResponseEntity<>(responseBodyWithMessage(LS_002, ex.getMessage()), HttpStatus.BAD_REQUEST);
+    }
+
+    @ExceptionHandler(value = {ValidationException.class})
+    public ResponseEntity<Object> handleInvalidPasswordOrUsernameException(ValidationException ex) {
+        return new ResponseEntity<>(responseBodyWithMessage(LS_003, ex.getMessage()), HttpStatus.BAD_REQUEST);
+    }
+
+    private String responseBodyWithMessage(ErrorCode code, String message) {
+        return Map.of(code, message).toString();
+    }
+}
diff --git a/src/main/java/com/csaba79coder/littersnap/value/ErrorCode.java b/src/main/java/com/csaba79coder/littersnap/value/ErrorCode.java
@@ -0,0 +1,6 @@
+package com.csaba79coder.littersnap.value;
+
+public enum ErrorCode {
+
+    LS_001, LS_002, LS_003
+}
