文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

SpringBoot 如何统一后端返回格式?老鸟们都是这样玩的!

2024-12-03 01:55

关注

大家好,我是飘渺。

今天我们来聊一聊在基于SpringBoot前后端分离开发模式下,如何友好的返回统一的标准格式以及如何优雅的处理全局异常。

首先我们来看看为什么要返回统一的标准格式?

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

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

第一种:返回 String

  1. @GetMapping("/hello"
  2. public String getStr(){ 
  3.   return "hello,javadaily"

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

  1. hello,javadaily 

第二种:返回自定义对象

  1. @GetMapping("/aniaml"
  2. public Aniaml getAniaml(){ 
  3.   Aniaml aniaml = new Aniaml(1,"pig"); 
  4.   return aniaml; 

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

  1.   "id": 1, 
  2.   "name""pig" 

第三种:接口异常

  1. @GetMapping("/error"
  2. public int error(){ 
  3.     int i = 9/0; 
  4.     return i; 

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

  1.   "timestamp""2021-07-08T08:05:15.423+00:00"
  2.   "status": 500, 
  3.   "error""Internal Server Error"
  4.   "path""/wrong" 

基于以上种种情况,如果你和前端开发人员联调接口她们就会很懵逼,由于我们没有给他一个统一的格式,前端人员不知道如何处理返回值。

还有甚者,有的同学比如小张喜欢对结果进行封装,他使用了Result对象,小王也喜欢对结果进行包装,但是他却使用的是Response对象,当出现这种情况时我相信前端人员一定会抓狂的。

所以我们项目中是需要定义一个统一的标准返回格式的。

定义返回标准格式

一个标准的返回格式至少包含3部分:

  1. status 状态值:由后端统一定义各种返回结果的状态码
  2. message 描述:本次接口调用的结果描述
  3. data 数据:本次返回的数据。
  1.   "status":"100"
  2.   "message":"操作成功"
  3.   "data":"hello,javadaily" 

当然也可以按需加入其他扩展值,比如我们就在返回对象中添加了接口调用时间

timestamp: 接口调用时间

定义返回对象

  1. @Data 
  2. public class ResultData { 
  3.    
  4.   private int status; 
  5.   private String message; 
  6.   private T data; 
  7.   private long timestamp ; 
  8.  
  9.  
  10.   public ResultData (){ 
  11.     this.timestamp = System.currentTimeMillis(); 
  12.   } 
  13.  
  14.  
  15.   public static  ResultData success(T data) { 
  16.     ResultData resultData = new ResultData<>(); 
  17.     resultData.setStatus(ReturnCode.RC100.getCode()); 
  18.     resultData.setMessage(ReturnCode.RC100.getMessage()); 
  19.     resultData.setData(data); 
  20.     return resultData; 
  21.   } 
  22.  
  23.   public static  ResultData fail(int code, String message) { 
  24.     ResultData resultData = new ResultData<>(); 
  25.     resultData.setStatus(code); 
  26.     resultData.setMessage(message); 
  27.     return resultData; 
  28.   } 
  29.  

定义状态码

  1. public enum ReturnCode { 
  2.      
  3.     RC100(100,"操作成功"), 
  4.      
  5.     RC999(999,"操作失败"), 
  6.      
  7.     RC200(200,"服务开启限流保护,请稍后再试!"), 
  8.      
  9.     RC201(201,"服务开启降级保护,请稍后再试!"), 
  10.      
  11.     RC202(202,"热点参数限流,请稍后再试!"), 
  12.      
  13.     RC203(203,"系统规则不满足要求,请稍后再试!"), 
  14.      
  15.     RC204(204,"授权规则不通过,请稍后再试!"), 
  16.      
  17.     RC403(403,"无访问权限,请联系管理员授予权限"), 
  18.      
  19.     RC401(401,"匿名用户访问无权限资源时的异常"), 
  20.      
  21.     RC500(500,"系统异常,请稍后重试"), 
  22.  
  23.     INVALID_TOKEN(2001,"访问令牌不合法"), 
  24.     ACCESS_DENIED(2003,"没有权限访问该资源"), 
  25.     CLIENT_AUTHENTICATION_FAILED(1001,"客户端认证失败"), 
  26.     USERNAME_OR_PASSWORD_ERROR(1002,"用户名或密码错误"), 
  27.     UNSUPPORTED_GRANT_TYPE(1003, "不支持的认证模式"); 
  28.  
  29.  
  30.  
  31.      
  32.     private final int code; 
  33.      
  34.     private final String message; 
  35.  
  36.     ReturnCode(int code, String message){ 
  37.         this.code = code; 
  38.         this.message = message; 
  39.     } 
  40.  
  41.  
  42.     public int getCode() { 
  43.         return code; 
  44.     } 
  45.  
  46.     public String getMessage() { 
  47.         return message; 
  48.     } 

统一返回格式

  1. @GetMapping("/hello"
  2. public ResultData getStr(){ 
  3.  return ResultData.success("hello,javadaily"); 

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

  1.   "status": 100, 
  2.   "message""hello,javadaily"
  3.   "data"null
  4.   "timestamp": 1625736481648, 
  5.   "httpStatus": 0 

这样确实已经实现了我们想要的结果,我在很多项目中看到的都是这种写法,在Controller层通过ResultData.success()对返回结果进行包装后返回给前端。

看到这里我们不妨停下来想想,这样做有什么弊端呢?

最大的弊端就是我们后面每写一个接口都需要调用ResultData.success()这行代码对结果进行包装,重复劳动,浪费体力;

而且还很容易被其他老鸟给嘲笑。

所以呢我们需要对代码进行优化,目标就是不要每个接口都手工制定ResultData返回值。

高级实现方式

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

ResponseBodyAdvice的作用:拦截Controller方法的返回值,统一处理返回值/响应体,一般用来统一返回格式,加解密,签名等等。

先来看下ResponseBodyAdvice的源码:

  1. public interface ResponseBodyAdvice { 
  2.    
  3.     boolean supports(MethodParameter var1, Class> var2); 
  4.  
  5.     
  6.     @Nullable 
  7.     T beforeBodyWrite(@Nullable T var1, MethodParameter var2, MediaType var3, Class> var4, ServerHttpRequest var5, ServerHttpResponse var6); 

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

  1.  
  2. @RestControllerAdvice 
  3. public class ResponseAdvice implements ResponseBodyAdvice { 
  4.     @Autowired 
  5.     private ObjectMapper objectMapper; 
  6.  
  7.     @Override 
  8.     public boolean supports(MethodParameter methodParameter, Class> aClass) { 
  9.         return true
  10.     } 
  11.  
  12.     @SneakyThrows 
  13.     @Override 
  14.     public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) { 
  15.         if(o instanceof String){ 
  16.             return objectMapper.writeValueAsString(ResultData.success(o)); 
  17.         }         
  18.         return ResultData.success(o); 
  19.     } 
  20. 需要注意两个地方:

    • @RestControllerAdvice注解

    @RestControllerAdvice是@RestController注解的增强,可以实现三个方面的功能:

    1. 全局异常处理
    2. 全局数据绑定
    3. 全局数据预处理
    • String类型判断
    1. if(o instanceof String){ 
    2.   return objectMapper.writeValueAsString(ResultData.success(o)); 
    3. }  

    这段代码一定要加,如果Controller直接返回String的话,SpringBoot是直接返回,故我们需要手动转换成json。

    经过上面的处理我们就再也不需要通过ResultData.success()来进行转换了,直接返回原始数据格式,SpringBoot自动帮我们实现包装类的封装。

    1. @GetMapping("/hello"
    2. public String getStr(){ 
    3.     return "hello,javadaily"

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

    1. @GetMapping("/hello"
    2. public String getStr(){ 
    3.   return "hello,javadaily"

    是不是感觉很完美,别急,还有个问题在等着你呢。

    接口异常问题

    此时有个问题,由于我们没对Controller的异常进行处理,当我们调用的方法一旦出现异常,就会出现问题,比如下面这个接口

    1. @GetMapping("/wrong"
    2. public int error(){ 
    3.     int i = 9/0; 
    4.     return i; 

    返回的结果为:

    这显然不是我们想要的结果,接口都报错了还返回操作成功的响应码,前端看了会打人的。

    别急,接下来我们进入第二个议题,如何优雅的处理全局异常。

    SpringBoot为什么需要全局异常处理器

    不用手写try...catch,由全局异常处理器统一捕获

    使用全局异常处理器最大的便利就是程序员在写代码时不再需要手写try...catch了,前面我们讲过,默认情况下SpringBoot出现异常时返回的结果是这样:

    1.   "timestamp""2021-07-08T08:05:15.423+00:00"
    2.   "status": 500, 
    3.   "error""Internal Server Error"
    4.   "path""/wrong" 

    这种数据格式返回给前端,前端是看不懂的,所以这时候我们一般通过try...catch来处理异常

    1. @GetMapping("/wrong"
    2. public int error(){ 
    3.     int i; 
    4.     try{ 
    5.         i = 9/0; 
    6.     }catch (Exception e){ 
    7.         log.error("error:{}",e); 
    8.         i = 0; 
    9.     } 
    10.     return i; 

    我们追求的目标肯定是不需要再手动写try...catch了,而是希望由全局异常处理器处理。

    对于自定义异常,只能通过全局异常处理器来处理

    1. @GetMapping("error1"
    2. public void empty(){ 
    3.  throw  new RuntimeException("自定义异常"); 

    当我们引入Validator参数校验器的时候,参数校验不通过会抛出异常,此时是无法用try...catch捕获的,只能使用全局异常处理器。

    SpringBoot集成参数校验请参考这篇文章SpringBoot开发秘籍 - 集成参数校验及高阶技巧

    如何实现全局异常处理器

    1. @Slf4j 
    2. @RestControllerAdvice 
    3. public class RestExceptionHandler { 
    4.      
    5.     @ExceptionHandler(Exception.class) 
    6.     @ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR) 
    7.     public ResultData exception(Exception e) { 
    8.         log.error("全局异常信息 ex={}", e.getMessage(), e); 
    9.         return ResultData.fail(ReturnCode.RC500.getCode(),e.getMessage()); 
    10.     } 
    11.  

    有三个细节需要说明一下:

    1. @RestControllerAdvice,RestController的增强类,可用于实现全局异常处理器
    2. @ExceptionHandler,统一处理某一类异常,从而减少代码重复率和复杂度,比如要获取自定义异常可以@ExceptionHandler(BusinessException.class)
    3. @ResponseStatus指定客户端收到的http状态码

    体验效果

    这时候我们调用如下接口:

    1. @GetMapping("error1"
    2. public void empty(){ 
    3.     throw  new RuntimeException("自定义异常"); 

    返回的结果如下:

    1.   "status": 500, 
    2.   "message""自定义异常"
    3.   "data"null
    4.   "timestamp": 1625795902556 

    基本满足我们的需求了。

    但是当我们同时启用统一标准格式封装功能ResponseAdvice和RestExceptionHandler全局异常处理器时又出现了新的问题:

    1.   "status": 100, 
    2.   "message""操作成功"
    3.   "data": { 
    4.     "status": 500, 
    5.     "message""自定义异常"
    6.     "data"null
    7.     "timestamp": 1625796167986 
    8.   }, 
    9.   "timestamp": 1625796168008 

    此时返回的结果是这样,统一格式增强功能会给返回的异常结果再次封装,所以接下来我们需要解决这个问题。

    全局异常接入返回的标准格式

    要让全局异常接入标准格式很简单,因为全局异常处理器已经帮我们封装好了标准格式,我们只需要直接返回给客户端即可。

    1. @SneakyThrows 
    2. @Override 
    3. public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) { 
    4.   if(o instanceof String){ 
    5.     return objectMapper.writeValueAsString(ResultData.success(o)); 
    6.   } 
    7.   if(o instanceof ResultData){ 
    8.     return o; 
    9.   } 
    10.   return ResultData.success(o); 

    关键代码:

    1. if(o instanceof ResultData){ 
    2.   return o; 

    如果返回的结果是ResultData对象,直接返回即可。

    这时候我们再调用上面的错误方法,返回的结果就符合我们的要求了。

    1.   "status": 500, 
    2.   "message""自定义异常"
    3.   "data"null
    4.   "timestamp": 1625796580778 

    好了,今天的文章就到这里了,希望通过这篇文章你能掌握如何在你项目中友好实现统一标准格式到返回并且可以优雅的处理全局异常。

     

     

     

    github地址:https://github.com/jianzh5/cloud-blog/

     

    免责声明:

    ① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

    ② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

    软考中级精品资料免费领

    • 历年真题答案解析
    • 备考技巧名师总结
    • 高频考点精准押题
    • 资料下载
    • 历年真题
    • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

      难度     813人已做
      查看
    • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

      难度     354人已做
      查看
    • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

      难度     318人已做
      查看
    • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

      难度     435人已做
      查看
    • 2024年上半年系统架构设计师考试综合知识真题

      难度     224人已做
      查看

    相关文章

    发现更多好内容
    咦!没有更多了?去看看其它编程学习网 内容吧