feat: add an api for downloading file (#30)

Co-authored-by: Rick <linuxsuren@users.noreply.github.com>

Author: LinuxSuRen

.github/workflows/build.yaml

@@ -24,6 +24,6 @@ jobs:
           PULL_REQUEST: ${{ github.event.number }}
         run: |
           sudo curl -L https://github.com/docker/compose/releases/download/v2.23.0/docker-compose-linux-x86_64 -o /usr/local/bin/docker-compose
-          sudo chmod u+x /usr/local/bin/docker-compose
+          sudo chmod +x /usr/local/bin/docker-compose
           export OWNER=$(echo ${{ github.repository_owner }} | tr '[:upper:]' '[:lower:]')
           make test-e2e

Makefile

@@ -6,6 +6,8 @@ run-demo:
 	cd e2e && docker compose up server
 run:
 	mvn package && java -jar target/demo-0.0.1-SNAPSHOT.jar
+run-8081:
+	mvn package && java -Dserver.port=8081 -jar target/demo-0.0.1-SNAPSHOT.jar
 local-test:
 	atest run -p e2e/test-suite.yaml
 

e2e/test-suite.yaml

@@ -110,3 +110,13 @@ items:
     api: /actuator/prometheus
     header:
       Authorization: "{{ .param.auth }}"
+
+## files
+- name: downloadFile
+  request:
+    api: /file/download
+    header:
+      Authorization: "{{ .param.auth }}"
+  expect:
+    header:
+      Content-Type: application/octet-stream

src/main/java/io/github/devopsws/demo/service/FileService.java

@@ -2,7 +2,13 @@
 
 import java.io.FileOutputStream;
 
+import org.springframework.core.io.Resource;
+import org.springframework.core.io.UrlResource;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
 import org.springframework.util.FileCopyUtils;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
@@ -12,9 +18,9 @@
 import io.github.devopsws.demo.model.Message;
 
 @RestController
-@RequestMapping("/upload")
+@RequestMapping("/file")
 public class FileService {
-    @PostMapping("/")
+    @PostMapping("/upload")
     public Message<?> upload(@RequestParam("file") MultipartFile file) {
         System.out.println("Received file uploading request");
         Message<String> message = new Message<String>();
@@ -34,4 +40,17 @@ public Message<?> upload(@RequestParam("file") MultipartFile file) {
         }
         return message;
     }
+
+    @GetMapping("/download")
+    public ResponseEntity<Resource> downloadFile() {
+        Resource resource = new UrlResource(FileService.class.getResource("/application.yml"));
+        if (resource.exists() || resource.isReadable()) {
+            return ResponseEntity.ok()
+                    .contentType(MediaType.APPLICATION_OCTET_STREAM)
+                    .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + resource.getFilename() + "\"")
+                    .body(resource);
+        } else {
+            return ResponseEntity.notFound().build();
+        }
+    }
 }