Spring-boot的文件上传和下载

た入场券 2024-04-17 15:09 32阅读 0赞

先创建一个upload的html文件，代码如下：

<!DOCTYPE html>
    <html>
    <head>
    	<meta charset="utf-8"> 
    	<title>文件上传和下载</title>
    </head>
    <body>
    <form action='http://localhost:8080/file/upload' method='post' enctype='multipart/form-data'>
    	<input type='file' name='file'>
    	<button type='submit'>上传</button>
    </form>
    </body>
    </html>

接下来是webjava的内容了。

@Controller
    @RequestMapping("/file/")
    public class FileController {
    	@RequestMapping("upload")
    	@ResponseBody
    	public String upload (@RequestParam("file") MultipartFile file) {
    		// 获取原始名字
    		String fileName = file.getOriginalFilename();
    		// 获取后缀名
    		// String suffixName = fileName.substring(fileName.lastIndexOf("."));
    		// 文件保存路径
    		String filePath = "d:/upload/";
    		// 文件重命名，防止重复
    		fileName = filePath + UUID.randomUUID() + fileName;
    		// 文件对象
    		File dest = new File(fileName);
    		// 判断路径是否存在，如果不存在则创建
    		if(!dest.getParentFile().exists()) {
    			dest.getParentFile().mkdirs();
    		}
    		try {
    			// 保存到服务器中
    			file.transferTo(dest);
    			return "上传成功";
    		} catch (Exception e) {
    			e.printStackTrace();
    		}
    		return "上传失败";
    	}
    }

运行起来吧，我们会发现页面上提示上传成功，并且在电脑的d:/upload/中存在一个我们刚刚上传的文件，文件上传就做好了，是不是很简单。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3R6eHNjOA_size_16_color_FFFFFF_t_70]  
接下来就是文件下载了，这个时候不再需要创建html文件了，我们可以直接输入网址进行下载，我们来看看核心代码，也是非常简单的。

@RequestMapping("download")
    	public void download(HttpServletResponse response) throws Exception {
    		// 文件地址，真实环境是存放在数据库中的
    		File file = new File("D:\\upload\\a.txt");
    		// 穿件输入对象
    		FileInputStream fis = new FileInputStream(file);
    		// 设置相关格式
    		response.setContentType("application/force-download");
    		// 设置下载后的文件名以及header
    		response.addHeader("Content-disposition", "attachment;fileName=" + "a.txt");
    		// 创建输出对象
    		OutputStream os = response.getOutputStream();
    		// 常规操作
    		byte[] buf = new byte[1024];
    		int len = 0;
    		while((len = fis.read(buf)) != -1) {
    			os.write(buf, 0, len);
    		}
    		fis.close();
    	}

运行起来，当我们在浏览器的地址栏中输入http://localhost:8080/file/upload时会发现，文件下载了，文件名为a.txt，即response.addHeader(“Content-disposition”, “attachment;fileName=” + “a.txt”);最后的这个文件名，这个可以自定义

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3R6eHNjOA_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/17/2ac1f36a92644a0384a30b14eaf68c72.png