乐趣区

怎么重复使用inputStream

jiezi

引语：

    之前做项目的时候遇到一个问题, 就是从网络中读取的图片要上传到 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.

退出移动版