完成文件对象管理刷新状态功能

2025-09-01 22:53:29 +08:00 · 2022-04-16 23:47:22 +08:00
parent 3b6c6aa3e7
commit 5c895d0ef5
6 changed files with 134 additions and 39 deletions
--- a/bookshelfplus/src/main/java/plus/bookshelf/Controller/Controller/FileController.java
+++ b/bookshelfplus/src/main/java/plus/bookshelf/Controller/Controller/FileController.java
@@ -95,17 +95,6 @@ public class FileController extends BaseController {
        return CommonReturnType.create(fileVOS);
    }

-    private FileVO convertFileVOFromModel(FileModel fileModel) {
-        if (fileModel == null) {
-            return null;
-        }
-        FileVO fileVO = new FileVO();
-        BeanUtils.copyProperties(fileModel, fileVO);
-        fileVO.setFileCreateAt(fileModel.getFileCreateAt().getTime());
-        fileVO.setFileModifiedAt(fileModel.getFileModifiedAt().getTime());
-        return fileVO;
-    }
-
    @ApiOperation(value = "【管理员】查询文件对象列表", notes = "查询文件列表")
    @RequestMapping(value = "object/list", method = {RequestMethod.POST}, consumes = {CONTENT_TYPE_FORMED})
    @ResponseBody
@@ -125,18 +114,56 @@ public class FileController extends BaseController {
        return CommonReturnType.create(fileObjectVOS);
    }

-    private FileObjectVO convertFileObjectVOFromModel(FileObjectModel fileObjectModel) {
-        if (fileObjectModel == null) {
-            return null;
+    @ApiOperation(value = "【管理员】更新文件对象上传状态", notes = "重新从 COS 对象存储中获取文件对象上传状态")
+    @RequestMapping(value = "object/refreshFileObjectStatus", method = {RequestMethod.POST}, consumes = {CONTENT_TYPE_FORMED})
+    @ResponseBody
+    public CommonReturnType refreshFileObjectStatus(@RequestParam(value = "token", required = false) String token,
+                                                    @RequestParam(value = "fileObjectId") Integer fileObjectId) throws InvocationTargetException, IllegalAccessException, BusinessException {
+
+        // 验证用户权限
+        UserModel userModel = userService.getUserByToken(redisTemplate, token);
+        if (userModel == null || !Objects.equals(userModel.getGroup(), "ADMIN")) {
+            throw new BusinessException(BusinessErrorCode.OPERATION_NOT_ALLOWED, "非管理员用户无权进行此操作");
        }
-        FileObjectVO fileObjectVO = new FileObjectVO();
-        BeanUtils.copyProperties(fileObjectModel, fileObjectVO);
-        try {
-            // 尝试将 FileStorageMedium 转为中文，如果没有成功，那么就保留英文
-            fileObjectVO.setStorageMediumType(FileStorageMediumEnum.valueOf(fileObjectModel.getStorageMediumType()).getStorageMediumDisplayName());
-        } catch (Exception e) {
+
+        // 获取旧文件权限
+        FileObjectModel fileObjectModel = fileObjectService.getFileObjectById(fileObjectId);
+        String fileSha1 = fileObjectModel.getFileSha1();
+        String oldUploadStatus = fileObjectModel.getUploadStatus();
+
+        QCloudCosUtils qCloudCosUtils = new QCloudCosUtils(qCloudCosConfig, cosPresignedUrlGenerateLogService);
+
+        // 判断对象是否存在
+        Boolean isExist = qCloudCosUtils.doesObjectExist(QCloudCosUtils.BOOK_SAVE_FOLDER, fileSha1);
+
+        Boolean isSuccess = false;
+
+        if (!isExist) {
+            // 现在文件不存在
+            switch (oldUploadStatus) {
+                case "NOT_EXIST": // 文件不存在，不需要更新
+                default:
+                    isSuccess = true;
+                    break;
+                case "SUCCESS": // 上传成功，但是文件不存在了，更新为不存在
+                case "UPLOADING": // 上传中，更新为不存在
+                    isSuccess = fileObjectService.updateFileStatus(fileObjectModel.getId(), "NOT_EXIST");
+                    break;
+            }
+        } else {
+            // 现在文件存在
+            switch (oldUploadStatus) {
+                case "SUCCESS": // 之前上传成功，现在存在，不做任何操作
+                default:
+                    isSuccess = true;
+                    break;
+                case "NOT_EXIST": // 之前不存在，现在存在，更新为上传成功
+                case "UPLOADING": // 之前上传中，现在存在，更新为上传成功
+                    isSuccess = fileObjectService.updateFileStatus(fileObjectModel.getId(), "SUCCESS");
+                    break;
+            }
        }
-        return fileObjectVO;
+        return CommonReturnType.create(isSuccess);
    }

    /**
@@ -196,7 +223,7 @@ public class FileController extends BaseController {
        // 判断对象是否存在
        Boolean isExist = qCloudCosUtils.doesObjectExist(bookSaveFolder, fileSha1);
        String url = null;
-                switch (httpMethodName) {
+        switch (httpMethodName) {
            case PUT:
                // 上传文件
                if (isExist) throw new BusinessException(BusinessErrorCode.PARAMETER_VALIDATION_ERROR, "文件已存在");
@@ -300,6 +327,31 @@ public class FileController extends BaseController {
        }
        return CommonReturnType.create("success");
    }
+
+    private FileVO convertFileVOFromModel(FileModel fileModel) {
+        if (fileModel == null) {
+            return null;
+        }
+        FileVO fileVO = new FileVO();
+        BeanUtils.copyProperties(fileModel, fileVO);
+        fileVO.setFileCreateAt(fileModel.getFileCreateAt().getTime());
+        fileVO.setFileModifiedAt(fileModel.getFileModifiedAt().getTime());
+        return fileVO;
+    }
+
+    private FileObjectVO convertFileObjectVOFromModel(FileObjectModel fileObjectModel) {
+        if (fileObjectModel == null) {
+            return null;
+        }
+        FileObjectVO fileObjectVO = new FileObjectVO();
+        BeanUtils.copyProperties(fileObjectModel, fileObjectVO);
+        try {
+            // 尝试将 FileStorageMedium 转为中文，如果没有成功，那么就保留英文
+            fileObjectVO.setStorageMediumType(FileStorageMediumEnum.valueOf(fileObjectModel.getStorageMediumType()).getStorageMediumDisplayName());
+        } catch (Exception e) {
+        }
+        return fileObjectVO;
+    }
 }

 /*
--- a/bookshelfplus/src/main/java/plus/bookshelf/Service/Impl/FileObjectServiceImpl.java
+++ b/bookshelfplus/src/main/java/plus/bookshelf/Service/Impl/FileObjectServiceImpl.java
@@ -78,7 +78,7 @@ public class FileObjectServiceImpl implements FileObjectService {
    }

    private FileObjectModel convertFromDataObject(FileObjectDO fileObjectDO) throws InvocationTargetException, IllegalAccessException {
-        if(fileObjectDO == null) {
+        if (fileObjectDO == null) {
            return null;
        }
        FileObjectModel fileObjectModel = new FileObjectModel();
@@ -103,7 +103,7 @@ public class FileObjectServiceImpl implements FileObjectService {
    }

    private FileObjectDO convertFromFileObjectModel(FileObjectModel fileObjectModel) {
-        if(fileObjectModel == null) {
+        if (fileObjectModel == null) {
            return null;
        }
        FileObjectDO fileObjectDO = new FileObjectDO();
@@ -213,7 +213,7 @@ public class FileObjectServiceImpl implements FileObjectService {
    }

    /**
-     * 通过文件路径获取文件
+     * 通过文件路径获取文件对象
     *
     * @param filePath 文件路径
     * @return
@@ -224,4 +224,19 @@ public class FileObjectServiceImpl implements FileObjectService {
        FileObjectModel fileObjectModel = convertFromDataObject(fileObjectDO);
        return fileObjectModel;
    }
+
+    /**
+     * 通过 Id 获取文件对象
+     *
+     * @param fileObjectId 文件对象 Id
+     * @return
+     * @throws InvocationTargetException
+     * @throws IllegalAccessException
+     */
+    @Override
+    public FileObjectModel getFileObjectById(Integer fileObjectId) throws InvocationTargetException, IllegalAccessException {
+        FileObjectDO fileObjectDO = fileObjectDOMapper.selectByPrimaryKey(fileObjectId);
+        FileObjectModel fileObjectModel = convertFromDataObject(fileObjectDO);
+        return fileObjectModel;
+    }
 }
--- a/bookshelfplus/src/main/java/plus/bookshelf/Service/Service/FileObjectService.java
+++ b/bookshelfplus/src/main/java/plus/bookshelf/Service/Service/FileObjectService.java
@@ -72,4 +72,14 @@ public interface FileObjectService {
     * @return
     */
    FileObjectModel getFileObjectByFilePath(String filePath) throws InvocationTargetException, IllegalAccessException;
+
+    /**
+     * 通过 Id 获取文件对象
+     *
+     * @param fileObjectId 文件对象 Id
+     * @return
+     * @throws InvocationTargetException
+     * @throws IllegalAccessException
+     */
+    FileObjectModel getFileObjectById(Integer fileObjectId) throws InvocationTargetException, IllegalAccessException;
 }