From 34f84d91c4c0fee4832f93c2b0e332f2bc868c7b Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Tue, 16 Apr 2024 16:41:46 +0800
Subject: [PATCH] 断点续飞

---
 src/main/java/com/dji/sample/media/controller/FileController.java |   25 ++++++++++++++++++++++---
 1 files changed, 22 insertions(+), 3 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 aaa400d..4834b0f 100644
--- a/src/main/java/com/dji/sample/media/controller/FileController.java
+++ b/src/main/java/com/dji/sample/media/controller/FileController.java
@@ -1,16 +1,21 @@
 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.MediaFileDTO;
+import com.dji.sample.media.model.MediaFileEntity;
 import com.dji.sample.media.model.MediaFileQueryParam;
 import com.dji.sample.media.service.IFileService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.net.URL;
+
+import static com.dji.sample.component.AuthInterceptor.TOKEN_CLAIM;
 
 /**
  * @author sean
@@ -31,14 +36,28 @@
      * @return
      */
     @GetMapping("/{workspace_id}/files")
-    public ResponseResult<PaginationData<MediaFileDTO>> getFilesList(@RequestParam(defaultValue = "1") Long page,
+    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<MediaFileDTO> filesList = fileService.getMediaFilesPaginationByWorkspaceId(workspaceId, page, pageSize,mediaFileQueryParam);
+                                                                      MediaFileQueryParam mediaFileQueryParam) {
+        PaginationData<MediaFileEntity> filesList = fileService.getMediaFilesPaginationByWorkspaceId(workspaceId, page, pageSize,mediaFileQueryParam);
         return ResponseResult.success(filesList);
     }
 
+    @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));
+    }
+
+
+
     /**
      * Query the download address of the file according to the media file id,
      * and redirect to this address directly for download.

--
Gitblit v1.9.3