From a7aca2f0ae9455d23a7c8c5c5f2a6210b6e1fb4f Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Fri, 26 Aug 2022 18:31:09 +0800 Subject: [PATCH] fix --- src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java | 12 ++++++++++-- 1 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java b/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java index 7729da2..53c9ed8 100644 --- a/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java +++ b/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java @@ -3,7 +3,9 @@ import cc.mrbird.febs.common.contants.AppContants; import cn.hutool.crypto.SecureUtil; import cn.hutool.crypto.asymmetric.KeyType; +import cn.hutool.crypto.asymmetric.RSA; import com.alibaba.fastjson.JSONObject; +import org.springframework.beans.factory.annotation.Value; import org.springframework.core.MethodParameter; import org.springframework.http.MediaType; import org.springframework.http.server.ServerHttpRequest; @@ -17,6 +19,10 @@ **/ @ControllerAdvice public class MyResponseBodyAdvise implements ResponseBodyAdvice { + + @Value("${system.debug}") + private boolean isDebug; + @Override public boolean supports(MethodParameter methodParameter, Class aClass) { return true; @@ -24,10 +30,12 @@ @Override public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) { - if (!AppContants.ENCRYPT_METHOD.contains(methodParameter.getMethod().getName())) { + if (!AppContants.ENCRYPT_METHOD.contains(methodParameter.getMethod().getName()) || isDebug) { + System.out.println(1111111); return o; } - return SecureUtil.rsa(AppContants.REQ_PRIVATE_KEY, AppContants.REQ_PUBLIC_KEY).encryptBase64(JSONObject.toJSONString(o), KeyType.PrivateKey); + RSA rsa = new RSA(AppContants.RESP_PRIVATE_KEY, AppContants.RESP_PUBLIC_KEY); + return rsa.encryptBase64(JSONObject.toJSONString(o), KeyType.PublicKey); } } -- Gitblit v1.9.1