将字节流转换为字节数组，无需额外空间

Posted 2021-04-21

我有一个ByteArrayOutputStream，它写入了大量数据，最终转换为字节数组并写入缓存：

try {
      ByteArrayOutputStream baos = new ByteArrayOutputStream();
      try (JsonGenerator jg = mapper.getFactory().createGenerator(baos)) {
        for (Object result : results) {
          jg.writeObject(result);
        }
      }

      localCache.put(cacheKey, baos.toByteArray());
    }
    catch (IOException e) {
      throw Throwables.propagate(e);
    } 

在这里，baos.toByteArray()在内存中创建了一个全新的数据副本，我试图避免这种情况。有没有办法将流转换为字节数组而不使用额外的内存？

答案

内部缓冲区和当前计数是Javadoc中记录的protected字段。这意味着您可以将ByteArrayOutputStream子类化并提供byte[] getBuffer()方法直接访问缓冲区。使用现有的size()方法确定存在多少数据。

public class MyBAOS extends ByteArrayOutputStream
{
    public MyBAOS()           { super(); }
    public MyBAOS(int size)   { super(size); }
    public byte[] getBuffer() { return buf; }
}