SpringBoot中干掉Whitelabel Error Page,返回自定义内容

SpringBoot中干掉Whitelabel Error Page,返回自定义内容

1. 引言

SpringBoot中对于错误请求的页面是长这样的,

SpringBoot中干掉Whitelabel Error Page,返回自定义内容_第1张图片

然而我们在访问在一些网站时,如果请求错误,一般都会有友好美观的提示,比如知乎这个,这比起一堆错误信息要友好的多了。

SpringBoot中干掉Whitelabel Error Page,返回自定义内容_第2张图片

我们可以根据项目业务来自定义错误请求(RequestMapping中没有映射到的请求)的处理,比如返回自定义错误页面或者Json字符串。

2. 分析

我们看看SpringBoot中对于错误请求是如何处理的。SpringBoot项目中搜索Whitelabel定位到类WhitelabelErrorViewConfiguration,可以看到它是ErrorMvcAutoConfiguration的一个静态内部类,而且正是这个类处理的错误请求的,代码中的defaultErrorView正是我们看到的默认错误页面。

@Configuration(proxyBeanMethods = false)
@ConditionalOnProperty(prefix = "server.error.whitelabel", name = "enabled", matchIfMissing = true)
@Conditional(ErrorMvcAutoConfiguration.ErrorTemplateMissingCondition.class)
protected static class WhitelabelErrorViewConfiguration {
     

	private final ErrorMvcAutoConfiguration.StaticView defaultErrorView = new ErrorMvcAutoConfiguration.StaticView();

	@Bean(name = "error")
	@ConditionalOnMissingBean(name = "error")
	public View defaultErrorView() {
     
		return this.defaultErrorView;
	}
	// and so on...
}

仔细研究这个代码,我们不难发现,我们至少有两种方法可以替换掉默认的实现自定义错误页面。

  • 入手点1:
@ConditionalOnProperty(prefix = "server.error.whitelabel", name = "enabled", matchIfMissing = true)
  • 入手点2:
@Bean(name = "error")
@ConditionalOnMissingBean(name = "error")

3. 尝试

3.1 尝试 1

@ConditionalOnProperty(prefix = "server.error.whitelabel", name = "enabled", matchIfMissing = true)

我们看到server.error.whitelabel.enabled控制了这个类是否装配,我们可以在配置中将其设为false

server:
  error:
    whitelabel:
      enabled: false
    path: /error

该配置类为ErrorProperties,默认的错误请求为/error,将 whitelabel 禁用后在 controller 中定义请求 /error返回自定义内容。

@Controller
public class ErrorController {
     

	@RequestMapping("/error")
	@ResponseBody
	public R error(HttpServletRequest request) {
     
		Integer status = (Integer) request.getAttribute(RequestDispatcher.ERROR_STATUS_CODE);
		if (Objects.nonNull(status)) {
     
			HttpStatus httpStatus = HttpStatus.valueOf(status);
			return RUtils.fail(httpStatus);
		}
		return RUtils.fail("Error");
	}

}

运行!一顿操作猛如虎,仔细一看原地杵,想法不错,但疯狂报错:

org.springframework.beans.factory.UnsatisfiedDependencyException: Error creating bean with name 'documentationPluginsBootstrapper' defined in URL [jar:file:/D:/Environment/Maven/repository/io/springfox/springfox-spring-web/2.9.2/springfox-spring-web-2.9.2.jar!/springfox/documentation/spring/web/plugins/DocumentationPluginsBootstrapper.class]: Unsatisfied dependency expressed through constructor parameter 1; nested exception is org.springframework.beans.factory.UnsatisfiedDependencyException: Error creating bean with name 'webMvcRequestHandlerProvider' defined in URL [jar:file:/D:/Environment/Maven/repository/io/springfox/springfox-spring-web/2.9.2/springfox-spring-web-2.9.2.jar!/springfox/documentation/spring/web/plugins/WebMvcRequestHandlerProvider.class]: Unsatisfied dependency expressed through constructor parameter 1; nested exception is org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'requestMappingHandlerMapping' defined in class path resource [xxx/xxx/xxx/config/WebMvcConfig.class]: Invocation of init method failed; nested exception is java.lang.IllegalStateException: Ambiguous mapping. Cannot map 'basicErrorController' method 
org.springframework.boot.autoconfigure.web.servlet.error.BasicErrorController#error(HttpServletRequest)
to {
      /error}: There is already 'errorController' bean method

具体报错原因未作分析,怀疑是我项目设置有问题,笔者使用的SpringBoot版本是2.3.6.RELEASE,大家可以尝试一些行不行。既然第一个方法尝试接着尝试失败,那就试试第二个吧。

3.2 尝试2

@Bean(name = "error")
@ConditionalOnMissingBean(name = "error")

这个操作更简单,我们只要向IoC 容器中注入一个名为errorBean即可,返回类型为View:

@Configuration
public class MyConfig {
     
    @Bean("error")
	public View error() {
     
		ModelAndView view = new ModelAndView(new MappingJackson2JsonView());
		return view.getView();
	}
}

其中new MappingJackson2JsonView()返回的Json字符串的 View ,当然你也可以根据业务需求灵活使用,如果想获取HttpServletRequest可以这样获取,但使用时需要注意判空,因为Bean实例化注入是可能获取不到HttpServletRequest而造成NPE,项目启动失败。

public static HttpServletRequest getRequest() {
   return Optional.ofNullable(RequestContextHolder.getRequestAttributes())
         .map(r -> ((ServletRequestAttributes) r).getRequest())
         .orElse(null);
}

结果当然是没问题的,会返回以下信息:

result

4. 总结

解决问题:

  1. 搜索引擎找解决方法
  2. 深入源码尝试自己解决

你可能感兴趣的:(SpringBoot,spring,boot,spring,java)