From 8d9a2d656e4ae007590c622e5f7c228adacdca49 Mon Sep 17 00:00:00 2001
From: rain <167982779@qq.com>
Date: Fri, 14 Jun 2024 10:11:36 +0800
Subject: [PATCH] 统一风格

---
 src/main/java/com/dji/sample/media/controller/FileController.java |   92 +++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 86 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/dji/sample/media/controller/FileController.java b/src/main/java/com/dji/sample/media/controller/FileController.java
index 25a6015..e32836a 100644
--- a/src/main/java/com/dji/sample/media/controller/FileController.java
+++ b/src/main/java/com/dji/sample/media/controller/FileController.java
@@ -1,15 +1,27 @@
 package com.dji.sample.media.controller;
 
+import com.dji.sample.common.model.CustomClaim;
+import com.dji.sample.common.model.PaginationData;
 import com.dji.sample.common.model.ResponseResult;
+import com.dji.sample.media.model.FileUploadDTO;
 import com.dji.sample.media.model.MediaFileDTO;
+import com.dji.sample.media.model.MediaFileEntity;
+import com.dji.sample.media.model.MediaFileQueryParam;
 import com.dji.sample.media.service.IFileService;
+import com.drew.imaging.ImageProcessingException;
+import org.apache.ibatis.annotations.Update;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import java.awt.*;
+import java.io.IOException;
+import java.net.URL;
 import java.util.List;
+
+import static com.dji.sample.component.AuthInterceptor.TOKEN_CLAIM;
 
 /**
  * @author sean
@@ -25,12 +37,80 @@
 
     /**
      * Get information about all the media files in this workspace based on the workspace id.
+     * 根据工作空间id获取有关此工作空间中所有媒体文件的信息。
+     *
      * @param workspaceId
      * @return
      */
     @GetMapping("/{workspace_id}/files")
-    public ResponseResult<List<MediaFileDTO>> getFilesList(@PathVariable(name = "workspace_id") String workspaceId) {
-        List<MediaFileDTO> filesList = fileService.getAllFilesByWorkspaceId(workspaceId);
+    public ResponseResult<PaginationData<MediaFileEntity>> getFilesList(@RequestParam(defaultValue = "1") Long page,
+                                                                        @RequestParam(name = "page_size", defaultValue = "10") Long pageSize,
+                                                                        @PathVariable(name = "workspace_id") String workspaceId,
+                                                                        MediaFileQueryParam mediaFileQueryParam) {
+        PaginationData<MediaFileEntity> filesList = fileService.getMediaFilesPaginationByWorkspaceId(workspaceId, page, pageSize, mediaFileQueryParam);
         return ResponseResult.success(filesList);
     }
+
+    @GetMapping("/{workspace_id}/listAdded")
+    public ResponseResult listIsadd(@RequestParam String dkbh,
+                                    @PathVariable(name = "workspace_id") String workspaceId) {
+        List<MediaFileEntity> entityList = fileService.listByIsadd(dkbh, workspaceId);
+        return ResponseResult.success(entityList);
+    }
+
+    @PutMapping("/examine")
+    public ResponseResult examineData(@RequestParam String fileId) {
+        return fileService.updateExamByFileId(fileId);
+    }
+
+    @GetMapping("/getMediaInfo")
+    public ResponseResult mediaInfo(@RequestParam String fileName) {
+        return ResponseResult.success(fileService.mediaInfo(fileName));
+    }
+
+    @GetMapping("/{workspace_id}/files/{job_id}")
+    public ResponseResult findFilesList(@PathVariable(name = "workspace_id") String workspaceId, @PathVariable(name = "job_id") String jobId) {
+        return ResponseResult.success(fileService.listMediaFileEntity(workspaceId, jobId));
+    }
+
+    @PostMapping("/{workspace_id}/updateFile")
+    public ResponseResult updateFile(HttpServletRequest request, @PathVariable(name = "workspace_id") String workspaceId, @RequestBody MediaFileEntity mediaFileEntity) {
+        CustomClaim claims = (CustomClaim) request.getAttribute(TOKEN_CLAIM);
+        mediaFileEntity.setUserId(claims.getId());
+        return ResponseResult.success(fileService.updateMediaFile(workspaceId, mediaFileEntity));
+    }
+
+    @DeleteMapping("/{workspace_id}/deleteFile")
+    public ResponseResult deleteFile(@PathVariable(name = "workspace_id") String workspaceId, @RequestParam String fileId) {
+        int count = fileService.deleteMedia(workspaceId, fileId);
+        if (count == 0) {
+            return ResponseResult.error("删除失败");
+        }
+        return ResponseResult.success();
+    }
+
+
+    /**
+     * Query the download address of the file according to the media file id,
+     * and redirect to this address directly for download.
+     *
+     * @param workspaceId
+     * @param fileId
+     * @param response
+     */
+    @GetMapping("/{workspace_id}/file/{file_id}/url")
+    public void getFileUrl(@PathVariable(name = "workspace_id") String workspaceId,
+                           @PathVariable(name = "file_id") String fileId, HttpServletResponse response) {
+
+        try {
+            URL url = fileService.getObjectUrl(workspaceId, fileId);
+            response.sendRedirect(url.toString());
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+    @PostMapping("/{workspace_id}/mark")
+    public void eaveMarkFile(@PathVariable(name = "workspace_id") String workspaceId,@RequestBody FileUploadDTO file) throws ImageProcessingException, IOException, FontFormatException {
+        fileService.saveMarkFile(workspaceId,file);
+    }
 }

--
Gitblit v1.9.3