java拦截接口返回参数ResponseBodyAdvice

@ControllerAdvice
public class MyEncryptResponseBodyAdvice implements ResponseBodyAdvice<Object> {
    private boolean encrypt;//业务参数
    @Autowired
    private SecretKeyConfig secretKeyConfig;//业务参数
    private static ThreadLocal<Boolean> encryptLocal = new ThreadLocal();//业务参数

    public MyEncryptResponseBodyAdvice() {
    }
	//返回true时才会执行beforeBodyWrite方法
    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
        this.encrypt = false;
        if (returnType.getMethod().isAnnotationPresent(MyEncrypt.class) && !this.secretKeyConfig.isDebug()) {
            this.encrypt = true;
        }

        return this.encrypt;
    }

    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
        Boolean status = (Boolean)encryptLocal.get();
        if (null != status && !status) {
            encryptLocal.remove();
            return body;
        } else {
            if (this.encrypt) {
                String publicKey = this.secretKeyConfig.getPublicKey();

                try {
                    String content = JSON.toJSONString(body);
                    if (!StringUtils.hasText(publicKey)) {
                        throw new NullPointerException("Please configure rsa.encrypt.privatekeyc parameter!");
                    }

                    String encodedData = AesUtil.aesEncrypt(content, publicKey);
                    //String result = new String(Base64.encodeBase64(encodedData));
                    return encodedData;
                } catch (Exception var13) {
                    var13.printStackTrace();
                }
            }

            return body;
        }
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

IT界的老菜鸟

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值