针对项目中间环节使用到的输入流复用（可以看做是复制拷贝）

输入流复用技巧

最新推荐文章于 2023-08-14 17:13:26 发布

原创最新推荐文章于 2023-08-14 17:13:26 发布 · 615 阅读

1 ·

CC 4.0 BY-SA版权

Java IO 专栏收录该内容

0 篇文章

订阅专栏

本文介绍了一种在Java中实现输入流复用的方法，通过使用StreamReuse类来避免将流转换为文件或byte[]时可能遇到的问题。这种方法适用于无法保存临时文件的场景。

一般情况下一个输入流 stream，正常情况使用了一次，stream就已经没有用了，不能起到第二次利用。一般情况都是采用先把stream转化为File 文件保存然后反复利用，这种比较多见,之所以写这个，就是因为有些情况下不能转化为文件保存起来利用输入输出流复用，转化为byte[] 经常会发生格式不正确或其他问题，这种项目中间环节不可能让你在项目中生成一个临时文件，那么这种BufferedInputStream 就很好的保存了最原始的输入流，
在使用时候只需要标记使用markUsed（），通过该方法会自动重置流，也就是说你下次getInputStream()的时候又是一个新的一模一样的流。

例如：

public String uploadFile(String bizCode, InputStream stream){
StreamReuse reuse = new StreamReuse(stream);
// 1.第一次使用了stream，正常情况，stream就已经没有用了，不能起到第二次利用。
public xxxxxxxxxxxxxxxxxx(stream);
// 2.标记流已经使用了
reuse.markUsed();
// 3.第二次使用,直接get一下就可以获取一个新的
public xxxxxxxxxxxxxxx(reuse.getInputStream());
}

StreamReuse 源码：

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;

/**
 * Desc: 输入流的备份
 * Author: zhanghl2
 * Date: 2016/12/20
 * Time: 10:52
 * Version:
 */
public class StreamReuse {
        private InputStream input;
        public StreamReuse(InputStream input) {
            if (!input.markSupported()) {
                this.input = new BufferedInputStream(input);
            } else {
                this.input = input;
            }
        }

        public InputStream getInputStream() {
            input.mark(Integer.MAX_VALUE);
            return input;
        }

        public void markUsed() throws IOException {
            input.reset();
        }
}