重复消费input stream的方法

最新推荐文章于 2024-01-30 16:05:49 发布

weixin_33797791

最新推荐文章于 2024-01-30 16:05:49 发布

阅读量105

点赞数

原文链接：https://my.oschina.net/go4it/blog/1526499

版权

为什么80%的码农都做不了架构师？>>>

序

本文介绍一下如何重复消费input stream，普通的inputStream，消费一次之后，就不能再用了，有时候需要重复消费的话，就必须自己缓存一下。这里定义了ReuseableStream类，可以用来实现这个目的。

ReuseableStream

public class ReuseableStream {

    private InputStream inputStream;

    public ReuseableStream(InputStream inputStream) {
        if (!inputStream.markSupported()) {
            this.inputStream = new BufferedInputStream(inputStream);
        } else {
            this.inputStream = inputStream;
        }
    }

    public InputStream open() {
        inputStream.mark(Integer.MAX_VALUE);
        return inputStream;
    }

    public void reset() throws IOException {
        inputStream.reset();
    }
}

开启并重复使用

ReuseableStream reuse = new ReuseableStream(IOUtils.toInputStream("hello", Charsets.UTF_8));
System.out.println(IOUtils.toString(reuse.open(),Charsets.UTF_8));
reuse.reset();
System.out.println(IOUtils.toString(reuse.open(),Charsets.UTF_8));