分享SpringBoot具体是怎样封装统一响应体的

2021-07-27 09:33:28 浏览数 (3)

一、前言

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

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

二、添加结果类枚举

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

2021052416032678

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

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

    // 其他异常处理方法
}

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

2021052416032681

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

  • HttpServletResponse
  • @ResponseStatus
  • ResponseEntity

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

public static ResponseEntity<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 new ResponseEntity<>(serverResponse, HttpStatus.BAD_REQUEST); // 使用 ResponseEntity 对象设置响应状态码
}

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

@RestControllerAdvice
public class GlobalExceptionHandler {
    @ExceptionHandler(value = MethodArgumentNotValidException.class)
    public ResponseEntity<ServerResponse> handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
        return ServerResponse.badRequest(Objects.requireNonNull(e.getBindingResult().getFieldError()).getDefaultMessage());
    }

    // 其他异常处理方法
}

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

2021052416032682

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

@Data
public class ServerResponse {
    private Boolean success;
    private Integer code;
    private String message;
    private Object data;

    // 构造方法设为私有
    private ServerResponse() {}

    /**
     * 200 请求成功
     * @param params 传给前端的参数
     * @return ResponseEntity<ServerResponse>
     */
    public static ResponseEntity<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 new ResponseEntity<>(serverResponse, HttpStatus.OK);
    }

    /**
     * 201 创建成功
     * @return ResponseEntity<ServerResponse>
     */
    public static ResponseEntity<ServerResponse> created() {
        ServerResponse serverResponse = new ServerResponse();
        serverResponse.setSuccess(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<ServerResponse>
     */
    public static ResponseEntity<ServerResponse> noContent() {
        return new ResponseEntity<>(null, HttpStatus.NO_CONTENT);
    }

    /**
     * 400 参数错误
     * @param message 自定义错误信息
     * @return ResponseEntity<ServerResponse>
     */
    public static ResponseEntity<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 new ResponseEntity<>(serverResponse, HttpStatus.BAD_REQUEST); // 使用 ResponseEntity 对象设置响应状态码
    }
}

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

@PostMapping("validateUser")
@ResponseStatus(code=HttpStatus.BAD_REQUEST, reason="参数异常")
public ServerResponse userValidate(@RequestBody @Validated UserDTO userDTO) {
    return ServerResponse.ok(userDTO);
}

到此这篇关于 SpringBoot 框架是如何封装统一响应体的文章就介绍到这了,想要了解更多相关 SpringBoot 封装统一响应体的其他内容请搜索W3Cschool以前的文章或继续浏览下面的相关文章，希望大家以后多多支持我们！


Java

1 人点赞