处理流中不同大小的字节数组

Question

我有一个方法“A”在每次调用时返回一个固定大小（例如 64 字节）的byte[outputSize]。我还有一个方法“B”需要输入可变大小的字节[inputSize]（例如，它可以小于或大于 64 字节）。 我想使用一个流来提供从“A”到“B”的 64 字节块序列，而不会丢失任何单个字节。

我尝试将这些 64 位块提供给 ByteArrayOutputStream “out”，直到达到 out.size > inputSize。 然后从“out”的副本创建一个ByteArrayInputStream“in”，以使用从“in”读取的byte[inputSize] 提供方法“B”。

然后重置“out”并将未读字节复制到“out”。 这是实现这一目标的正确方法还是有更简单/更快/更好的方法？ 这是一个伪代码示例：

ByteArrayOutputStream out = new ByteArrayOutputStream();

public byte[] A() {
    return next64bytes;
}

public void B(byte[] bytes) throws IOException {
    while (out.size() < bytes.length) {
        out.write(A()); // feeding "out" with 64 bytes
    }
    ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
    in.read(bytes); // feeding "bytes" with bytes.length bytes
    byte unread[];
    in.read(unread = new byte[in.available()]);
    out.reset();
    out.write(unread); // basically writing unread bytes of "in" in an empty "out"
}

Answer 1

使用ByteArrayOutputStream 作为缓冲区并不是超级高效，但它确实使代码更简洁，这通常更重要。以下是您可以做得更好的方法：

public void B(byte[] bytes) {
    while (out.size() < bytes.length) {
        byte[] chunk = A();
        out.write(chunk, 0, chunk.length);
    }
    byte[] buffered = out.toByteArray();
    out.reset();
    System.arraycopy(buffered, 0, bytes, 0, bytes.length);
    out.write(buffered, bytes.length, buffered.length - bytes.length);
}

Answer 2

为什么不简单地使用 ByteArrayOutputStream.writeTo(B)？

编辑：所以在拖出你真正想做的事情之后；为此忘记流。这个问题可以用流来解决，但它并不比复制字节简单。由于流是为 I/O 设计的，因此您最终会得到额外的异常处理，但对问题没有额外的价值。

使用成员变量来记住之前调用 B() 的剩余字节，并使用普通循环填充数组，当剩余部分用完时从 A() 请求更多数据。