将 Java 字节读取器转换为 InputStream

Question

考虑一个通用的字节读取器，它实现了以下简单的 API，以从无法访问的数据结构中读取未指定数量的字节：

public interface ByteReader
{
    public byte[] read() throws IOException; // Returns null only at EOF
}

如何将上述内容有效地转换为标准Java InputStream，以便使用InputStream 类定义的所有方法的应用程序按预期工作？

一个简单的解决方案是将InputStream 子类化为

1. 根据InputStream 的read(...) 方法的需要调用ByteReader 的read() 方法
2. 缓冲在 byte[] 数组中检索到的字节
3. 按预期返回字节数组的一部分，例如，每当调用 InputStream read() 方法时，一次返回 1 个字节。

但是，这需要更多的工作才能提高效率（例如，为了避免多字节数组分配）。此外，为了让应用程序扩展到较大的输入大小，将所有内容读入内存然后进行处理不是一种选择。

有什么可以使用的想法或开源实现吗？

Answer 1

我假设，通过您使用“convert”，替换是可以接受的。

最简单的方法是使用ByteArrayInputStream，它已经提供了您正在寻找的所有功能（但必须包装现有数组），或者使用任何其他已经提供的InputStream从各种来源读取数据。

看来您可能正在冒着重新发明轮子的风险。如果可能的话，我会考虑完全废弃您的ByteReader 界面，而是使用以下选项之一：

1. 替换为ByteInputStream。
2. 使用各种其他 InputStream 类（取决于数据源）。
3. 使用您的自定义实现扩展 InputStream。

我会坚持使用现有的 InputStream 类。我不知道您的代码是如何构建的，但您可以，例如，将getInputStream() 方法添加到您当前的数据源，并让它们返回一个适当的已经存在的InputStream（或必要时的自定义子类）。

顺便说一下，我建议在您自己的 IO 类中避免使用术语 Reader，因为在 Java SDK 中已经大量使用 Reader 来指示对编码字符数据进行操作的流读取器（而不是 InputStream它通常对原始字节数据进行操作）。

Answer 2

围绕返回的数组创建多个ByteArrayInputStream 实例，并在提供连接的流中使用它们。例如，您可以为此使用 SequenceInputStream。

诀窍是实现一个可以使用ByteReader 类的Enumeration<ByteArrayInputStream>。

编辑：我已经实现了这个答案，但最好创建自己的 InputStream 实例。不幸的是，这个解决方案不能让你优雅地处理IOException。

---

final Enumeration<ByteArrayInputStream> basEnum = new Enumeration<ByteArrayInputStream>() {

    ByteArrayInputStream baos;
    boolean ended;

    @Override
    public boolean hasMoreElements() {
        if (ended) {
            return false;
        }

        if (baos == null) {
            getNextBA();
            if (ended) {
                return false;
            }
        }

        return true;
    }

    @Override
    public ByteArrayInputStream nextElement() {
        if (ended) {
            throw new NoSuchElementException();
        }
        if (baos.available() != 0) {
            return baos;
        }

        getNextBA();
        return baos;
    }

    private void getNextBA() {

        byte[] next;
        try {
            next = byteReader.read();
        } catch (IOException e) {
            throw new IllegalStateException("Issues reading byte arrays");
        }
        if (next == null) {
            ended = true;
            return;
        }
        this.baos = new ByteArrayInputStream(next);
    }
};

SequenceInputStream sis = new SequenceInputStream(basEnum);