Java课程实验 Thymeleaf 模板应用

package as.lwl.control; import org.apache.commons.io.FileUtils; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpStatus; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.stereotype.Controller; import org.springframework.ui.Model; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletRequest; import java.io.File; import java.net.URLEncoder; import java.util.UUID; @Controller public class FileController { String pathParent = "G:/File"; //向文件上传页面跳转 @GetMapping("/toUpload") public String toUpload(){ return "file/upload"; } //文件上传管理 @PostMapping("/uploadFile") public String uploadFile(MultipartFile[] fileUpload, Model model) { //上传成功返回状态信息 model.addAttribute("uploadStatus", "上传成功！"); for (MultipartFile file : fileUpload) { //获取文件名以及后缀名 String fileName = file.getOriginalFilename(); //重新生成文件名 fileName = UUID.randomUUID()+"_"+fileName; //指定上传文件本地存储目录，不存在则需创建 String dirPath = pathParent+"/upload/"; File filePath = new File(dirPath); if(!filePath.exists()){ filePath.mkdirs(); } try { file.transferTo(new File(dirPath+fileName)); } catch (Exception e) { e.printStackTrace(); //上传失败返回失败信息 model.addAttribute("uploadStatus","上传失败： "+e.getMessage()); } } //携带上传状态信息回调到文件上传页面 return "file/upload"; } //向文件下载页面跳转 @GetMapping("/toDownload") public String toDownload(){ return "file/download"; } @GetMapping("/download") //所有类型文件下载管理 public ResponseEntity<byte[]> fileDownload(HttpServletRequest request, String filename) throws Exception{ //指定要下载的文件根路径 String dirPath = pathParent+"/download/"; //创建该文件对象 File file = new File(dirPath + File.separator + filename); //设置响应头 HttpHeaders headers = new HttpHeaders(); //通知浏览器已下载方式打开（下载前对文件名进行转码） filename=getFilename(request,filename); headers.setContentDispositionFormData("attachment",filename); //定义以流的方式下载返回文件数据 headers.setContentType(MediaType.APPLICATION_OCTET_STREAM); try { return new ResponseEntity<>(FileUtils.readFileToByteArray(file), headers, HttpStatus.OK); } catch (Exception e) { e.printStackTrace(); return new ResponseEntity<byte[]>(e.getMessage().getBytes(),HttpStatus.EXPECTATION_FAILED); } } //根据浏览器的不同编码设置，返回编码的文件名 private String getFilename(HttpServletRequest request, String filename) throws Exception { //IE不同版本User-Agent中出现的关键字 String[] IEBrowserKeyWords = {"MSIE", "Trident", "Edge"}; //获取请求头代理信息 String userAgent = request.getHeader("User-Agent"); for (String keyWord : IEBrowserKeyWords) { if (userAgent.contains(keyWord)) { //IE内核浏览器，统一为UTF-8编码显示，并对转换的+进行更正 return URLEncoder.encode(filename, "UTF-8").replace("+"," "); } } //火狐等其他浏览器统一为ISO-8859-1编码显示 return new String(filename.getBytes("UTF-8"), "ISO-8859-1"); } }