快速上手

一般我们在写服务的时候，会有统一的返回格式，在这篇文章中用RespMsg表示。

比如根据用户的id要获取用户的信息。

第一版代码如下

@PostMapping(value = "user/{id}")public RespMsg getUserById(@PathVariable Long id) {    try {        return RespMsg.success(userService.getUserById(id));    } catch (BusinessException e) {        log.error(e.getMessage(), e);        return RespMsg.failed(e.getErrCode(), e.getMessage());    }}

看了上一篇你真的了解spring boot全局异常处理吗，就可以改写为第二版

@PostMapping(value = "user/{id}")public RespMsg getUserById(@PathVariable Long id) throws BusinessException {    return RespMsg.success(userService.getUserById(id));}

看了这篇，就可以改写为第三版

@PostMapping(value = "user/{id}")public User getUserById(@PathVariable Long id) throws BusinessException {    return userService.getUserById(id);}

实现方式也很简单，通过实现ResponseBodyAdvice搭配@ControllerAdvice即可

@ControllerAdvicepublic class ResponseWrapperAdvice implements ResponseBodyAdvice {    @Override    public boolean supports(MethodParameter returnType, Class converterType) {        return true;    }    @Override    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {        //如果已经是RespMsg类型的则直接返回        if (body instanceof RespMsg) {            return body;        }        //如果不是，则封装        return RespMsg.success(body);    }}
  • supports方法用于判断是否支持，如果不支持则不会调用beforeBodyWrite。具体可见RequestResponseBodyAdviceChain的processBody方法。既然是统一，这里都返回true了。
  • beforeBodyWrite方法用于改写响应。

然后就大功告成了，so easy。

问题分析

上面的其实有一点小坑，下面详细的分析。

匹配到错误的HttpMessageConverter

如果接口返回的是String，比如

@PostMapping(value = "test")public String test() {    return "test";}

就会报错

从我们定义的RespMsg强转为String报错，从异常信息信息可以看出报错发生在StringHttpMessageConverter的getContentLength方法，跟了代码发现是在AbstractHttpMessageConverter的addDefaultHeaders中，调用了getContentLength方法，而这个方法被StringHttpMessageConverter方法重写了，在AbstractHttpMessageConverter的getContentLength声明如下：

protected Long getContentLength(T t, @Nullable MediaType contentType) throws IOException

而StringHttpMessageConverter中重写时的声明变为了：

protected Long getContentLength(String str, @Nullable MediaType contentType)

注意第一个类型变量发生了变化，报错也是因为这里有个隐式的类型转换，跟代码时这里第一个变量其实已经被封装为RespMsg，强转String肯定会报错。

这个顺序在ResponseBodyAdvice的注释中有提到

Allows customizing the response after the execution of an @ResponseBody or a ResponseEntity controller method but before the body is written with an HttpMessageConverter.

允许在被 @ResponseBody注解或者返回ResponseEntity 的Controller方法执行后自定义响应，但是是在HttpMessageConverter写之前。

具体代码可以参考AbstractMessageConverterMethodProcessor的writeWithMessageConverters方法。

等等，为什么是StringHttpMessageConverter，而不是MappingJackson2HttpMessageConverter

还是AbstractMessageConverterMethodProcessor的writeWithMessageConverters方法中有如下代码：

会按messageConverters顺序检测是否canWrite，至于为什么StringHttpMessageConverter能匹配上，因为Controller中的方法返回的是String类型。

首先想到的解决办法是改变一下messageConverters里面元素的顺序，找到了加载messageConverters代码，具体可见RequestMappingHandlerAdapter的构造方法。

public RequestMappingHandlerAdapter() {    StringHttpMessageConverter stringHttpMessageConverter = new StringHttpMessageConverter();    stringHttpMessageConverter.setWriteAcceptCharset(false);  // see SPR-7316    this.messageConverters = new ArrayList<>(4);    this.messageConverters.add(new ByteArrayHttpMessageConverter());    this.messageConverters.add(stringHttpMessageConverter);    try {        this.messageConverters.add(new SourceHttpMessageConverter<>());    }    catch (Error err) {        // Ignore when no TransformerFactory implementation is available    }    this.messageConverters.add(new AllEncompassingFormHttpMessageConverter());}

可以看到StringHttpMessageConverter是第二个元素。

想改变messageConverters的顺序，可以通过实现WebMvcConfigurer的configureMessageConverters方法，调整list中的元素位置。我试了一下，调整完之后不会报错了，但是会不会引起别的问题还不确定，毕竟这是一个较大的改动。

第二种方案就是在Controller层中包装为RespMsg，这样就避免了匹配到StringHttpMessageConverter。这种方案可能更加安全。

ResponseBodyAdvice接口的泛型处理

注意到ResponseBodyAdvice是支持泛型的，刚才为了统一返回，所以写了Object，但是如果你只想用于某些类型，就可以直接指定类型，比如

@ControllerAdvicepublic class UserWrapperAdvice implements ResponseBodyAdvice<User> {    @Override    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {        return false;    }    @Override    public User beforeBodyWrite(User body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {        return null;    }}

这时需要注意的是一定要在supports方法中判断什么类型支持，什么类型不支持，spring并不会根据你指定的类型判断调不调用beforeBodyWrite方法，只会根据supports的返回判断，比如说我返回了一个RespMsg，而supports返回true，然后就会调用beforeBodyWrite，这时会发生强制转换（RespMsg->User），就会报错了。

总结

  1. 可以通过实现ResponseBodyAdvice搭配@ControllerAdvice来统一数据的返回格式。（配合统一异常处理）
  2. 根据方法的返回类型，可能导致匹配到错误的HttpMessageConverter，这时可以通过修改返回类型避免这种错误。
  3. ResponseBodyAdvice接口支持泛型，需要严格的判断类型是否符合，否则可能会因为强转不成功报错。

看到了这里一定是真爱了，关注微信公众号【憨憨的春天】第一时间获取更新