SpringBoot2.3 修改响应头、添加更新token、解决在过滤器中修改失败

前言

使用Spring Boot开发项目的过程中，难免会遇到向响应头中写入一些信息，例如更新了的token，用户详情等信息。在请求头中添加，请求时就可以通过过滤器，拦截器等验证或者获得请求头传递的信息。但是如果使用@RestController注解，在过滤器、拦截器中修改响应头是无效的。那么怎么解决呢？

解决方式

话不多说，直接上代码，以在响应头中添加token为例

@ControllerAdvice
public class HeaderModifierAdvice implements ResponseBodyAdvice<Object> {

    @Override
    public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object o,
                                  MethodParameter methodParameter,
                                  MediaType mediaType,
                                  Class<? extends HttpMessageConverter<?>> aClass,
                                  ServerHttpRequest serverHttpRequest,
                                  ServerHttpResponse serverHttpResponse) {
        // 转换对象
        HttpServletRequest request = ((ServletServerHttpRequest) serverHttpRequest).getServletRequest();
        HttpServletResponse response = ((ServletServerHttpResponse) serverHttpResponse).getServletResponse();
        // 获得token
        String token = JwtTokenUtils.refresh(request);
        if (StringUtils.isNotBlank(token)) {
            /*刷新token*/
            response.addHeader(JwtTokenUtils.TOKEN_KEY, token);
        }
        return o;
    }
}

1、新建一个类，实现ResponseBodyAdvice接口，并且实现beforeBodyWrite方法。
2、在beforeBodyWrite方法中写入自己的添加逻辑。
注意 beforeBodyWrite方法中的HttpServletRequest和HttpServletResponse需要从对应的对象获得。其次supports方法返回值为true即可。