文章详情

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

请输入下面的图形验证码

提交验证

短信预约提醒成功

优雅地统一处理接口返回值的最佳实践

2024-11-30 04:44

关注

为什么要统一接口返回值?

2. 开发流程

定义统一返回值的包装类

public class R {
  private Integer code ;
  private Object data ;
  private String message ;
  public R(Integer code, Object data, String message) {
    this.code = code ;
    this.data = data ;
    this.message = message ;
  }
  public static R success(Object data) {
    return new R(200, data, "success") ;
  }
  public static R failure(String message) {
    return new R(500, null, message) ;
  }
}

自定义ResponseBodyAdvice

@RestControllerAdvice
public class PackResponseBodyAdvice implements ResponseBodyAdvice {
  @Resource
  private ObjectMapper objectMapper ;
  @Override
  public boolean supports(MethodParameter returnType, Class> converterType) {
    // 只有返回值不是R类型的时候才通过该Advice进行处理
    return !returnType.getParameterType().equals(R.class) ;
  }
  @Override
  public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
      Class> selectedConverterType, ServerHttpRequest request,
      ServerHttpResponse response) {
    // 统一返回值处理
    return R.success(body) ;
  }


}

测试接口

@RestController
@RequestMapping("/advices")
public class AdviceController {
  @GetMapping("/str")
  public String str() {
    return "success" ;
  }
  @GetMapping("/{id}")
  public User body(@PathVariable("id") Long id) {
    return new User(id, "张三 - " + new Random().nextInt(1000)) ;
  }
}

首先,测试接口/advices/{id}

图片

处理了最终的返回结果。

继续测试/advices/str

图片

程序出错了

控制台输出

java.lang.ClassCastException: class com.pack.common.dto.R cannot be cast to class java.lang.String (com.pack.common.dto.R is in unnamed module of loader 'app'; java.lang.String is in module java.base of loader 'bootstrap')
  at org.springframework.http.converter.StringHttpMessageConverter.addDefaultHeaders(StringHttpMessageConverter.java:44) ~[spring-web-5.3.27.jar:5.3.27]
  at org.springframework.http.converter.AbstractHttpMessageConverter.write(AbstractHttpMessageConverter.java:211) ~[spring-web-5.3.27.jar:5.3.27]

出现ClassCastException错误,是由于Controller接口返回值是String,那么匹配到的HttpMessageConverter是StringHttpMessageConverter处理,而该转换器的调用是在ResponseBodyAdvice之后执行,这时候的字符串已经被转换成了R对象,所以最后在write时就出现了类型转换错误。通过如下方式处理

public Object beforeBodyWrite(Object body, 
      MethodParameter returnType, 
      MediaType selectedContentType,
      Class> selectedConverterType, ServerHttpRequest request,
      ServerHttpResponse response) {
  if (body instanceof String) {
    try {
      return this.objectMapper.writeValueAsString(R.success(body)) ;
    } catch (JsonProcessingException e) {
      e.printStackTrace();
    }
  }
  
  return R.success(body) ;
}

针对返回值是String类型的正常了。

通过自定义注解排除那些不需要处理的接口

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface NoR {
}

修改PackResponseBodyAdvice#supports方法,添加NoR注解的判断

public boolean supports(MethodParameter returnType, Class> converterType) {
  // 方法上或者是类上没有NoR注解
  return (!returnType.hasMethodAnnotation(NoR.class) 
            || AnnotatedElementUtils.hasAnnotation(returnType.getContainingClass(), NoR.class)) 
         && !returnType.getParameterType().equals(R.class)  ;
}

这样就可以控制具体哪些方法不进行处理了。

通过自定义ResponseBodyAdvice,我们可以实现接口统一返回值的处理,从而提高了接口的可读性、扩展性和错误处理能力。同时,这也有助于保持代码的清晰和规范。通过学习和实践,我们可以更好地利用SpringMVC相应的功能,开发出更优秀的应用程序。

免责声明:

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

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

软考中级精品资料免费领

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

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

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

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

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

    难度     224人已做
    查看

相关文章

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