javadev / pt-backend

import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.boot.web.servlet.error.ErrorAttributes;

import org.springframework.boot.web.servlet.error.ErrorController;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import org.springframework.web.context.request.WebRequest;

@RestController

class CustomErrorController implements ErrorController {

    @Value("${logging.level.org.springframework.web}")

    private String loggingLevel;

    @Autowired

    private ErrorAttributes errorAttributes;

    @RequestMapping(value = PATH)

    ErrorDTO error(WebRequest webRequest, HttpServletResponse response) {

        // Appropriate HTTP response code (e.g. 404 or 500) is automatically set by Spring. 

        // Here we just define response body.

        return new ErrorDTO(response.getStatus(), getErrorAttributes(webRequest, loggingLevel));

    }

    @Override

    public String getErrorPath() {

        return PATH;

    }

}