一：上传单个pdf文件并用hash重命名及判断文件是不是已经上传过，可以参考hash图片
@PostMapping("/uploadPdfFileSingle")
public String uploadPdfFileSingle(@RequestPart("file") MultipartFile file) {
System.err.println("开始文件上传");
if (file.isEmpty()) {
System.err.println("File cannot be empty");
}
if (!"application/pdf".equals(file.getContentType())) {
System.err.println("Only PDF files are allowed");
}
try {
// 生成文件哈希可以参考图片hash'那篇文章
String fileHash = calculateFileHash(file.getInputStream());
// 处理扩展名
String originalFileName = StringUtils.cleanPath(Objects.requireNonNull(file.getOriginalFilename()));
String fileExtension = originalFileName.contains(".")
? originalFileName.substring(originalFileName.lastIndexOf("."))
: "";
// 使用哈希值作为文件名
String storedFileName = fileHash + fileExtension;
// 创建目录 pdpdf是配置文件配置的根目录
Path uploadPath = Paths.get(pgPdf).toAbsolutePath().normalize();
if (!Files.exists(uploadPath)) {
Files.createDirectories(uploadPath);
}
// 检查文件是否已存在
Path targetLocation = uploadPath.resolve(storedFileName);
if (Files.exists(targetLocation)) {
System.err.println("File is exist"); // 文件已存在
}
// 保存文件
Files.copy(file.getInputStream(), targetLocation, StandardCopyOption.REPLACE_EXISTING);
System.err.println(storedFileName);
} catch (IOException ex) {
ex.printStackTrace();
throw new RuntimeException("File storage failed");
}
}
二：对应的下载方法
@GetMapping("/downLoadPdfFileSingle")
public void downLoadPdfFileSingle(@RequestParam("fileName") String fileName, HttpServletResponse response) throws Exception {
//传的文件名就是上传时打印的文件名storedFileName
File f = new File(pgPdf+fileName);
if (!f.exists()) {
response.sendError(404, "File not found!");
return;
}
BufferedInputStream br = new BufferedInputStream(new FileInputStream(f));
byte[] buf = new byte[1024];
int len = 0;
response.reset(); // 非常重要
//下载
response.setContentType("application/x-msdownload");
response.setHeader("Content-Disposition", "attachment; filename=" + f.getName());
OutputStream out = response.getOutputStream();
while ((len = br.read(buf)) > 0)
out.write(buf, 0, len);
br.close();
out.close();
}