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

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

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值