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;
 }