tieshengkeji/tiesheng-springboot
1
1
Fork 0
Code Pull Requests Actions Packages Releases Activity

perf： 加解密数据处理

Browse Source
This commit is contained in:
曾文豪
2023-03-07 15:16:26 +08:00
parent c0fb149558
commit 0094bf2e69
8 changed files with 75 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
springboot-encrypt/src/main/java/com/tiesheng/encrypt/config/EncryptResponseBodyAdvice.java
View File

@@ -1,11 +1,10 @@
package com.tiesheng.encrypt.config;
import cn.hutool.core.annotation.AnnotationUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import cn.hutool.log.LogFactory;
import com.tiesheng.util.CommonUtil;
import com.tiesheng.util.config.EncryptConfig;
import com.tiesheng.util.pojos.ApiResp;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
@@ -21,7 +20,7 @@ import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
* @author hao
*/
@ControllerAdvice
public class EncryptResponseBodyAdvice implements ResponseBodyAdvice<Object> {
public class EncryptResponseBodyAdvice implements ResponseBodyAdvice<ApiResp> {
@Autowired
EncryptConfig encryptConfig;
@@ -32,25 +31,19 @@ public class EncryptResponseBodyAdvice implements ResponseBodyAdvice<Object> {
}
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends
public ApiResp beforeBodyWrite(ApiResp body, MethodParameter returnType, MediaType selectedContentType, Class<? extends
HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
try {
String content = JSONUtil.toJsonStr(body);
String respData = JSONUtil.parseObj(content).getStr("data");
if (StrUtil.isEmpty(respData)) {
// 无需加密
Object data = body.getData();
if (data == null || !body.successful()) {
return body;
}
JSONObject resp = JSONUtil.parseObj(content);
resp.set("encrypt", true);
if (resp.getInt("code") == 200) {
resp.set("data", encryptConfig.encrypt(respData));
}
return resp;
body.setEncrypt(true);
body.setData(encryptConfig.encrypt(CommonUtil.writeJsonString(data)));
return body;
} catch (Exception var17) {
LogFactory.get().info("加密数据异常", var17);
}