spring-boot中统一修改http响应体内容的方法

作者: 向上生长之路 | 来源:发表于2021-09-25 22:43 被阅读0次

spring-boot中统一修改http响应体内容的方法
JavaEE中统一修改http响应体内容的方法
HTTP协议入门教程，一文就够了！
网络教程
Web API02-HTTP协议基础
HTTP协议笔记
HTTP请求行、请求头、请求体详解
HTTP协议
spring-boot启动图标更改
关于http请求的一些问题

在spring-boot开发中，有时候会遇到一些统一修改http接口响应体内容的场景，比如统一增加一些响应头信息、统一增加一些数据包装处理、数据转换等场景，spring框架中已经为我们提供了这种能力，即ResponseBodyAdvice切面的使用，ResponseBodyAdvice就是spring框架中预留的钩子，它作用在Controller方法执行完成之后，http响应体写回客户端之前，这个时候我们就能方便的织入一些自己的业务逻辑处理了

工作原理

查看ResponseBodyAdvice接口的源码中，提供了两个方法

supports方法用于判断beforeBodyWrite方法的执行与否，返回值为布尔类型，返回true即执行beforeBodyWrite方法；可以在实现类中编写实现逻辑，来根据MethodParameter以及HttpMessageConverter类型判断是否需要改写http响应体
beforeBodyWrite方法提供了用于修改http响应体的能力，我们要编写的核心逻辑就是这个方法所在了，该方法中提供了Controller方法的返回值，ServerHttpRequest 、ServerHttpResponse 对象等，在实现类方法中，我们拿到这些对象，根据需要来编写修改逻辑

public interface ResponseBodyAdvice<T> {
    boolean supports(MethodParameter var1, Class<? extends HttpMessageConverter<?>> var2);

    @Nullable
    T beforeBodyWrite(@Nullable T var1, MethodParameter var2, MediaType var3, Class<? extends HttpMessageConverter<?>> var4, ServerHttpRequest var5, ServerHttpResponse var6);
}

使用方式

自定义一个实现ResponseBodyAdvice接口的实现类，添加注解@ControllerAdvice，来拦截Controller控制器的执行

示例代码：https://github.com/netbuffer/spring-boot-demo/blob/master/src/main/java/cn/netbuffer/springboot/demo/controller/CustomResponseBodyAdvice.java

@Slf4j
@ControllerAdvice
public class CustomResponseBodyAdvice implements ResponseBodyAdvice {

    @Override
    public boolean supports(MethodParameter methodParameter, Class aClass) {
        //根据需要来判断是否允许执行beforeBodyWrite
        Class targetClass = methodParameter.getMethod().getDeclaringClass();
        log.debug("supports execute methodParameter={} targetClass={} class={}", methodParameter, targetClass, aClass);
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object data, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
        //data即为Controller方法执行后写回的值
        log.debug("beforeBodyWrite data={}", data);
        //这里可以拿到request信息来做特定的逻辑处理
        if (serverHttpRequest.getURI().getPath().equals("/ua")) {
            //测试添加http响应头
            serverHttpResponse.getHeaders().set("x-abc", "header-value");
        }
        //测试根据不同数据类型，来做不同的包装处理、转换等操作
        if (data instanceof String) {
            return "[" + data + "]";
        } else if (data instanceof Map) {
            ((Map) data).put("extData", "value");
            return data;
        } else {
            return data;
        }
    }

}

建立测试用的控制器UserAgentController，用于返回User-Agent头信息
示例代码：https://github.com/netbuffer/spring-boot-demo/blob/master/src/main/java/cn/netbuffer/springboot/demo/controller/UserAgentController.java

@RestController
@RequestMapping("/ua")
public class UserAgentController {

    @GetMapping
    public String ua(@RequestHeader("User-Agent") String userAgent) {
        return userAgent;
    }
}

建立测试用于返回Map类型数据的映射方法
完整示例代码：https://github.com/netbuffer/spring-boot-demo/blob/master/src/main/java/cn/netbuffer/springboot/demo/controller/BeanController.java

    @GetMapping(value = {"map"})
    public Map getMap() {
        Map data = new HashMap();
        data.put("name", "test");
        data.put("age", "30");
        return data;
    }