实现文件的上传和下载
package com.atguigu.mvc.controller;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpSession;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.UUID;
@Controller
public class FileUpAndDownController {
@RequestMapping("/testDown")
public ResponseEntity<byte[]> testResponseEntity(HttpSession session) throws IOException {
ServletContext servletContext = session.getServletContext();
String realPath = servletContext.getRealPath("/static/img/wallhaven-x8k5wz.png");
System.out.println(realPath);
InputStream is = new FileInputStream(realPath);
byte[] bytes = new byte[is.available()];
is.read(bytes);
MultiValueMap<String, String> headers = new HttpHeaders();
headers.add("Content-Disposition", "attachment;filename=wallhaven-x8k5wz.pg");
HttpStatus statusCode = HttpStatus.OK;
ResponseEntity<byte[]> responseEntity = new ResponseEntity<>(bytes, headers, statusCode);
is.close();
return responseEntity;
}
@RequestMapping("/testUp")
public String testUp(MultipartFile photo ,HttpSession httpSession) throws IOException {
String originalFilename = photo.getOriginalFilename();
String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
String uuid = UUID.randomUUID().toString();
originalFilename = uuid + suffix;
ServletContext httpSessionServletContext = httpSession.getServletContext();
String photoPath = httpSessionServletContext.getRealPath("photo");
File file = new File(photoPath);
if (!file.exists()) file.mkdir();
String finalPath = photoPath + File.separator + originalFilename;
photo.transferTo(new File(finalPath));
return "success";
}
}
<bean id="multipartResolver" class="org.springframework.web.multipart.commons.CommonsMultipartResolver"/>
<!DOCTYPE html>
<html lang="en" xmlns:th="http://www.thymeleaf.org">
<head>
<meta charset="UTF-8">
<title>测试文件的上传和下载</title>
</head>
<body>
<a th:href="@{/testDown}">下载 wallhaven-x8k5wz.png</a><br>
<form th:action="@{/testUp}" method="post" enctype="multipart/form-data">
头像:<input type="file" name="photo"><br>
<input type="submit" value="上传">
</form>
</body>
</html>