package com.bell.web.controller;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import lombok.extern.slf4j.Slf4j;
@RestController
@RequestMapping("/files")
@Slf4j
public class FileController {
private String parent = "D:/files";
@GetMapping("/{filename:.+}")//:.+ 保证获取到文件的扩展名
public void download(@PathVariable String filename,
HttpServletResponse response) throws Exception {
if (!StringUtils.hasText(filename)) {
response.getWriter().write("filename has error");
return;
}
try (InputStream inputStream = new FileInputStream(new File(parent, filename));
OutputStream outputStream = response.getOutputStream();){
response.setContentType("application/x-download");
response.setHeader("Content-Disposition", "attachment;filename="+filename);
IOUtils.copy(inputStream, outputStream);
} catch (Exception e) {
throw e;
}
}
@PostMapping
public String upload(MultipartFile file) throws Exception {
log.info(file.getName());
log.info(file.getOriginalFilename());
log.info(file.getSize()+"size");
String originalFilename = file.getOriginalFilename();
String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
File localfile = new File(parent, System.currentTimeMillis()+suffix);
file.transferTo(localfile);
return localfile.getAbsolutePath();
}
}