修改404异常处理方式

2026-11-01 23:03:41 +00:00 · 2017-08-08 11:44:02 +08:00
parent 9bc76872b0
commit dbff9a3e13
2 changed files with 175 additions and 0 deletions
@@ -0,0 +1,118 @@
+package org.itkk.udf.core.exception.handle;
+
+import org.itkk.udf.core.RestResponse;
+import org.itkk.udf.core.exception.ErrorResult;
+import org.springframework.boot.autoconfigure.web.AbstractErrorController;
+import org.springframework.boot.autoconfigure.web.ErrorAttributes;
+import org.springframework.boot.autoconfigure.web.ErrorProperties;
+import org.springframework.boot.autoconfigure.web.ErrorViewResolver;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.stereotype.Controller;
+import org.springframework.util.Assert;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.ResponseBody;
+import org.springframework.web.servlet.ModelAndView;
+import org.springframework.web.servlet.NoHandlerFoundException;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.*;
+
+/**
+ * ErrorController
+ */
+@Controller
+@RequestMapping("${server.error.path:${error.path:/error}}")
+public class ExceptionController extends AbstractErrorController {
+
+    /**
+     * errorProperties
+     */
+    private final ErrorProperties errorProperties;
+
+    /**
+     * Create a new {@link ExceptionController} instance.
+     *
+     * @param errorAttributes the error attributes
+     * @param errorProperties configuration properties
+     */
+    public ExceptionController(ErrorAttributes errorAttributes, ErrorProperties errorProperties) {
+        this(errorAttributes, errorProperties, new ArrayList<>());
+    }
+
+    /**
+     * Create a new {@link ExceptionController} instance.
+     *
+     * @param errorAttributes    the error attributes
+     * @param errorProperties    configuration properties
+     * @param errorViewResolvers error view resolvers
+     */
+    public ExceptionController(ErrorAttributes errorAttributes, ErrorProperties errorProperties, List<ErrorViewResolver> errorViewResolvers) {
+        super(errorAttributes, errorViewResolvers);
+        Assert.notNull(errorProperties, "ErrorProperties must not be null");
+        this.errorProperties = errorProperties;
+    }
+
+    @Override
+    public String getErrorPath() {
+        return this.errorProperties.getPath();
+    }
+
+    /**
+     * html错误
+     *
+     * @param request  request
+     * @param response response
+     * @return ModelAndView
+     */
+    @RequestMapping(produces = "text/html")
+    public ModelAndView errorHtml(HttpServletRequest request, HttpServletResponse response) {
+        HttpStatus status = getStatus(request);
+        Map<String, Object> model = Collections.unmodifiableMap(getErrorAttributes(request, isIncludeStackTrace(request, MediaType.TEXT_HTML)));
+        response.setStatus(status.value());
+        ModelAndView modelAndView = resolveErrorView(request, response, status, model);
+        return (modelAndView == null ? new ModelAndView("error", model) : modelAndView);
+    }
+
+    /**
+     * json错误
+     *
+     * @param request request
+     * @return ResponseEntity
+     */
+    @RequestMapping
+    @ResponseBody
+    public ResponseEntity<RestResponse<String>> error(HttpServletRequest request) {
+        Map<String, Object> body = getErrorAttributes(request, isIncludeStackTrace(request, MediaType.ALL));
+        HttpStatus status = getStatus(request);
+        ErrorResult errorResult = new ErrorResult();
+        errorResult.setType(NoHandlerFoundException.class.getName());
+        errorResult.setMessage("No handler found for " + request.getMethod() + " " + body.get("path").toString());
+        errorResult.setDate(new Date());
+        return new ResponseEntity<>(new RestResponse<>(status, errorResult), status);
+    }
+
+    /**
+     * Determine if the stacktrace attribute should be included.
+     *
+     * @param request  the source request
+     * @param produces the media type produced (or {@code MediaType.ALL})
+     * @return if the stacktrace attribute should be included
+     */
+    protected boolean isIncludeStackTrace(HttpServletRequest request, MediaType produces) {
+        ErrorProperties.IncludeStacktrace include = getErrorProperties().getIncludeStacktrace();
+        return include == ErrorProperties.IncludeStacktrace.ALWAYS || include == ErrorProperties.IncludeStacktrace.ON_TRACE_PARAM && getTraceParameter(request);
+    }
+
+    /**
+     * Provide access to the error properties.
+     *
+     * @return the error properties
+     */
+    protected ErrorProperties getErrorProperties() {
+        return this.errorProperties;
+    }
+
+}
@@ -0,0 +1,57 @@
+package org.itkk.udf.core.exception.handle;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.AutoConfigureBefore;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
+import org.springframework.boot.autoconfigure.web.*;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.servlet.DispatcherServlet;
+
+import javax.servlet.Servlet;
+import java.util.List;
+
+/**
+ * ErrorControllerConfig
+ */
+@Configuration
+@ConditionalOnWebApplication
+@ConditionalOnClass({Servlet.class, DispatcherServlet.class})
+@AutoConfigureBefore(WebMvcAutoConfiguration.class)
+@EnableConfigurationProperties(ResourceProperties.class)
+public class ExceptionControllerConfig {
+    /**
+     * errorViewResolvers
+     */
+    @Autowired(required = false)
+    private List<ErrorViewResolver> errorViewResolvers;
+
+    /**
+     * serverProperties
+     */
+    private final ServerProperties serverProperties;
+
+    /**
+     * 构造函数
+     *
+     * @param serverProperties serverProperties
+     */
+    public ExceptionControllerConfig(ServerProperties serverProperties) {
+        this.serverProperties = serverProperties;
+    }
+
+    /**
+     * exceptionController
+     *
+     * @param errorAttributes errorAttributes
+     * @return ExceptionController
+     */
+    @Bean
+    public ExceptionController exceptionController(ErrorAttributes errorAttributes) {
+        return new ExceptionController(errorAttributes, this.serverProperties.getError(), this.errorViewResolvers);
+    }
+
+
+}