如果你正在寻找一种方法来规范化你的RESTful API的返回值,那么这篇文章将是你的理想选择。通过阅读这篇文章,你将了解到如何使用Spring ResponseBodyAdvice来改善你的API设计和实现。
为什么要统一接口返回值?
定义统一返回值的包装类
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
@RestControllerAdvicepublic class PackResponseBodyAdvice implements ResponseBodyAdvice<Object> { @Resource private ObjectMapper objectMapper ; @Override public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) { // 只有返回值不是R类型的时候才通过该Advice进行处理 return !returnType.getParameterType().equals(R.class) ; } @Override public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> 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<? extends HttpMessageConverter<?>> 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<? extends HttpMessageConverter<?>> converterType) { // 方法上或者是类上没有NoR注解 return (!returnType.hasMethodAnnotation(NoR.class) || AnnotatedElementUtils.hasAnnotation(returnType.getContainingClass(), NoR.class)) && !returnType.getParameterType().equals(R.class) ;}
这样就可以控制具体哪些方法不进行处理了。
通过自定义ResponseBodyAdvice,我们可以实现接口统一返回值的处理,从而提高了接口的可读性、扩展性和错误处理能力。同时,这也有助于保持代码的清晰和规范。通过学习和实践,我们可以更好地利用SpringMVC相应的功能,开发出更优秀的应用程序。
本文链接:http://www.28at.com/showinfo-26-38112-0.html优雅地统一处理接口返回值的最佳实践
声明:本网页内容旨在传播知识,若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。邮件:2376512515@qq.com
上一篇: 一次 K8s 升级,竟然导致滴滴故障 12 小时?
下一篇: Python中的单例模式,这些你都会吗?