前言

好久沒有更新spring Boot 這個專案了。最近看了一下docker 的知識，後期打算將spring boot 和docker 結合起來。剛好最近有一個上傳檔案的工作呢，剛好就想起這個腳手架，將檔案上傳和下載整理進來。

配置

在application.properties 中增加上傳檔案存放的路徑配置

#檔案上傳目錄
file.upload.url=E:/test

controller 層

上傳檔案和下載檔案都比較簡單，我們就直接在controller層來編寫。也不用在pom.xml 中增加什麼依賴。所以直接上程式碼。
在controller 包下建立一個file包，在file 包下建立一個FileController 類。

@RestController
@RequestMapping("file")
@Slf4j
public class FileController {
    @Value("${file.upload.url}")
    private String uploadFilePath;
        
        @RequestMapping("/upload")
    public String httpUpload(@RequestParam("files") MultipartFile files[]){
        JSONObject object=new JSONObject();
        for(int i=0;i<files.length;i++){
            String fileName = files[i].getOriginalFilename();  // 檔名
            File dest = new File(uploadFilePath +'/'+ fileName);
            if (!dest.getParentFile().exists()) {
                dest.getParentFile().mkdirs();
            }
            try {
                files[i].transferTo(dest);
            } catch (Exception e) {
                log.error("{}",e);
                object.put("success",2);
                object.put("result","程式錯誤，請重新上傳");
                return object.toString();
            }
        }
        object.put("success",1);
        object.put("result","檔案上傳成功");
        return object.toString();
    }

}
 

上面的程式碼看起來有點多，其實就是一個上傳的方法，首先通過 MultipartFile 接收檔案。這裡我用的是file[] 陣列接收檔案，這是為了相容多檔案上傳的情況，如果只用file 接收，然後在介面上傳多個檔案的話，只會接收最後一個檔案。這裡大家注意一下。看自己的需求，我這裡相容多檔案所以用陣列接收。

然後遍歷files 獲取檔案，下面這段程式碼是判斷檔案在所在目錄是否存在，如果不存在就建立對應的目錄。

 File dest = new File(uploadFilePath +'/'+ fileName);
            if (!dest.getParentFile().exists()) {
                dest.getParentFile().mkdirs();
            }
 

 files[i].transferTo(dest);

就是將檔案存放到對應的伺服器，這裡有一點需要說明一下，如果我們上傳重複的檔案會怎麼樣麼？上傳重複的檔案不會報錯，後上傳的檔案會直接覆蓋已經上傳的檔案。

整體程式碼就是這樣。現在就可以實現檔案的上傳操作。

測試

我們寫好之後，基本上傳功能就已經實現了，我們現在來測試一下。啟動專案後我們用postman 請求，因為我們需要上傳檔案，用get 方式請求不了。

可以看到檔案上傳成功了，由此可見，springboot檔案上傳一個方法就搞定了。

檔案下載

其實檔案下載，不太建議用介面做，因為檔案下載一般都是下載一些靜態檔案，我們可以先將檔案處理好，然後通過Nginx 服務下載靜態檔案，這樣速度會快很多。但是這裡我們還是寫一下。程式碼也很簡單，就一個方法，也寫在fileController 類中

 @RequestMapping("/download")
    public String fileDownLoad(HttpServletResponse response, @RequestParam("fileName") String fileName){
        File file = new File(downloadFilePath +'/'+ fileName);
        if(!file.exists()){
            return "下載檔案不存在";
        }
        response.reset();
        response.setContentType("application/octet-stream");
        response.setCharacterEncoding("utf-8");
        response.setContentLength((int) file.length());
        response.setHeader("Content-Disposition", "attachment;filename=" + fileName );

        try(BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));) {
            byte[] buff = new byte[1024];
            OutputStream os  = response.getOutputStream();
            int i = 0;
            while ((i = bis.read(buff)) != -1) {
                os.write(buff, 0, i);
                os.flush();
            }
        } catch (IOException e) {
            log.error("{}",e);
            return "下載失敗";
        }
        return "下載成功";
    }

程式碼也很簡單，就是根據檔名判斷是否存在檔案，不存在就提示沒有檔案，存在就將檔案下載下來。response設定返回檔案的格式，以檔案流的方式返回，採用utf-8 字符集，設定下載後的檔名。然後就是以檔案流的方式下載檔案了。

測試的話也簡單，我們啟動專案，訪問介面
http://localhost:9090/zlflovemm/file/download?fileName=11 http://localhost:9090/zlflovemm/file/download?fileName=1.rar


可以看到如果檔案存在，會直接下載，不會提示下載成功或者失敗。

# 刪除檔案

刪除檔案是很簡單的，我這裡講一下刪除檔案下所有資料夾和檔案。並做一個定時任務，每天清理一次。
```

@Scheduled(cron="0 0 3 * * ?")
private void deleteFiles(){
    deleteFile(new File(deleteFilePath));
}

public void deleteFile(File file){
    //判斷檔案不為null或檔案目錄存在
    if (file == null || !file.exists()){
        log.info("暫無檔案");
        return;
    }
    //取得這個目錄下的所有子檔案物件
    File[] files = file.listFiles();
    //遍歷該目錄下的檔案物件
    for (File f: files){
        //列印檔名
        String name = f.getName();
        log.info(name);
        //判斷子目錄是否存在子目錄,如果是檔案則刪除
        if (f.isDirectory()){
            deleteFile(f);
        }else {
            f.delete();
        }
    }
    //刪除空資料夾  for迴圈已經把上一層節點的目錄清空。
    file.delete();
}

```

# 番外
到此為止，我們常用的映象和容器的操作就會使用啦。都是一些命令。忘記的可以--help 檢視一下。

好了，就說這麼多啦
程式碼上傳到github：
https://github.com/QuellanAn/zlflovemm

後續加油♡

歡迎大家關注個人公眾號 "程式設計師愛酸奶"

分享各種學習資料，包含java，linux，大資料等。資料包含視訊文件以及原始碼，同時分享本人及投遞的優質技術博文。

如果大家喜歡記得關注和分享喲❤

相關推薦