RequestBodyAdvice 和 ResponseBodyAdvice增强器使用

前言:
  日常开发中,我们常常需要对@RequestBody的参数进行各种处理,例如加解密、打印日志,这些东西我们可以用到RequestBodyAdvice 和 ResponseBodyAdvice来对请求前后进行处理,本质上他俩都是AOP,这里做加解密处理记录。

1 RequestBodyAdvice

@RestControllerAdvice
public class CustomRequestControllerAdvice implements RequestBodyAdvice {
}

RequestBodyAdvice接口中提供了四个实现方法:
在这里插入图片描述
  (1)supports:返回true代表开启支持,false不支持。
  (2)beforeBodyRead:body读取前进行处理。
  (3)afterBodyRead:body读取后进行处理。
  (4)handleEmptyBody:body为空时处理。
这里使用beforeBodyRead进行解密处理演示:
  定义一个内部类,通过构造参数进行处理

/**
     * @author:lis
     * @createTime:2019-11-08 10:42
     * @description:内部类处理header和body数据解密
     * @modifyTime:
     * @modifyDescription:
     */
    class CustomHttpInputMessage implements HttpInputMessage {
        private HttpHeaders headers;

        private InputStream body;

        public CustomHttpInputMessage(HttpInputMessage inputMessage) throws Exception {

            //TODO 此处header解密处理
            headers = inputMessage.getHeaders();
            /**
             * 这里有个严重声明!!!
             * IOUtils.toString(inputMessage.getBody(), "UTF-8")取完数据后一定需要缓存!!
             * 写法如下:
             */
            String bodyStr = IOUtils.toString(inputMessage.getBody(), "UTF-8");
            //TODO 此处body解密处理
            String data = AESUtil.decrypt(bodyStr,environment.getProperty("dataKey"),environment.getProperty("dataOffset"));
            body = IOUtils.toInputStream(data, "UTF-8");
        }

        public InputStream getBody() {
            return this.body;
        }

        public HttpHeaders getHeaders() {
            return this.headers;
        }
    }

  注意:上述代码中的声明,因为requestBody的数据是通过流来获取的,所以在使用HttpInputMessage获取数据后,一定要记录下来,放回去记录的,因为该数据只能取一次,第二次获取将返回NULL!

2 ResponseBodyAdvice

ResponseBodyAdvice接口提供了两种实现方法
在这里插入图片描述
  (1)supports:返回true代表开启支持,false不支持。
  (2)beforeBodyWrite:body返回给页面参数之前处理。
这里的beforeBodyWrite可用于返回数据加密处理,业务代码不做演示了。

  • 5
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值