wangjun 3 долоо хоног өмнө
parent
commit
3170e24f5c

+ 21 - 12
code/base/nckd-jimin-base-helper/src/main/java/nckd/base/helper/CusFileServiceExt.java

@@ -17,26 +17,35 @@ public class CusFileServiceExt extends FilePathService implements FileServiceExt
     public InputStream encode(String originalPath, InputStream in) {
     public InputStream encode(String originalPath, InputStream in) {
         logger.info("--------------SEC 附件加密 "+originalPath+"----------------");
         logger.info("--------------SEC 附件加密 "+originalPath+"----------------");
 
 
-        long filesize = FileSECUtils.getFileSizeByPath(originalPath);
-        InputStream inForSEC = FileSECUtils.decodeFileForSEC(filesize,in);
-        if(inForSEC == null){
-            return in;
+        InputStream inForSEC = null;
+        int isEncryption = FileSECUtils.checkFileIsEncryptionRest(in);
+        if(isEncryption != 1){
+
+            long filesize = FileSECUtils.getFileSizeByPath(originalPath);
+            inForSEC = FileSECUtils.decodeFileForSEC(filesize,in);
+            if(inForSEC != null){
+                return inForSEC;
+            }
+        }else{
+            logger.info("--------------SEC 附件已加密 "+originalPath+"----------------");
         }
         }
-        return inForSEC;
+        return in;
     }
     }
 
 
     @Override
     @Override
     public InputStream decode(String originalPath, InputStream in) {
     public InputStream decode(String originalPath, InputStream in) {
         logger.info("--------------SEC 附件解密 "+originalPath+"----------------");
         logger.info("--------------SEC 附件解密 "+originalPath+"----------------");
 
 
-        long filesize = FileSECUtils.getFileSizeByPath(originalPath);
-        InputStream inForSEC = FileSECUtils.decodeFileForSEC(filesize,in);
-
-        if(inForSEC == null){
-            return in;
+        InputStream inForSEC = null;
+        int isEncryption = FileSECUtils.checkFileIsEncryptionRest(in);
+        if(isEncryption == 1) {
+            long filesize = FileSECUtils.getFileSizeByPath(originalPath);
+            inForSEC = FileSECUtils.decodeFileForSEC(filesize, in);
+            if (inForSEC != null) {
+                return inForSEC;
+            }
         }
         }
-
-        return inForSEC;
+        return in;
     }
     }