springBoot 简单优雅是实现文件上传和下载
前言
好久沒有更新spring Boot 這個項目了。最近看了一下docker 的知識,后期打算將spring boot 和docker 結合起來。剛好最近有一個上傳文件的工作呢,剛好就想起這個腳手架,將文件上傳和下載整理進來。
配置
在application.properties 中增加上傳文件存放的路徑配置
#文件上傳目錄 file.upload.url=E:/testcontroller 層
上傳文件和下載文件都比較簡單,我們就直接在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=11http://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();}總結
以上是生活随笔為你收集整理的springBoot 简单优雅是实现文件上传和下载的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 第四范式荣获“工业和信息化系统抗击新冠肺
- 下一篇: 第四范式荣获2020年度信创工委会 “卓