角色

hx-file/src/main/java/com/fjhx/file/utils/ObsFileUtil.java

@@ -1,10 +1,13 @@
 package com.fjhx.file.utils;
 
 import cn.hutool.extra.spring.SpringUtil;
+import com.fjhx.file.entity.FileInfoSelectDto;
+import com.fjhx.file.entity.FileInfoVo;
 import com.fjhx.file.entity.ObsFile;
 import com.fjhx.file.service.FileInfoService;
 
 import java.util.List;
+import java.util.Map;
 
 public class ObsFileUtil {
 
@@ -72,4 +75,17 @@ public class ObsFileUtil {
         fileInfoService.removeFile(businessId);
     }
 
+    public static Map<Long, List<FileInfoVo>> getFileMap(List<Long> businessIdList) {
+        FileInfoSelectDto dto = new FileInfoSelectDto();
+        dto.setBusinessIdList(businessIdList);
+        return fileInfoService.getList(dto);
+    }
+
+    public static Map<Long, List<FileInfoVo>> getFileMap(List<Long> businessIdList, Integer fileType) {
+        FileInfoSelectDto dto = new FileInfoSelectDto();
+        dto.setBusinessIdList(businessIdList);
+        dto.setFileType(fileType);
+        return fileInfoService.getList(dto);
+    }
+
 }