Java InputStream 读取问题

Question

我有一个 Java 类，我通过 InputStream 读取数据

    byte[] b = null;
    try {
        b = new byte[in.available()];
        in.read(b);
    } catch (IOException e) {
        e.printStackTrace();
    }

当我从 IDE (Eclipse) 运行我的应用程序时，它可以完美运行。

但是当我导出我的项目并将它打包在一个 JAR 中时，读取命令不会读取所有数据。我该如何解决？

这个问题主要发生在 InputStream 是一个文件 (~10kb) 时。

谢谢！

Answer 1

通常我更喜欢在从输入流中读取时使用固定大小的缓冲区。正如 evilone 指出的那样，使用 available() 作为缓冲区大小可能不是一个好主意，因为例如，如果您正在读取远程资源，那么您可能事先不知道可用字节。您可以阅读 InputStream 的 javadoc 以获得更多信息。

这里是我通常用来读取输入流的代码sn-p：

byte[] buffer = new byte[BUFFER_SIZE];

int bytesRead = 0;
while ((bytesRead = in.read(buffer)) >= 0){
  for (int i = 0; i < bytesRead; i++){
     //Do whatever you need with the bytes here
  }
}

我在这里使用的 read() 版本将尽可能地填充给定的缓冲区，并且 返回实际读取的字节数。这意味着您的缓冲区可能包含尾随垃圾数据，因此仅使用最多 bytesRead 的字节非常重要。

注意(bytesRead = in.read(buffer)) >= 0这一行，InputStream 规范中没有任何内容说 read() 不能读取 0 字节。 您可能需要处理 read() 读取 0 字节的特殊情况情况取决于你的情况。对于本地文件，我从未遇到过这种情况；但是，在读取远程资源时，我实际上看到 read() 不断读取 0 个字节，导致上述代码进入无限循环。我通过计算读取 0 字节的次数来解决无限循环问题，当计数器超过阈值时，我将抛出异常。你可能不会遇到这个问题，但请记住这一点:)

出于性能原因，我可能不会为每次读取创建新的字节数组。

Answer 2

当 InputStream 耗尽时，read() 将返回 -1。还有一个版本的read 接受一个数组，这允许您进行分块读取。它在 InputStream 结束时返回实际读取的字节数或-1。将此与动态缓冲区（例如 ByteArrayOutputStream）结合起来，得到以下结果：

InputStream in = ...
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
int read;
byte[] input = new byte[4096];
while ( -1 != ( read = in.read( input ) ) ) {
    buffer.write( input, 0, read );
}
input = buffer.toByteArray()

这大大减少了您必须调用的方法数量，并允许 ByteArrayOutputStream 更快地增长其内部缓冲区。

Answer 3

File file = new File("/path/to/file");

try {
   InputStream is = new FileInputStream(file);
   byte[] bytes = IOUtils.toByteArray(is);

   System.out.println("Byte array size: " + bytes.length);
} catch (IOException e) {
   e.printStackTrace();
}

Answer 4

下面是一段代码，用于下载文件（*.Png、*.Jpeg、*.Gif、...）并将其写入代表 HttpServletResponse 的 BufferedOutputStream。

BufferedInputStream inputStream = bo.getBufferedInputStream(imageFile);
try {
    ByteArrayOutputStream buffer = new ByteArrayOutputStream();
    int bytesRead = 0;
    byte[] input = new byte[DefaultBufferSizeIndicator.getDefaultBufferSize()];
    while (-1 != (bytesRead = inputStream.read(input))) {
        buffer.write(input, 0, bytesRead);
    }
    input = buffer.toByteArray();

    response.reset();
    response.setBufferSize(DefaultBufferSizeIndicator.getDefaultBufferSize());
    response.setContentType(mimeType);
    // Here's the secret. Content-Length should equal the number of bytes read.
    response.setHeader("Content-Length", String.valueOf(buffer.size()));
    response.setHeader("Content-Disposition", "inline; filename=\"" + imageFile.getName() + "\"");

    BufferedOutputStream outputStream = new BufferedOutputStream(response.getOutputStream(), DefaultBufferSizeIndicator.getDefaultBufferSize());
    try {
        outputStream.write(input, 0, buffer.size());
    } finally {
        ImageBO.close(outputStream);
    }
} finally {
    ImageBO.close(inputStream);
}

希望这会有所帮助。