MultipartFile 对上传的文件大小进行限制

@Slf4j
@Component
public class MultipartFileUtil {

private final static Integer FILE_SIZE = 5;//文件上传限制大小
private final static String FILE_UNIT = "M";//文件上传限制单位(B,K,M,G)

    /**
     * @param len  文件长度
     * @param size 限制大小
     * @param unit 限制单位(B,K,M,G)
     * @描述 判断文件大小
     */
    public static boolean checkFileSize(Long len, int size, String unit) {
        double fileSize = 0;
        if ("B".equalsIgnoreCase(unit)) {
            fileSize = (double) len;
        } else if ("K".equalsIgnoreCase(unit)) {
            fileSize = (double) len / 1024;
        } else if ("M".equalsIgnoreCase(unit)) {
            fileSize = (double) len / 1048576;
        } else if ("G".equalsIgnoreCase(unit)) {
            fileSize = (double) len / 1073741824;
        }
        return !(fileSize > size);
    }

    //文件上传调用
    public static String upload(MultipartFile file) {
        boolean flag = checkFileSize(file.getSize(), FILE_SIZE, FILE_UNIT);
        if (!flag) {
            throw new RuntimeException("上传文件大小超出限制");
        }
    }

}
PS:以上基础需要项目本身设定好最大文件阈值,在阈值的基础上进行限制大小提示,如果不设置,文件上传大小默认1MB,超出则抛异常

#设置单个文件最大请求100MB,最多一次请求2个文件(具体设定看自己需求)
spring.servlet.multipart.max-file-size=100MB
spring.servlet.multipart.max-request-size=200MB

#设置文件上传大小不进行限制
spring.servlet.multipart.max-file-size=-1
spring.servlet.multipart.max-request-size=-1

/**
     * @描述 文件下载
     * @参数 [response, filePath, filename]
     * @返回值 javax.servlet.http.HttpServletResponse
     * @创建时间 2021/6/29
     */
    public static void download(HttpServletResponse response, String filePath, String filename) {
        try {
            // path是指欲下载的文件的路径
            File file = new File(filePath);
            // 以流的形式下载文件
            InputStream fis = new BufferedInputStream(new FileInputStream(filePath));
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            // 清空response
            response.reset();
            // 设置response的Header
            response.setContentType("application/octet-stream; charset=UTF-8");
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "utf-8"));
            response.addHeader("Content-Length", "" + file.length());
            OutputStream toClient = new BufferedOutputStream(response.getOutputStream());
            toClient.write(buffer);
            toClient.flush();
            toClient.close();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }
  • 4
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Java中,你可以使用MultipartFile对象来对文件大小和格式进行限制。下面是一个示例代码,演示了如何对文件对象MultipartFile进行文件大小限制为1MB,并且要求文件格式为PDF: ```java import org.springframework.web.multipart.MultipartFile; public class FileUploadService { private static final long MAX_FILE_SIZE = 1 * 1024 * 1024; // 1MB public void uploadFile(MultipartFile file) { // 检查文件大小 if (file.getSize() > MAX_FILE_SIZE) { throw new IllegalArgumentException("File size exceeds the maximum limit (1MB)."); } // 检查文件格式 String fileExtension = getFileExtension(file.getOriginalFilename()); if (!"pdf".equalsIgnoreCase(fileExtension)) { throw new IllegalArgumentException("Only PDF files are allowed."); } // 执行文件上传操作 // ... } // 获取文件扩展名 private String getFileExtension(String fileName) { int dotIndex = fileName.lastIndexOf("."); if (dotIndex > 0 && dotIndex < fileName.length() - 1) { return fileName.substring(dotIndex + 1).toLowerCase(); } return ""; } } ``` 在上述示例中,`uploadFile`方法接收一个MultipartFile对象作为参数。首先,通过调用`getSize()`方法获取文件的大小,然后与预定义的最大文件大小(1MB)进行比较,如果文件大小超过限制,则抛出异常。 接下来,通过调用`getOriginalFilename()`方法获取文件名,并通过`getFileExtension`方法获取文件扩展名。如果文件扩展名不是"pdf"(不区分大小写),则抛出异常。 请注意,上述示例中的`MAX_FILE_SIZE`和`getFileExtension`方法是示例中的一些自定义常量和方法,你可以根据实际需求进行调整和扩展。 这只是一个简单的文件上传限制示例,实际应用中可能还需要考虑其他安全性和业务需求。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值