inputstream重复使用方法
程序员文章站
2022-04-08 21:39:46
...
在项目中需要对同个文件流进行两个操作,一个是上传文件流到HDFS上,一个是上传文件流到solr建立文件索引。由于流只能读取一次。于是要想办法让流可以重复使用。
在网络上查找到一个方法,就是将inputstream转换成ByteArrayOutputStream,要用流时再用baos转换回来就可以了。
代码如下:
ByteArrayOutputStream baos = new ByteArrayOutputStream();
byte[] buffer = new byte[1024];
int len;
while ((len = input.read(buffer)) > -1 ) {
baos.write(buffer, 0, len);
}
baos.flush();
InputStream stream1 = new ByteArrayInputStream(baos.toByteArray());
InputStream stream2 = new ByteArrayInputStream(baos.toByteArray());