Springboot上传文件,但是使用CommonsMultipartResolver解析 并 使用 Multipartfile接收到文件后发现文件变大,且文件打不开,文件损坏

Springboot上传文件,

使用 CommonsMultipartResolver 解析

使用 Multipartfile 接收到文件后发现文件变大,且文件打不开,文件损坏

这是因为使用了自定义 CustomHttpServletRequestWrapper 包装类继承了 HttpServletRequestWrapper ,获取request 并读取了inputstream把inputsream转为string。

而文件上传时,inpustream转为string,就会出现解析出错,文件损坏。

解决:克隆inputstream 并保存 byte[] ,读取时 byte[] 转换为 inputstream ,这样可以多次读取inputstream

下面是助手类,克隆inputstream 并保存 byte[]



import java.io.*;

public class FileUtils {

    //TODO 克隆Inputstream 转换为byte[]
    public static byte[] cloneInputStream(InputStream inputStream){
        try {
            ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
            byte[] buffer = new byte[1024];
            int len;
            while ((len = inputStream.read(buffer)) > -1){
                byteArrayOutputStream.write(buffer,0,len);
            }
            byteArrayOutputStream.flush();
            byteArrayOutputStream.close();
            inputStream.close();
            return byteArrayOutputStream.toByteArray();
        }catch (IOException e){
            e.printStackTrace();
        }
        return null;
    }

    //TODO 使用byte[] 获取 克隆Inputstream
    public static InputStream getInputStream(byte[] bytes){
        InputStream inputStreamNew = null;
        if(bytes != null){
            inputStreamNew = new ByteArrayInputStream(bytes);
        }
        return inputStreamNew;
    }

    //TODO 使用byte[] 获取 克隆Inputstream
    public static ByteArrayInputStream getByteArrayInputStream(byte[] bytes){
        ByteArrayInputStream inputStreamNew = null;
        if(bytes != null){
            inputStreamNew = new ByteArrayInputStream(bytes);
        }
        return inputStreamNew;
    }
}

 下面是自定义 CustomHttpServletRequestWrapper 包装类继承了 HttpServletRequestWrapper



import FileUtils;

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;

public class CustomHttpServletRequestWrapper extends HttpServletRequestWrapper {

    // 保存request body的数据
    private String body;
    // 保存body byte[]
    private byte[] bodyBytes;




    // 解析request的inputStream(即body)数据,转成字符串
    public CustomHttpServletRequestWrapper(HttpServletRequest request) {
        super(request);
        StringBuilder stringBuilder = new StringBuilder();
        BufferedReader bufferedReader = null;
        InputStream inputStream = null;
        try {
            inputStream = request.getInputStream();
            //TODO 克隆 inputstream 到 byte[]
            bodyBytes = FileUtils.cloneInputStream(inputStream);
            inputStream = FileUtils.getInputStream(bodyBytes);

            if (inputStream != null) {
                bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
                char[] charBuffer = new char[128];
                int bytesRead = -1;
                while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
                    stringBuilder.append(charBuffer, 0, bytesRead);
                }
            } else {
                stringBuilder.append("");
            }
        } catch (IOException ex) {
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                }
                catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (bufferedReader != null) {
                try {
                    bufferedReader.close();
                }
                catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        body = stringBuilder.toString();
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        //TODO 使用 byte[] 转换为 getByteArrayInputStream , 这样文件上传时就不会出现文件损坏
        final ByteArrayInputStream byteArrayInputStream = FileUtils.getByteArrayInputStream(bodyBytes);  
        ServletInputStream servletInputStream = new ServletInputStream() {
            @Override
            public boolean isFinished() {
                return false;
            }
            @Override
            public boolean isReady() {
                return false;
            }
            @Override
            public void setReadListener(ReadListener readListener) {
            }
            @Override
            public int read() throws IOException {
                return byteArrayInputStream.read();
            }
        };
        return servletInputStream;
    }


    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(this.getInputStream()));
    }
    public String getBody() {
        return this.body;
    }
    // 赋值给body字段
    public void setBody(String body) {
        this.body = body;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
使用@Async异步处理上传文件时,可能会遇到文件不存在的异常。这是由于异步处理会在上传文件完成之前就返回,导致文件还未上传完成,异步处理就已经开始了。 解决方法是使用CompletableFuture类来等待文件上传完成后再处理,示例代码如下: ``` @Service public class FileService { @Autowired private MinioClient minioClient; @Async public CompletableFuture<String> uploadFile(MultipartFile file, String bucketName, String objectName) { try { minioClient.putObject(bucketName, objectName, file.getInputStream(), file.getContentType()); return CompletableFuture.completedFuture("File uploaded successfully"); } catch (Exception e) { return CompletableFuture.failedFuture(e); } } public String handleFileUpload(MultipartFile file) throws Exception { String bucketName = "my-bucket"; String objectName = file.getOriginalFilename(); CompletableFuture<String> future = uploadFile(file, bucketName, objectName); future.thenAccept(result -> { // 处理上传文件成功的逻辑 }).exceptionally(ex -> { // 处理上传文件失败的逻辑 return null; }).join(); return "File upload started"; } } ``` 在handleFileUpload方法中,先调用uploadFile方法上传文件,然后使用CompletableFuture类等待文件上传完成后再进行处理。在future.thenAccept方法中处理上传文件成功的逻辑,在future.exceptionally方法中处理上传文件失败的逻辑。最后调用join方法等待异步处理完成。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值