Java INPUTSTREAM如何实现重复使用

引语：

之前做项目的时候遇到一个问题,就是从网络中读取的图片要上传到oss,而且要对图片进行裁剪和压缩,其中上传和裁剪都要使用到图片的inputstream,

又因为inputstream不能重复读,导致裁剪是成功的,而上传是失败的.我们今天就提供两种方法来解决,inputstream不能重复读的问题.

问题分析:

inputstream的内部有个pos指针,当读取的时候指针会不断的移动,当移动到末尾的时候,就无法再次读取了.

我们写个简单的例子来看下:

string text = "测试inputstream内容";
inputstream inputstream = new bytearrayinputstream(text.getbytes());
byte[] readarray = new byte[inputstream.available()];
int readcount1 = inputstream.read(readarray);
system.out.println("读取了" + readcount1 + "个字节");

byte[] readarray2 = new byte[inputstream.available()];
int readcount2 = inputstream.read(readarray2);
system.out.println("读取了" + readcount2 + "个字节");
/**
* 执行结果是
* 读取了23个字节
* 读取了-1个字节
*/

从执行结果可以看出确实inputstream的设计是只能读取一次.

注意: 这里稍微提一下inputstream.available()这个方法,本地的文件可以直接知道文件的大小,但是如果是网络中的数据,这个方法最好不要用,因为传输的时候不是连续的,数据的大小会读取不准

问题解决:

那么我们实际项目中应该怎么解决呢?总不能就真的只使用一次inputsteam吧.我们来看解决方法:

方法一:

使用bytearrayoutputstream来缓存字节,然后每次读取从缓存的bytearrayoutputstream中拿取.

很自然的想到把inputstream的缓存起来(当然不一定说是要放在bytearrayoutputstream,其他的方式也可以,都是缓存起来的思路,实现方式有很多种,这种比较方便)

string text = "测试inputstream内容";
inputstream rawinputstream = new bytearrayinputstream(text.getbytes());
bytearrayoutputstream outputstream = new bytearrayoutputstream();
byte[] buffer = new byte[1024];
int len;
while ((len = rawinputstream.read(buffer)) > -1) {
outputstream.write(buffer, 0, len);
}
outputstream.flush();
inputstream in1 = new bytearrayinputstream(outputstream.tobytearray());
inputstream in2 = new bytearrayinputstream(outputstream.tobytearray());
int readcount1 = in1.read(buffer);
int readcount2 = in2.read(buffer);
system.out.println("读取了" + readcount1 + "个字节");
system.out.println("读取了" + readcount2 + "个字节");
/**
* 执行结果是
* 读取了23个字节
* 读取了23个字节
*

这里是先将inputstream的数据读取到output中,然后要反复使用inputstream中的内容的时候,我们将output中的数据取出(很神奇的设定,output可以反复取,input只能读一次)

方法二:

其实inputstream中有操作指针的方法,mark和reset,听名字就知道是标记和重置.在使用inputsteam前我们标记下inputstream指针的位置,读取完之后,重置,然后就可以反复使用了.我们看代码:

string text = "测试inputstream内容";
inputstream rawinputstream = new bytearrayinputstream(text.getbytes());
byte[] readarray = new byte[1024];
rawinputstream.mark(0);
int readcount1 = rawinputstream.read(readarray);
rawinputstream.reset();
int readcount2 = rawinputstream.read(readarray);
system.out.println("读取了" + readcount1 + "个字节");
system.out.println("读取了" + readcount2 + "个字节");

总结:

1.inputstream只能读取一次,也就是说只能调用read()或者其他的带参数的read()方法一次,在下次调用读取出来是-1,做项目的时候不要忘记这一点了,可能会导致有些坑出现;

2.可以使用缓存或者mark/reset方法来重复使用inputstream,这里要注意的是如果inputstream如果内容很多,缓存不是一个好办法,因为在使用完之前会占用大量的内存(我遇到过这样的,上传很多图片然后还有缓存,导致内存不够就一直fullgc,然后cpu先爆了);

3.还有一个小点就是别忘了关闭使用完的inputstream/outputsteam.

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。