Spring Boot實現(xiàn)對文件進行壓縮下載功能

更新時間：2023年09月18日 10:04:36 作者：Jamie Chyi

在Web應用中,文件下載功能是一個常見的需求,特別是當你需要提供用戶下載各種類型的文件時,本文將演示如何使用Spring Boot框架來實現(xiàn)一個簡單而強大的文件下載功能,需要的朋友跟隨小編一起學習吧

在Web應用中，文件下載功能是一個常見的需求，特別是當你需要提供用戶下載各種類型的文件時。本文將演示如何使用Spring Boot框架來實現(xiàn)一個簡單而強大的文件下載功能。我們將創(chuàng)建一個RESTful API，通過該API，用戶可以下載問價為ZIP壓縮文件。

1. 創(chuàng)建Spring Boot 項目

首先，確保你已經創(chuàng)建了一個Spring Boot項目，并在項目中添加了所需的依賴。在這個示例中，我們將使用Spring Boot的Web模塊和Spring的MVC框架。

2. 編寫下載控制器

創(chuàng)建一個名為 DownloadController 的RESTful控制器，用于處理文件下載請求。在這個控制器中，我們將定義一個 downloadStudentWork 方法，用于下載學生作品的ZIP壓縮文件。具體文件流來源與具體業(yè)務。

@RestController
@RequestMapping("/download")
public class DownloadController {
    @GetMapping("/studentWork")
    public ResponseEntity<StreamingResponseBody> downloadStudentWork() {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
        String encodedFileName = "作品名稱-學生姓名.zip";
        try {
            encodedFileName = URLEncoder.encode(encodedFileName, StandardCharsets.UTF_8.toString());
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        headers.setContentDispositionFormData("attachment", encodedFileName);
        StreamingResponseBody responseBody = outputStream -> {
            try (ZipOutputStream zipOut = new ZipOutputStream(outputStream)) {
                // 假設這是學生的作品內容視頻文件流
                InputStream videoStream = getStudentVideoStream();
                addToZip(zipOut, videoStream, "作品視頻.mkv");
                // 添加更多附件，如果有的話
                zipOut.finish();
            } catch (IOException e) {
                // 處理異常
            }
        };
        return new ResponseEntity<>(responseBody, headers, HttpStatus.OK);
    }
    private void addToZip(ZipOutputStream zipOut, InputStream inputStream, String fileName) throws IOException {
        ZipEntry zipEntry = new ZipEntry(fileName);
        zipOut.putNextEntry(zipEntry);
        byte[] buffer = new byte[1024];
        int bytesRead;
        while ((bytesRead = inputStream.read(buffer)) != -1) {
            zipOut.write(buffer, 0, bytesRead);
        }
        zipOut.closeEntry();
        inputStream.close();
    }
    // 獲取文件流
    private InputStream getStudentVideoStream() throws FileNotFoundException {
        // 附件信息
        FileInputStream inputStream = new FileInputStream("C:\\Users\\28111\\Videos\\2023-09-14 21-30-36.mkv");
        return inputStream;
    }
}