Spring Boot统一返回体的踩坑记录

Spring Boot统一返回体的踩坑记录

前言

在Spring Boot项目中我们可以通过RestControllerAdvice配合实现ResponseBodyAdvice接口来保证Spring MVC接口具有统一的返回格式，以保证前端同学能够封装统一的数据接收工具。但是很多网上的文章并没有对实际开发中的细节作出更多的讲解。今天胖哥就来分享一下我的采坑经历，也算作一个总结。

控制作用范围

我记得在前面关于Swagger3的文章中提过，如果我们不指定范围将导致Swagger无法识别接口的元信息。因此如果你使用了Swagger必须指定其范围，这里你可以通过指定扫描包来指定其作用域：

@RestControllerAdvice("cn.felord.controller")

如果你的Spring MVC控制器有统一的父类控制器的话，

@RestController

@RequestMapping("/foo")

public class FooController extends BaseController {

//todo 省略

}

也可以这样：

@RestControllerAdvice(assignableTypes = BaseController.class)

白名单

有些接口可能根据业务需要或者协议需要不能使用统一返回体，例如支付的通知应答。这就需要一个类似白名单的机制来绕过统一返回体控制器通知类。我们可以借助于ResponseBodyAdvice的下列方法实现：

boolean supports(MethodParameter returnType, Class extends HttpMessageConverter>> converterType);

这个方法如果返回false就表示不执行统一返回体的封装逻辑。这里我推荐注解实现。定义一个标记注解，可以定义在类上或者方法上：

@Documented

@Inherited

@Target({ElementType.ANNOTATION_TYPE, ElementType.METHOD})

@Retention(RetentionPolicy.RUNTIME)

public @interface IgnoreRestBody {

}

然后上面的supports方法这样实现:

@Override

public boolean supports(MethodParameter returnType, Class extends HttpMessageConverter>> converterType) {

return !returnType.hasMethodAnnotation(http://IgnoreRestBody.class);

}

如果某个Controller下所有的方法都绕过，就把这个注解标记在控制器类上；如果只想忽略某个方法上就把它标记在该方法上即可。

返回独立字符串的问题

有些接口我们会返回一个字符串:

@GetMapping("/get")

public String getStr(){

//返回了一个字符串

return "felord.cn";

}

我们希望这个字符串被统一返回体处理，类似这样：

{

code: 200,

data: "felord.cn",

msg: "返回成字符串",

}

但是你会发现并没有达到期望的效果，会抛出类型转换异常。这是因为当我们的Sprinhttp://g MVC接口返回数据时，会根据Content-Type来选择一个HttpMessageConverter来处理，而字符串在不声明Content-Type的情况下优先使用StringHttpMessageConverter ，就导致了转换异常，需要设定成MappingJackson2HttpMessageConverter用Jackson来处理，Spring MVC的对应配置如下：

@Configuration(proxyBeanMethods = false)

public class SpringMvcConfiguration implements WebMvcConfigurer {

@Override

public void configureMessageConverters(List> convertehttp://rs) {

// 解决 String 统一封装RestBody的问题

converters.add(0, new MappingJackson2HttpMessageConverter());

}

}

嗯，这样就起效了！你以为这样就完了？你会发现你的jsON序列化不按照你设置的策略执行了。因为你new了一个而不是采用系统初始化的那个。解决方法为，将Spring IoC中的ObjectMapper注入到MappingJackson2HttpMessageConverter中去。或者你使用Debug调试出系统默认的MappingJackson2HttpMessageConverter的位置，比如我的索引为7，就可以这样配置：

@Configuration(proxyBeanMethods = false)

public class SpringMvcConfiguration implements WebMvcConfigurer {

@Override

public void configureMessageConverters(List> converters) {

// 解决 String 统一封装RestBody的问题

HttpMessageConverter> httpMessageConverter = converters.get(7);

if (!(httpMessageConverter instanceof MappingJackson2HttpMessageConverter)) {

// 确保正确，如果有改动就重新debug

throw new RuntimeException("MappingJackson2HttpMessageConverter is not here");

}

converters.add(0, httpMessageConverter);

}

}

Data的类型问题

曾经一个安卓开发同学说，你这统一结构中的data如果是数组：

{

code: 200,

data: ['a','b'],

msg: "返回成字符串",

}

后续如果data添加其它与数组没有关系的属性就不兼容了，你应该保证这个data是个Map。是的，这也是问题，实际中发现不仅仅是数组，如果是int、long等原始类型或者String类型都面临这种情况，需要加一个额外的判断body是不是可能改变data类型的类型：

private boolean checkPrimitive(Object body) {

Class> clazz = body.getClass();

return clazz.isPrimitive()

|| clazz.isArray()

|| Collection.class.isAssignableFrom(clazz)

|| body instanceof Number

|| body instanceof Boolean

|| body instanceof Character

|| body instanceof String;

}

然后我们在ResponseBodyAdvice实现中增加一个判断：

// 增强扩展性

if (checkPrimitive(body)) {

return RestBody.okData(Collections.singletonMap("result", body));

}

就解决问题了。

总结

今天对Spring Boot中统一返回体的一些细节问题进行了分享，希望能够帮助你解决一些实际开发中遇到的同样问题。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。