近日上传文件遇到诸多问题,我以为上传就一个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("上传失败,请联系管理员");
}
}
如果有补充,或者不对的地方请指正,感谢。
版权归原作者 干饭的草莓田田圈 所有, 如有侵权,请联系我们删除。