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

接口的统一返回

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

图片

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

这个Result中有几个字段:

  • code:状态码

  • message:状态信息

  • data:装载正真返回的数据

  • exception:异常数据

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

图片

图片

调用接口,返回格式为:

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

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

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

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

图片

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

我们来看看效果:

图片

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

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

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

全局的异常捕获

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

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

图片

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

图片

也可以这么抛:

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

格式符合预期。

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

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

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

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

自定义业务异常

图片

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

优雅抛异常

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

图片

这里只截取了部分代码,里面其实只有两个方法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;
 }

我们测试结果:

图片

符合预期。

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

  • 如何实现接口统一返回

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

  • 如何优雅的抛异常

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

猜你喜欢

转载自blog.csdn.net/wujialv/article/details/111879692