Error controller should return the status based on the response status

Author: pushkar-g

src/main/java/com/bnc/sbjb/rest/DefaultErrorController.java

@@ -1,21 +1,22 @@
 package com.bnc.sbjb.rest;
 
+import com.bnc.sbjb.model.api.CustomError;
+import javax.servlet.http.HttpServletResponse;
 import org.springframework.boot.web.servlet.error.ErrorController;
 import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.ResponseStatus;
 import org.springframework.web.bind.annotation.RestController;
 
 @RestController
 public class DefaultErrorController implements ErrorController {
 
     private static final String PATH = "/error";
 
-    @ResponseStatus(HttpStatus.NOT_FOUND)
     @RequestMapping(value = PATH)
-    public void error() {
-        // Blank method since we don't need to do anything to handle unknown paths.
-        // Any forked service may choose to handle some requests differently.
+    public ResponseEntity<CustomError> error(HttpServletResponse response) {
+        HttpStatus status = HttpStatus.valueOf(response.getStatus());
+        return ResponseEntity.status(response.getStatus()).body(new CustomError(status, status.getReasonPhrase()));
     }
 
     @Override