百木园-与人分享，
就是让自己快乐。
当前位置：百木园 > 野生技术 > 正文

SpringBoot 如何统一后端返回格式

2022-04-26 分类：野生技术 阅读(212) 评论(0)

在前后端分离的项目中后端返回的格式一定要友好，不然会对前端的开发人员带来很多的工作量。那么SpringBoot如何做到统一的后端返回格式呢？今天我们一起来看看。

为什么要对SpringBoot返回统一的标准格式

在默认情况下，SpringBoot的返回格式常见的有三种：

返回String

@GetMapping(\"/hello\")
public String hello() {
    return  \"hello\";
}

此时调用接口获取到的返回值是这样：

hello

返回自定义对象

@GetMapping(\"/student\")
public Student getStudent() {
        Student student = new Student();
        student.setId(1);
        student.setName(\"didiplus\");
        return  student;
}


//student的类
@Data
public class Student {
    private Integer id;
    private String name;
}

此时调用接口获取到的返回值是这样：

{\"id\":1,\"name\":\"didiplus\"}

接口异常

@GetMapping(\"/error\")
public int error(){
    int i = 9/0;
    return i;
}

此时调用接口获取到的返回值是这样：
image.png

SpringBoot的版本是v2.6.7，

定义返回对象

package com.didiplus.common.web.response;

import lombok.Data;

import java.io.Serializable;

/**
 * Author: didiplus
 * Email: 972479352@qq.com
 * CreateTime: 2022/4/24
 * Desc: Ajax 返 回 JSON 结 果 封 装 数 据
 */

@Data
public class Result<T> implements Serializable {

    /**
     * 是否返回成功
     */
    private boolean success;

    /**
     * 错误状态
     */
    private int code;

    /***
     * 错误信息
     */
    private String msg;

    /**
     * 返回数据
     */
    private T data;

    /**
     * 时间戳
     */
    private long timestamp ;


    public Result (){
        this.timestamp = System.currentTimeMillis();
    }
    /**
     * 成功的操作
     */
    public static <T> Result<T> success() {
        return  success(null);
    }

    /**
     * 成 功 操 作 , 携 带 数 据
     */
    public static <T> Result<T> success(T data){
        return success(ResultCode.RC100.getMessage(),data);
    }

    /**
     * 成 功 操 作, 携 带 消 息
     */
    public static <T> Result<T> success(String message) {
        return success(message, null);
    }

        /**
         * 成 功 操 作, 携 带 消 息 和 携 带 数 据
         */
    public static <T> Result<T> success(String message, T data) {
        return success(ResultCode.RC100.getCode(), message, data);
    }

    /**
     * 成 功 操 作, 携 带 自 定 义 状 态 码 和 消 息
     */
    public static <T> Result<T> success(int code, String message) {
        return success(code, message, null);
    }

    public static <T> Result<T> success(int code,String message,T data) {
        Result<T> result = new Result<T>();
        result.setCode(code);
        result.setMsg(message);
        result.setSuccess(true);
        result.setData(data);
        return result;
    }

    /**
     * 失 败 操 作, 默 认 数 据
     */
    public static <T> Result<T> failure() {
        return failure(ResultCode.RC100.getMessage());
    }

    /**
     * 失 败 操 作, 携 带 自 定 义 消 息
     */
    public static <T> Result<T> failure(String message) {
        return failure(message, null);
    }

    /**
     * 失 败 操 作, 携 带 自 定 义 消 息 和 数 据
     */
    public static <T> Result<T> failure(String message, T data) {
        return failure(ResultCode.RC999.getCode(), message, data);
    }

    /**
     * 失 败 操 作, 携 带 自 定 义 状 态 码 和 自 定 义 消 息
     */
    public static <T> Result<T> failure(int code, String message) {
        return failure(ResultCode.RC999.getCode(), message, null);
    }

    /**
     * 失 败 操 作, 携 带 自 定 义 状 态 码 , 消 息 和 数 据
     */
    public static <T> Result<T> failure(int code, String message, T data) {
        Result<T> result = new Result<T>();
        result.setCode(code);
        result.setMsg(message);
        result.setSuccess(false);
        result.setData(data);
        return result;
    }

    /**
     * Boolean 返 回 操 作, 携 带 默 认 返 回 值
     */
    public static <T> Result<T> decide(boolean b) {
        return decide(b, ResultCode.RC100.getMessage(), ResultCode.RC999.getMessage());
    }

    /**
     * Boolean 返 回 操 作, 携 带 自 定 义 消 息
     */
    public static <T> Result<T> decide(boolean b, String success, String failure) {
        if (b) {
            return success(success);
        } else {
            return failure(failure);
        }
    }
}

定义状态码

package com.didiplus.common.web.response;

import lombok.Getter;

/**
 * Author: didiplus
 * Email: 972479352@qq.com
 * CreateTime: 2022/4/24
 * Desc: 统 一 返 回 状 态 码
 */
public enum ResultCode {
    /**操作成功**/
    RC100(100,\"操作成功\"),
    /**操作失败**/
    RC999(999,\"操作失败\"),
    /**服务限流**/
    RC200(200,\"服务开启限流保护,请稍后再试!\"),
    /**服务降级**/
    RC201(201,\"服务开启降级保护,请稍后再试!\"),
    /**热点参数限流**/
    RC202(202,\"热点参数限流,请稍后再试!\"),
    /**系统规则不满足**/
    RC203(203,\"系统规则不满足要求,请稍后再试!\"),
    /**授权规则不通过**/
    RC204(204,\"授权规则不通过,请稍后再试!\"),
    /**access_denied**/
    RC403(403,\"无访问权限,请联系管理员授予权限\"),
    /**access_denied**/
    RC401(401,\"匿名用户访问无权限资源时的异常\"),
    /**服务异常**/
    RC500(500,\"系统异常，请稍后重试\"),

    INVALID_TOKEN(2001,\"访问令牌不合法\"),
    ACCESS_DENIED(2003,\"没有权限访问该资源\"),
    CLIENT_AUTHENTICATION_FAILED(1001,\"客户端认证失败\"),
    USERNAME_OR_PASSWORD_ERROR(1002,\"用户名或密码错误\"),
    UNSUPPORTED_GRANT_TYPE(1003, \"不支持的认证模式\");

    /**自定义状态码**/
    @Getter
    private final int code;

    /**
     * 携 带 消 息
     */
    @Getter
    private final String message;
    /**
     * 构 造 方 法
     */
    ResultCode(int code, String message) {

        this.code = code;

        this.message = message;
    }
}

统一返回格式

    @GetMapping(\"/hello\")
    public Result<String> hello() {
        return  Result.success(\"操作成功\",\"hello\");
    }

此时调用接口获取到的返回值是这样：

{\"success\":true,\"code\":100,\"msg\":\"操作成功\",\"data\":\"hello\",\"timestamp\":1650785058049}

这样确实已经实现了我们想要的结果，我在很多项目中看到的都是这种写法，在Controller层通过Result.success()对返回结果进行包装后返回给前端。这样显得不够专业而且不够优雅。 所以呢我们需要对代码进行优化，目标就是不要每个接口都手工制定Result返回值。

高级实现方式

要优化这段代码很简单，我们只需要借助SpringBoot提供的ResponseBodyAdvice即可。

ResponseBodyAdvice的源码：

public interface ResponseBodyAdvice<T> {
		/**
		* 是否支持advice功能
		* true 支持，false 不支持
		*/
    boolean supports(MethodParameter var1, Class<? extends HttpMessageConverter<?>> var2);

	  /**
		* 对返回的数据进行处理
		*/
    @Nullable
    T beforeBodyWrite(@Nullable T var1, MethodParameter var2, MediaType var3, Class<? extends HttpMessageConverter<?>> var4, ServerHttpRequest var5, ServerHttpResponse var6);
}

只需要编写一个具体实现类即可

@RestControllerAdvice
public class ResponseAdvice  implements ResponseBodyAdvice<Object> {


    @Autowired
    ObjectMapper objectMapper;

    @Override
    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
        return true;
    }

    @SneakyThrows
    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response)  {
        if (body instanceof  String){
            return objectMapper.writeValueAsString(Result.success(ResultCode.RC100.getMessage(),body));
        }
        return Result.success(ResultCode.RC100.getMessage(),body);
    }
}

需要注意两个地方：
@RestControllerAdvice注解 @RestControllerAdvice是@RestController注解的增强，可以实现三个方面的功能：

  1. 全局异常处理
  2. 全局数据绑定
  3. 全局数据预处理

String类型判断

        if (body instanceof  String){
            return objectMapper.writeValueAsString(Result.success(ResultCode.RC100.getMessage(),body));
        }

这段代码一定要加，如果Controller直接返回String的话，SpringBoot是直接返回，故我们需要手动转换成json。 经过上面的处理我们就再也不需要通过ResultData.success()来进行转换了，直接返回原始数据格式，SpringBoot自动帮我们实现包装类的封装。

    @GetMapping(\"/hello\")
    public String hello() {
        return \"hello,didiplus\";
    }


    @GetMapping(\"/student\")
    public Student getStudent() {
        Student student = new Student();
        student.setId(1);
        student.setName(\"didiplus\");
        return student;
    }

此时我们调用接口返回的数据结果为：

{
	\"success\": true,
	\"code\": 100,
	\"msg\": \"操作成功\",
	\"data\": \"hello,didiplus\",
	\"timestamp\": 1650786993454
}

本文来自博客园，作者：北根娃，转载请注明原文链接：https://www.cnblogs.com/alanlin/p/16191008.html

来源：https://www.cnblogs.com/alanlin/p/16191008.html
本站部分图文来源于网络，如有侵权请联系删除。

未经允许不得转载：百木园 » SpringBoot 如何统一后端返回格式
标签：

相关推荐

  • 暂无文章

热门文章