Просмотр исходного кода

10/27 xx "GUID改为小写guid,查询文件信息添加后缀信息"

xiexiang 1 год назад
Родитель
Сommit
9708c5a0f1

+ 16 - 2
src/main/java/com/example/fms/controller/FileMangerController.java

@@ -26,6 +26,7 @@ import java.nio.charset.StandardCharsets;
 import java.util.List;
 
 /**
+ * 文件系统管理接口
  * @Author xiexiang
  * @Date 2023/6/6
  */
@@ -76,7 +77,7 @@ public class FileMangerController {
         //根据文件GUID获取文件信息
         byte[] fileData = fileManagerService.downloadFile(fileId);
         LambdaQueryWrapper<SystemFile> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(SystemFile::getGUID, fileId);
+        queryWrapper.eq(SystemFile::getGuid, fileId);
         SystemFile systemFile = systemFileService.getOne(queryWrapper);
         String fileName = systemFile.getOriginalName();
         try {
@@ -97,8 +98,21 @@ public class FileMangerController {
     @Operation(summary = "获取文件信息")
     public List<SystemFile> getFileData(@RequestBody List<String> fileIds) {
         LambdaQueryWrapper<SystemFile> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.in(SystemFile::getGUID, fileIds);
+        queryWrapper.in(SystemFile::getGuid, fileIds);
         List<SystemFile> systemFiles = systemFileService.list(queryWrapper);
+        systemFiles.forEach(systemFile -> {
+            //定义一个空字符串装载文件名后缀
+            String extension = "";
+            //文件名
+            String name = systemFile.getFileName();
+            //获取最后一个点的位置
+            int dotIndex = name.lastIndexOf(".");
+            if (dotIndex > 0 && dotIndex < name.length() - 1) {
+                //获取文件名后缀
+                extension = name.substring(dotIndex + 1);
+            }
+            systemFile.setType(extension);
+        });
         return systemFiles;
     }
 

+ 4 - 1
src/main/java/com/example/fms/domain/SystemFile.java

@@ -21,7 +21,7 @@ public class SystemFile extends BaseEntity<SystemFile> {
      * 唯一标识id
      */
     @TableField("GUID")
-    private String GUID;
+    private String guid;
 
     /**
      * 存储位置
@@ -83,4 +83,7 @@ public class SystemFile extends BaseEntity<SystemFile> {
     @TableField("IS_DELETE")
     private Integer isDelete;
 
+    @TableField(exist = false)
+    private String type;
+
 }

+ 3 - 3
src/main/java/com/example/fms/service/FileMangerService.java

@@ -77,7 +77,7 @@ public class FileMangerService {
             DownloadSysFileDTO downloadSysFileDTO = new DownloadSysFileDTO();
             //2.根据传入文件id查询到文件信息
             LambdaQueryWrapper<SystemFile> queryWrapper = new LambdaQueryWrapper<>();
-            queryWrapper.eq(SystemFile::getGUID, fileId);
+            queryWrapper.eq(SystemFile::getGuid, fileId);
             SystemFile systemFileVO = systemFileService.getOne(queryWrapper);
             //3.文件信息标记为已经删除的,返回null
             if (systemFileVO.getIsDelete().equals(1)) {
@@ -128,7 +128,7 @@ public class FileMangerService {
             for (int i = 0; i < guIds.size(); i++) {
                 //3.1.2 获取需要更新的对象
                 LambdaQueryWrapper<SystemFile> queryWrapper = new LambdaQueryWrapper<>();
-                queryWrapper.eq(SystemFile::getGUID, guIds.get(i));
+                queryWrapper.eq(SystemFile::getGuid, guIds.get(i));
                 SystemFile systemFileVO = systemFileService.getOne(queryWrapper);
                 //3.1.3 将对象的是否删除字段置为1
                 systemFileVO.setIsDelete(1);
@@ -151,7 +151,7 @@ public class FileMangerService {
             for (int i = 0; i < guIds.size(); i++) {
                 //3.2.2 根据GUID到表中查询该文件记录的数据
                 LambdaQueryWrapper<SystemFile> queryWrapper = new LambdaQueryWrapper<>();
-                queryWrapper.eq(SystemFile::getGUID, guIds.get(i));
+                queryWrapper.eq(SystemFile::getGuid, guIds.get(i));
                 SystemFile systemFileVO = systemFileService.getOne(queryWrapper);
                 //3.2.3 根据pType判断存储在服务器的哪个地址下
                 int pType = systemFileVO.getPType();

+ 2 - 2
src/main/java/com/example/fms/service/SystemFileService.java

@@ -50,12 +50,12 @@ public class SystemFileService extends ServiceImpl<SystemFileMapper, SystemFile>
 //                //给实体类赋值登陆人id
 //                systemFile.setCreateId(personnelVO.getId());
                 String guid = this.giveFileGUID(systemFile.getOriginalName());
-                systemFile.setGUID(guid);
+                systemFile.setGuid(guid);
                 systemFile.setCreateId(1);
                 systemFile.setPType(sourceId);
                 //插入数据
                 systemFile.insert();
-                GUIDs.add(systemFile.getGUID());
+                GUIDs.add(systemFile.getGuid());
             }
             return GUIDs;
         } else {