MockMultipartFile 将 File转为InputStream时,大文件会报错的问题

原代码:

multipartFile = new MockMultipartFile(originFileName, inputStream);

现象:

在这里插入图片描述

在 Java 中,MockMultipartFile 是 Spring Framework 测试模块中的一个工具类,它用于在测试环境中模拟 MultipartFile 对象

解决:

使用 CommonsMultipartFile

    // File 转 MultipartFile
    private MultipartFile createMultipartFile(String filePath, InputStream inputStream) throws IOException {
        // 创建临时文件
        int lastIndexOf = filePath.lastIndexOf(".");
        if (lastIndexOf > 0) {
            filePath = filePath.substring(0, lastIndexOf);
        }
        filePath = filePath.substring(filePath.lastIndexOf(File.separator) + 1);
        File tempFile = File.createTempFile("upload_", "_" + filePath);
        logger.info("tempFile: " + tempFile.getAbsolutePath());
        tempFile.deleteOnExit(); // 确保程序退出时删除临时文件

        // 将输入流写入临时文件
        try (FileOutputStream outputStream = new FileOutputStream(tempFile)) {
            byte[] buffer = new byte[8192];
            int bytesRead;
            while ((bytesRead = inputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, bytesRead);
            }
        }

        // 使用 DiskFileItem 创建 MultipartFile
        FileItemFactory factory = new DiskFileItemFactory();
        String textFieldName = "file";
        FileItem item = factory.createItem(textFieldName, "application/octet-stream", false, filePath);

        try (InputStream tempInputStream = new FileInputStream(tempFile);
             OutputStream os = item.getOutputStream()) {
            byte[] buffer = new byte[8192];
            int bytesRead;
            while ((bytesRead = tempInputStream.read(buffer)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
        }

        return new CommonsMultipartFile(item);
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值