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

通常我更喜欢在从输入流读取时使用固定大小的缓冲区。正如vilone指出的那样，使用available()作为缓冲区大小可能不是一个好主意，因为，比如说，如果您正在读取远程资源，那么您可能无法提前知道可用字节。您可以阅读 的 javadoc输入流以获得更多洞察。

这是我通常用于读取输入流的代码片段：

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);
}

希望这有帮助。