本文由 简悦 SimpRead 转码， 原文地址 blog.csdn.net
近日上传文件遇到诸多问题，我以为上传就一个 io 流就能搞定的东西，结果写完给我同事测试一下漏洞百出，问题 1. 单次上传文件超过 10MB 就会报错，文件太大。问题 2. 文件限制文件类型。问题 3. 即使使用逻辑限制了类型只要有人有心上传两个文件，第一个是能过校验第二个就不会校验了。那么我们一起来解决一下这些问题。
问题 1、
上传文件这个有很多方案可以解决，可以直接一个异常捕获返回文件太大，也可以将文件容量扩大，而这里我是两个搭配使用，配置了 yml 文件，还配置了异常捕获，也可以直接在上传的代码那捕获这个异常 MaxUploadSizeExceededException ，我这是新建了一个 IExceptionHandler 直接全局进行配置了
spring:
servlet:
multipart:
max-file-size: 50MB # 最大上传大小限制为50MB。任何尝试上传超过这个大小的单个文件都会被拒绝。
max-request-size: 50MB # 整个请求的最大大小限制为50MB。这包括所有上传文件的总和以及其他请求参数的大小
全局捕获异常
@ExceptionHandler(MaxUploadSizeExceededException.class)
public JsonResult handlerAnyException(MaxUploadSizeExceededException e) {
e.printStackTrace();
var status = HttpStatus.INTERNAL_SERVER_ERROR;
return JsonResult.custom(
status,
"文件过大",
null
);
}
问题 2、
我当前项目需要只能上传 PDF 文件，所以我的文件中是这样限制的，直接在逻辑中限制文件类型（完整代码放文章最后）
if (!pdfFile.getContentType().equals("application/pdf")) {
return JsonResult.badRequest("请上传PDF文件");
}
问题 3、
问题 3 的话我觉得无伤大雅，因为只要只接收一个文件参数，那么就算传再多文件，也只会读取第一个。
以下是完整代码
@Override
@Transactional
public JsonResult uploadPdfFile(MultipartFile pdfFile) {
if (pdfFile.isEmpty()) {
return JsonResult.badRequest("请上传文件");
}
if (!pdfFile.getContentType().equals("application/pdf")) {
return JsonResult.badRequest("请上传PDF文件");
}
try {
// 获取文件名
String fileName = pdfFile.getOriginalFilename();
String prefixPath = Constent.FILE_URI + UserHolder.getUser().getUserId() + "/";
// 拼接文件路径，判断是否有该路径没有则文件夹
File dirPath = new File(prefixPath);
if (!dirPath .exists()) dirPath .mkdirs();
OutputStream out = new FileOutputStream(prefixPath + fileName);
out.write(pdfFile.getBytes());
// 将文件信息入库
files = new Files();
files.setFileName(fileName);
files.setFilePath(prefixPath + fileName);
filesService.save(files);
out.flush();
out.close();//关闭输出流
return JsonResult.success(files);
} catch (IOException e) {
log.error(e.getMessage());
return JsonResult.badRequest("上传失败，请联系管理员");
}
}
如果有补充，或者不对的地方请指正，感谢。