From 9b2eedb85d53ca32610c32c6e50b5230ab3b16cf Mon Sep 17 00:00:00 2001
From: sean.zhou <sean.zhou@dji.com>
Date: Fri, 22 Jul 2022 20:16:03 +0800
Subject: [PATCH] V1.1.0 for dock

---
 src/main/java/com/dji/sample/media/controller/FileController.java |   35 ++++++++++++++++++++++++++++-------
 1 files changed, 28 insertions(+), 7 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..8fa8451 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,15 @@
 package com.dji.sample.media.controller;
 
+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.service.IFileService;
 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 java.util.List;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URL;
 
 /**
  * @author sean
@@ -29,8 +29,29 @@
      * @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<MediaFileDTO>> getFilesList(@RequestParam(defaultValue = "1") Long page,
+                               @RequestParam(name = "page_size", defaultValue = "10") Long pageSize,
+                               @PathVariable(name = "workspace_id") String workspaceId) {
+        PaginationData<MediaFileDTO> filesList = fileService.getJobsPaginationByWorkspaceId(workspaceId, page, pageSize);
         return ResponseResult.success(filesList);
     }
+
+    /**
+     * Query the download address of the file according to the media file fingerprint,
+     * and redirect to this address directly for download.
+     * @param workspaceId
+     * @param fingerprint
+     * @param response
+     */
+    @GetMapping("/{workspace_id}/file/{fingerprint}/url")
+    public void getFileUrl(@PathVariable(name = "workspace_id") String workspaceId,
+                           @PathVariable String fingerprint, HttpServletResponse response) {
+
+        try {
+            URL url = fileService.getObjectUrl(workspaceId, fingerprint);
+            response.sendRedirect(url.toString());
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
 }

--
Gitblit v1.9.3