ZQ/laws-sinotruk-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

文件下载加解密

Browse Source
This commit is contained in:
梁琦涛
2023-11-27 09:40:15 +08:00
parent 86e0040303
commit 021426ba4a
2 changed files with 48 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
laws-modules-docking/src/main/java/com/jero/modules/docking/download/config/DownloadInterceptor.java
-7
View File
@@ -3,8 +3,6 @@ package com.jero.modules.docking.download.config;
import com.jero.modules.docking.download.service.DownloadDecryptFileService;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;
@@ -34,11 +32,6 @@ public class DownloadInterceptor implements HandlerInterceptor{
*/
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
// HandlerMethod handlerMethod = (HandlerMethod) handler;
// Method method = handlerMethod.getMethod();
// String methodName = method.getName();
// System.out.println("====拦截到了方法："+methodName+"，preHandle====");
// other
return true;
}
laws-modules-docking/src/main/java/com/jero/modules/docking/download/service/impl/DownloadDecryptFileServiceImpl.java
+48 -48
View File
@@ -8,14 +8,18 @@ import com.jero.modules.docking.download.service.DownloadDecryptFileService;
import com.jero.modules.docking.utils.IntekeyUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.entity.ContentType;
import org.apache.shiro.SecurityUtils;
import org.jetbrains.annotations.NotNull;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.stereotype.Service;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.util.ContentCachingResponseWrapper;
import org.springframework.web.util.WebUtils;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@@ -84,8 +88,8 @@ public class DownloadDecryptFileServiceImpl implements DownloadDecryptFileServic
@Override
public void downloadDecryptFile(HttpServletRequest request, HttpServletResponse response, Object handler) {
// 判断返回值是否为文件流，否则其他接口返回值会变更成字符串，导致前端无法解析
String responseBody = null;
HandlerMethod handlerMethod = (HandlerMethod) handler;
String responseBody = null;
try {
String returnType = handlerMethod.getMethod().getReturnType().getName();
if(!Objects.equals(returnType,"void") && !Objects.equals(returnType,"org.springframework.web.servlet.ModelAndView")){
@@ -93,71 +97,67 @@ public class DownloadDecryptFileServiceImpl implements DownloadDecryptFileServic
}
responseBody = ContentCachingWrapperFilter.getResponseBody(response);
if(StringUtils.isBlank(responseBody)){
return;
throw new JeroBootException(ResultCommon.ERROR);
}
} catch (IOException e) {
e.printStackTrace();
throw new JeroBootException(ResultCommon.ERROR);
}
MultipartFile file = getMultipartFile(responseBody);
// 访问路径
String str = request.getRequestURI();
boolean b = isBoolean(str, viewUrl);
boolean b1 = isBoolean(str, downloadUrl);
if(b){
// 预览，进行解密
writeFile(response, file, null);
}else if(b1){
// 下载文件 先解密，判断组织域，加密
writeFile(response, file, null);
int scope = getScope();
// 加密
writeFile(response, file, scope);
}else{
// excel 判断组织域，加密
int scope = getScope();
// 加密
writeFile(response, file, scope);
}
}
private void writeFile(HttpServletResponse response, MultipartFile file, Integer scope) {
try (InputStream is = IntekeyUtils.DecryptFile(encryptUrl, encryptAppCode, encryptSecretKey, scope, file)) {
writeResponse(response, is);
} catch (Exception e) {
throw new JeroBootException(ResultCommon.ERROR);
}
}
@NotNull
private MultipartFile getMultipartFile(String responseBody) {
MultipartFile file;
try {
InputStream inputStream = new ByteArrayInputStream(responseBody.getBytes(StandardCharsets.UTF_8));
try (InputStream inputStream = new ByteArrayInputStream(responseBody.getBytes(StandardCharsets.UTF_8))){
file = new MockMultipartFile(ContentType.APPLICATION_OCTET_STREAM.toString(), inputStream);
} catch (IOException e) {
e.printStackTrace();
throw new JeroBootException(ResultCommon.ERROR);
}
// 访问路径
String str = request.getRequestURI();
return file;
}
private boolean isBoolean(String str, List<String> viewUrl) {
boolean b = false;
for (String s : viewUrl) {
if(s.contains(str)){
if (s.contains(str)) {
b = true;
break;
}
}
boolean b1 = false;
for (String s : downloadUrl) {
if(s.contains(str)){
b1 = true;
break;
}
}
if(b){
// 预览，进行解密
try (InputStream is = IntekeyUtils.DecryptFile(decryptUrl, decryptAppCode, decryptSecretKey, null, file)){
writeResponse(response,is);
}catch (Exception e){
throw new JeroBootException(ResultCommon.ERROR);
}
}else if(b1){
// 下载文件 先解密，判断组织域，加密
try (InputStream is = IntekeyUtils.DecryptFile(decryptUrl, decryptAppCode, decryptSecretKey, null, file)){
writeResponse(response,is);
}catch (Exception e){
throw new JeroBootException(ResultCommon.ERROR);
}
int scope = getScope();
// 加密
try (InputStream is = IntekeyUtils.DecryptFile(encryptUrl, encryptAppCode, encryptSecretKey, scope, file)){
writeResponse(response,is);
}catch (Exception e){
throw new JeroBootException(ResultCommon.ERROR);
}
return;
}else{
// excel 判断组织域，加密
int scope = getScope();
// 加密
try (InputStream is = IntekeyUtils.DecryptFile(encryptUrl, encryptAppCode, encryptSecretKey, scope, file)){
writeResponse(response,is);
}catch (Exception e){
throw new JeroBootException(ResultCommon.ERROR);
}
}
return b;
}
private int getScope() {