0


spring boot上传文件,并限制上传文件类型、上传文件大小

  1. 近日上传文件遇到诸多问题,我以为上传就一个io流就能搞定的东西,结果写完给我同事测试一下漏洞百出,问题1.单次上传文件超过10MB就会报错,文件太大。问题2.文件限制文件类型。问题3.即使使用逻辑限制了类型只要有人有心上传两个文件,第一个是能过校验第二个就不会校验了。那么我们一起来解决一下这些问题。

问题1、

上传文件这个有很多方案可以解决,可以直接一个异常捕获返回文件太大,也可以将文件容量扩大,而这里我是两个搭配使用,配置了yml文件,还配置了异常捕获,也可以直接在上传的代码那捕获这个异常 MaxUploadSizeExceededException ,我这是新建了一个IExceptionHandler直接全局进行配置了

  1. spring:
  2. servlet:
  3. multipart:
  4. max-file-size: 50MB # 最大上传大小限制为50MB。任何尝试上传超过这个大小的单个文件都会被拒绝。
  5. max-request-size: 50MB # 整个请求的最大大小限制为50MB。这包括所有上传文件的总和以及其他请求参数的大小

全局捕获异常

  1. @ExceptionHandler(MaxUploadSizeExceededException.class)
  2. public JsonResult handlerAnyException(MaxUploadSizeExceededException e) {
  3. e.printStackTrace();
  4. var status = HttpStatus.INTERNAL_SERVER_ERROR;
  5. return JsonResult.custom(
  6. status,
  7. "文件过大",
  8. null
  9. );
  10. }

问题2、

  1. 我当前项目需要只能上传PDF文件,所以我的文件中是这样限制的,直接在逻辑中限制文件类型(完整代码放文章最后)
  1. if (!pdfFile.getContentType().equals("application/pdf")) {
  2. return JsonResult.badRequest("请上传PDF文件");
  3. }

问题3、

问题3的话我觉得无伤大雅,因为只要只接收一个文件参数,那么就算传再多文件,也只会读取第一个。

以下是完整代码

  1. @Override
  2. @Transactional
  3. public JsonResult uploadPdfFile(MultipartFile pdfFile) {
  4. if (pdfFile.isEmpty()) {
  5. return JsonResult.badRequest("请上传文件");
  6. }
  7. if (!pdfFile.getContentType().equals("application/pdf")) {
  8. return JsonResult.badRequest("请上传PDF文件");
  9. }
  10. try {
  11. // 获取文件名
  12. String fileName = pdfFile.getOriginalFilename();
  13. String prefixPath = Constent.FILE_URI + UserHolder.getUser().getUserId() + "/";
  14. // 拼接文件路径,判断是否有该路径没有则文件夹
  15. File dirPath = new File(prefixPath);
  16. if (!dirPath .exists()) dirPath .mkdirs();
  17. OutputStream out = new FileOutputStream(prefixPath + fileName);
  18. out.write(pdfFile.getBytes());
  19. // 将文件信息入库
  20. files = new Files();
  21. files.setFileName(fileName);
  22. files.setFilePath(prefixPath + fileName);
  23. filesService.save(files);
  24. out.flush();
  25. out.close();//关闭输出流
  26. return JsonResult.success(files);
  27. } catch (IOException e) {
  28. log.error(e.getMessage());
  29. return JsonResult.badRequest("上传失败,请联系管理员");
  30. }
  31. }

如果有补充,或者不对的地方请指正,感谢。

标签: spring boot 后端 java

本文转载自: https://blog.csdn.net/sole_legal/article/details/136678569
版权归原作者 干饭的草莓田田圈 所有, 如有侵权,请联系我们删除。

“spring boot上传文件,并限制上传文件类型、上传文件大小”的评论:

还没有评论