1. 首页
  2. >
  3. 技术代码
  4. >
  5. java技术

SpringBoot中如何实现接口的统一返回和异常的统一捕获

接口的统一返回

在开发公司接口时,发现Controller层的接口返回都需要用一个Result包裹,如下图所示:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

图示代码中无论是创建接口或者查询接口,这里都需要用一个Result去接收,我们来看看Result的结构:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

这个Result中有几个字段:

code:状态码

message:状态信息

data:装载正真返回的数据

exception:异常数据

然后我们测试下接口,看看返回样式:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

SpringBoot中如何实现接口的统一返回和异常的统一捕获

调用接口,返回格式为:

{   "code": 200,   "message""SUCCESS",   "data": {       ...   }   "exception": xxx } 

写完后我就在思考,每个接口都需要包一层Result,感觉影响了接口的可读性。作为开发来说这样包裹一层也很麻烦。

那么有没有一种方法,可以通过框架自动包裹一层Result,开发只要在Controller层直接返回实体就行了呢?

还真有这样的方法,能实现Controller层接口的统一返回:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

如上代码,我们使用ResponseBodyAdvice来拦截Controller层方法默认返回参数。说白了就是个拦截器。主要是看beforeBodyWrite()方法,在这个方法中,如果Controller中的返回已经是Result,那就直接返回Result。如果不是,那就使用Result去包装。

我们来看看效果:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

如上代码,我们直接返回实体,我们看看Swagger的返回:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

swagger的返回的格式是我们拦截器的格式。

这里我又有一个疑问,接口正常返回已经被Result包裹了,如果接口抛异常该怎么返回一样的格式呢?

全局的异常捕获

这里就需要全局的异常捕捉了。关于全局异常捕捉,相信很多童鞋都知道了:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

我们需要写一个捕捉类,在上面添加@ControllerAdvice注解,然后编写处理异常的方法:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

我们添加@ResponseBody注解和@ExceptionHandler注解,这里的value = Exception.class,表示我们捕捉Exception类型的异常。我们就可以直接抛异常了:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

也可以这么抛:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

我们测试,如果代码异常,接口就会返回:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

格式符合预期。

但是这里还是有一个问题,每次我都需要写if语句:

if(条件) {    throw new RuntimeException("..."); } 

这样写还是太麻烦,而且啥异常都抛RuntimeException。这样还是太粗糙了。

于是我决定自定义一个业务异常,并且封装一些异常抛出方法,说干就干。

自定义业务异常

SpringBoot中如何实现接口的统一返回和异常的统一捕获

我们定义一个业务异常,这里面封装了异常状态码,和异常信息数据。

优雅抛异常

然后我们编写一个业务异常判断类:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

这里只截取了部分代码,里面其实只有两个方法checkArgument()checkNotNull()。他们有什么用呢?如果要做非空校验,一般是这么做的:

if (updateEntity == null) {     throw new RuntimeException("传入参数为null"); } 

但是现在可以这么做了:

BusinessExceptionAssert.checkNotNull(updateEntity, "参数不能为null"); 

抛出的是我自定义的业务异常。

如果是一般的逻辑校验呢?以前的代码是这么写的:

if(!"1".equals(id)) {    throw new RuntimeException(String.format("参数id【%s】只能为1",id)); } 

现在可以有更优雅的实现方式:

BusinessExceptionAssert.checkArgument("1".equals(id), 500"参数id【%s】只能为1",id); 

因为抛的是业务异常,我们还可以自定义异常码。

完整测试代码:

 @ApiOperation(value = "test-测试异常", httpMethod = "POST", notes = "测试代码,勿动")  @PostMapping(value = "testBusinessException")  public int testBusinessException(@RequestParam String id) {      BusinessExceptionAssert.checkArgument("1".equals(id), 500"参数id【%s】只能为1",id);      return 1;  } 

我们测试结果:

SpringBoot中如何实现接口的统一返回和异常的统一捕获

符合预期。

这篇文章写到这里就结束了,本文主要讲解了:

如何实现接口统一返回

如何自定义业务异常,并被统一捕获

如何优雅的抛异常

如果还有改进的,欢迎大家积极交流。