浅谈SpringBoot如何封装统一响应体

浅谈SpringBoot如何封装统一响应体

一、前言

在上一篇 SpringBoot 参数校验 中我们对参数校验添加了异常处理，但是还是有不规范的地方，没有用统一响应体进行返回，在这篇文章中介绍如何封装统一响应体。

关于统一响应体的封装，没有一个标准答案，我在各种技术社区看了一遍，汇总了一个复用性比较好的方案。

二、添加结果类枚举

在项目目录下面建一个 responseEntity 的 package，然后在里面建一个 ResultEnum 枚举类，添加如下代码：

这边介绍一下枚举类的用法。枚举类的作用实际上就是定义常量，如果不使用枚举类，通常采用静态常量来表示：

public static final Integer OK_CODE = 200;

public static final String OK_MESSAGE = "成功";

public static final Integer BAD_REQUEST_CODE = 400;

public static final String BAD_REQUEST_MESSAGE = "参数错误";

这样的话存在一些问题，一是字段表意不明，特别是看别人的代码时，会很懵逼；第二当业务规模增大之后，可能要维护成百上千的静态常量，如果都写在一个文件里面，容易造成命名混淆，阅读也比较麻烦。

然后使用枚举类定义常量就比较方便，相当于一个接口，使用时只需要封装内部的数据类型，并且限定数据域。而且对于不同的枚举变量，可以调用不同的处理方法（实现枚举类的抽象方法可以做到这一点）。关于枚举类的一些知识点汇总如下：

使用enum定义的枚举类默认继承了java.lang.Enum，实现了java.lang.Comparable接口，且不能继承其他类，也不可以被继承。但枚举类可以实现一个或多个接口；

枚举类的所有实例必须放在第一行显示，不需使用new，不需显示调用构造方法，每个变量都是public static final修饰的，最终以分号结束。在之后的反编译中，我们就可以理解枚举类其实也是颗语法糖；

枚举类的构造方法是私有的，默认的就是 private，所以不用再添加 private；

枚举类内部常用的方法：

valueOf() ：返回当前枚举类的name属性，如果没有，则throw new java.lang.IllegalArgumentException()；

values() ：是编译器自动生成的方法，Enum中并没有该方法，返回包括所有枚举变量的数组；

toString() 和 name() ：两个方法一样，返回当前枚举类变量的name属性，如果觉得不够用，可以覆盖默认的 toString ，结合 SWITCH CASE 来灵活的实现 toString() 方法；

ordinal() ：枚举类会给所有的枚举变量一个默认的次序，该次序从0开始，是根据我们定义的次序来排序的。而ordinal()方法就是获取这个次序（或者说下标）；

compareTo() ：比较的是两个枚举变量的次序，返回两个次序相减后的结果；

定义了枚举类之后，在类的上面添加 lombok 的 @Getter 注解，给对象的每个属性添加 getter 方法，方便后面获取常量。例如要获取 OK 的状态码，就可以这样写：

ResultEnum.OK.getCode()

这边再解释下 @Data 、@Getter 和 @Setter 的区别：

@Data：注解在类上；提供类所有属性的 getter 和 setter 方法，此外还提供了equals、canEqual、hashCode、toString 方

@Getter：注解在属性上：为属性提供 getter 方法；注解再类上表示当前类中所有属性都生成getter方法

@Setter：注解在属性上:为属性提供 setter 方法；注解再类上表示当前类中所有属性都生成setter方法

三、添加统一结果类

还是在 responseEntity 目录下面，建一个 ServerResponse 类，添加如下代码：

@Data

public class ServerResponse {

private Boolean success;

private Integer code;

private String message;

private Object data;

// 构造方法设为私有

private ServerResponse() {}

public static ServerResponse ok(Object params) {

ServerResponse serverResponse = new ServerResponse();

serverResponse.setSuccess(ResultEnum.OK.getSuccess());

serverResponse.setCode(ResultEnum.OK.getCode());

serverResponse.setMessage(ResultEnum.OK.getMessage()); // 成功展示默认提示信息

serverResponse.setData(params); // 返回传入的参数

return serverResponse;

}

public static ServerResponse badRequest(@Nullable String message) {

ServerResponse serverResponse = new ServerResponse();

serverResponse.setSuccess(ResultEnum.BAD_REQUEST.getSuccess());

serverResponse.setCode(ResultEnum.BAD_REQUEST.getCode());

serverResponse.setMessage(message != null ? message : ResultEnum.BAD_REQUEST.getMessage()); // 校验失败传入指定的提示信息

serverResponse.setData(null); // 校验失败不返回参数

return serverResponse;

}

}

在上面的代码中，成员变量和构造方法都是私有的，只有静态方法向外暴露。然后处理成功的方法，message 展示默认提示信息，即定义在枚举类里面的常量，data 是需要传给前端的 jsON 参数；处理参数错误的方法，message 展示传进去的错误信息，如果传的是 null ，则展示默认提示信息，即定义在枚举类里面的常量，data 是传给前端的参数，但是在参数错误的情况下就不需要传了，因此是 null 。

这边有一个问题，暂时不清楚 Java 是否支持函数参数可选，本人测试发现定义函数的时候有参数，但是调用的时候不传，IDE 会给错误提示，因此这里通过传 null 来解决

四、控制层返回

在定义了统一结果类之后，就可以在接口中使用了。还是用之前那个方法，通过 POST 请求获取用户信息，再原封不动返回过去：

@PostMapping("validateUser")

public ServerResponse userValidate(@RequestBody @Validated UserDTO userDTO) {

return ServerResponse.ok(null);

}

这边先给参数传 null ，不给前端进行返回，看一下响应的结果：

然后传递参数：

@PostMapping("validateUser")

public ServerResponse userValidate(@RequestBody @Validated UserDTO userDTO) {

return ServerResponse.ok(userDTO);

}

看一下响应的结果：

五、异常处理类使用统一响应体

然后我们给异常处理的方法也添加统一响应体：

@RestControllerAdvice

public class GlobalExceptionHandler {

@ExceptionHandler(value = MethodArgumentNotValidException.class)

public ServerResponse handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {

// 注意传进去的表达式有可能是 null

// 因此在 ServerResponse 对 message 是否为 null 进行了判断

// 如果是 null 就展示默认的提示内容

return ServerResponse.badRequest(Objects.requireNonNull(e.getBindingResult().getFieldError()).getDefaultMessage());

}

// 其他异常处理方法

}

然后我们模拟一下参数异常的情况：

这样看起来是正常了，但是存在一个问题，后端判断参数异常的时候，因为我们捕获了异常，所以返回给前端的状态码还是 200 ，如何让状态码改为 400 呢？在 SpringBoot 中指定 HTTP 状态码主要有三种方式：

HttpServletResponse

@ResponseStatus

ResponseEntity

这边使用第三种方式，具体的用法看一下代码应该就明白了。我们把刚才统一结果类的方法修改下：

public static ResponseEntity badRequest(@Nullable String message) {

ServerResponse serverResponse = new ServerResponse();

serverResponse.setSuccess(ResultEnum.BAD_REQUEST.getSuccess());

serverResponse.setCode(ResultEnum.BAD_REQUEST.getCode());

serverResponse.setMessage(message != null ? message : ResultEnum.BAD_REQUEST.getMessage()); // 校验失败传入指定的提示信息

serverResponse.setData(null); // 校验失败不返回参数

return new ResponseEntity<>(serverResponse, HttpStatus.BAD_REQUEST); // 使用 ResponseEntity 对象设置响应状态码

}

可以看到我们用一个 ResponseEntity 对象包裹了我们封装的响应体，然后返回了这个对象。其中第二个参数就是状态码，HttpStatus.BAD_REQUEST 就代表 400 。然后我们还要修改下异常处理类的返回类型：

@RestControllerAdvice

public class GlobalExceptionHandler {

@ExceptionHandler(value = MethodArgumentNotValidException.class)

public ResponseEntity handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {

return ServerResponse.badRequest(Objects.requireNonNull(e.getBindingResult().getFieldError()).getDefaultMessage());

}

// 其他异常处理方法

}

再来调试一下，这下状态码正常了：

在统一结果类中所有的方法都根据上面的示例进行修改即可，我这边添加了几个方法，给各位参考下，具体可以根据业务场景进行添加：

@Data

public class ServerResponse {

private Boolean success;

private Integer code;

private String message;

private Object data;

// 构造方法设为私有

private ServerResponse() {}

/**

* 200 请求成功

* @param params 传给前端的参数

* @return ResponseEntity

*/

public static ResponseEntity ok(Object params) {

ServerResponse serverResponse = new ServerResponse();

serverRespoWuWpIqoonse.setSuccess(ResultEnum.OK.getSuccess());

serverResponse.setCode(ResultEnum.OK.getCode());

serverResponse.setMessage(ResultEnum.OK.getMessage()); // 成功展示默认提示信息

serverResponse.setData(params); // 返回传入的参数

return new ResponseEntity<>(serverResponse, HttpStatus.OK);

}

/**

* 201 创建成功

* @return ResponseEntity

*/

public static ResponseEntity created() {

ServerResponse serverResponse = new ServerResponse();

serverResponse.setShttp://uccess(ResultEnum.CREATED.getSuccess());

serverResponse.setCode(ResultEnum.CREATED.getCode());

serverResponse.setMessage(ResultEnum.CREATED.getMessage());

// serverResponse.setData(null);

return new ResponseEntity<>(serverResponse, HttpStatus.CREATED);

}

/**

* 204 请求成功，没有响应体

* @return ResponseEntity

*/

public static ResponseEntity noContent() {

return new ResponseEntity<>(null, HttpStatus.NO_CONTENT);

}

/**

* 400 参数错误

* @param message 自定义错误信息

* @return ResponseEntity

*/

public static ResponseEntity badRequest(@Nullable String message) {

ServerResponse serverResponse = new ServerResponse();

serverResponse.setSuccess(ResultEnum.BAD_REQUEST.getSuccess());

serverResponse.setCode(ResultEnum.BAD_REQUEST.getCode());

serverResponse.setMessage(message != null ? message : ResultEnum.BAD_REQUEST.getMessage()); // 校验失败传入指定的提示信息

// serverResponse.setData(null); // 校验失败不返回参数

return new ResponseEntity<>(serverResponse, HttpStatus.BAD_REQUEST); // 使用 ResponseEntity 对象设置响应状态码

}

}

此外，@ResponseStatus 也是一种设置状态码常用的方法，只需要在 Controller 方法中加一个注解就可以：

@PostMapping("validateUser")

@ResponseStatus(code=HttpStatus.BAD_REQUEST, reasoWuWpIqoon="参数异常")

public ServerResponse userValidate(@RequestBody @Validated UserDTO userDTO) {

return ServerResponse.ok(userDTO);

}

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。