我在Spring Boot 统一返回体中踩的坑

在Spring Boot项目中咱们能够经过RestControllerAdvice配合实现ResponseBodyAdvice<T>接口来保证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<T>的下列方法实现：

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(IgnoreRestBody.class);
}

若是某个Controller下全部的方法都绕过，就把这个注解标记在控制器类上；若是只想忽略某个方法上就把它标记在该方法上便可。

返回独立字符串的问题

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

@GetMapping("/get")
public String getStr(){
    //返回了一个字符串
    return "felord.cn";
}

咱们但愿这个字符串被统一返回体处理，相似这样：

{
    code: 200,
    data: "felord.cn",
    msg: "返回成字符串",
}

可是你会发现并无达到指望的效果，会抛出类型转换异常。这是由于当咱们的Spring MVC接口返回数据时，会根据Content-Type来选择一个HttpMessageConverter来处理，而字符串在不声明Content-Type的状况下优先使用StringHttpMessageConverter ，就致使了转换异常，须要设定成MappingJackson2HttpMessageConverter用Jackson来处理，Spring MVC的对应配置以下：

@Configuration(proxyBeanMethods = false)
public class SpringMvcConfiguration implements WebMvcConfigurer {
    @Override
    public void configureMessageConverters(List<HttpMessageConverter<?>> converters) {
        // 解决 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<HttpMessageConverter<?>> 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<T>实现中增长一个判断：

// 加强扩展性
        if (checkPrimitive(body)) {
            return RestBody.okData(Collections.singletonMap("result", body));
        }

就解决问题了。

总结

今天对Spring Boot中统一返回体的一些细节问题进行了分享，但愿可以帮助你解决一些实际开发中遇到的一样问题。多多关注：码农小胖哥 分享更多有用的编程知识。

关注公众号：Felordcn 获取更多资讯

我的博客：https://felord.cn