浏览代码

附件类加方法测试

wanghaiwu 6 天之前
父节点
当前提交
2833989daf
共有 1 个文件被更改,包括 16 次插入0 次删除
  1. 16 0
      code/base/nckd-jimin-base-helper/src/main/java/nckd/base/helper/CusFileServiceExt.java

+ 16 - 0
code/base/nckd-jimin-base-helper/src/main/java/nckd/base/helper/CusFileServiceExt.java

@@ -8,6 +8,7 @@ import kd.bos.service.attachment.FilePathService;
 
 import java.io.File;
 import java.io.InputStream;
+import java.util.Map;
 
 /**
  * 自定义附件服务
@@ -49,4 +50,19 @@ public class CusFileServiceExt extends FilePathService {
         return in;
     }
 
+    @Override
+    public InputStream beforeWriteToResponse(String originalPath, InputStream in, String userAgent) {
+        logger.info("--------------beforeWriteToResponse测试 "+originalPath+"----------------");
+
+        if("preview".equals(userAgent)){
+            in = FileSECUtils.processFileWithSEC(originalPath, in);
+        }
+
+        return super.beforeWriteToResponse(originalPath, in, userAgent);
+    }
+
+    @Override
+    public Map<String, Object> beforeWrite(String originalPath, InputStream in, String userAgent, Map<String, Object> params) {
+        return super.beforeWrite(originalPath, in, userAgent, params);
+    }
 }