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

本文讲述了在上传文件时遇到的问题,如文件大小超过限制、文件类型限制以及如何通过Spring配置和全局异常处理来解决这些问题。作者提供了详细代码示例,包括配置文件大小限制、检查文件类型以及处理MaxUploadSizeExceededException异常。
摘要由CSDN通过智能技术生成

        近日上传文件遇到诸多问题,我以为上传就一个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("上传失败,请联系管理员");
        }
    }

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

  • 9
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Spring Boot中,可以通过配置文件application.properties或application.yml来限制上传文件大小。如果你想使用XML文件来进行配置的话,可以在Spring Boot项目的classpath下创建一个名为`applicationContext.xml`的文件,并在其中添加以下内容: ``` <bean id="multipartResolver" class="org.springframework.web.multipart.commons.CommonsMultipartResolver"> <property name="maxUploadSize" value="1000000" /> </bean> ``` 其中,`maxUploadSize`属性设置了最大的上传文件大小,单位为字节。上述配置将允许上传的文件大小最大为1MB。你可以根据需要进行修改。 另外,如果你想针对不同的路径设置不同的上传文件大小限制,可以使用`RequestMappingHandlerMapping`来进行配置。具体的做法是在`applicationContext.xml`文件中添加以下内容: ``` <bean id="handlerMapping" class="org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping"> <property name="interceptors"> <list> <bean class="org.springframework.web.servlet.handler.MappedInterceptor"> <constructor-arg> <util:constant static-field="org.springframework.web.servlet.HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE" /> </constructor-arg> <constructor-arg> <bean class="com.example.UploadSizeInterceptor" /> </constructor-arg> </bean> </list> </property> </bean> ``` 其中,`UploadSizeInterceptor`是自定义的拦截器,用于对上传文件大小进行限制。你需要在`UploadSizeInterceptor`中实现`preHandle`方法,并在其中根据请求路径设置不同的上传文件大小限制。具体的做法可以参考以下代码: ``` public class UploadSizeInterceptor extends HandlerInterceptorAdapter { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { String path = request.getServletPath(); if (path.startsWith("/upload1")) { CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(); multipartResolver.setMaxUploadSize(1000000L); if (!multipartResolver.isMultipart(request)) { throw new Exception("上传文件格式不正确"); } } else if (path.startsWith("/upload2")) { CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(); multipartResolver.setMaxUploadSize(2000000L); if (!multipartResolver.isMultipart(request)) { throw new Exception("上传文件格式不正确"); } } return true; } } ``` 上述代码中,对于请求路径以`/upload1`开头的请求,设置最大上传文件大小为1MB;对于请求路径以`/upload2`开头的请求,设置最大上传文件大小为2MB。你可以根据需要进行修改。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值