如何使用输入流标记和重置功能？

Question

我正在尝试从文件中读取第一个8192字节，并通过返回布尔值的方法运行这些字节。

这个布尔值告诉我文件是否属于特定类型。

如果该方法在文件类型的字节上返回 true，则我想获取剩余的字节并通过其他方法运行它们。如果为 false，则通过其他方法运行剩余的字节。

我正在尝试使用标记，但没有成功。

private final void handleFile(InputStream inputStream) {

   BufferedInputStream bis = new BufferedInputStream(inputStream);
   bis.mark(8192);
   byte[] startingBytes = inputStreamToByteArray(bis);

   if(startingBytes.length == 0) { return; }

   byte[] finalBytes;
   if(isFileType(startingBytes)) {
      bis.reset();
      finalBytes = inputStreamToByteArray(bis);
      methodForFinalBytes(finalBytes);
   } else {
      // Do other stuff;
   }
}

private byte[] inputStreamToByteArray(InputStream inputStream) {
   ByteArrayOutputStream baos = new ByteArrayOutputStream();
   byte[] buffer = new byte[8192];

   try {
      while(inputStream.read(buffer) != -1) {
          baos.write(buffer);
      }
   } catch(IOException ioe) {
      ioe.printStackTrace();
   }

return baos.toByteArray();
}

问题是从我停下来的地方继续，同时将字节数组保持在块中（用于处理大文件）。此外，我只返回了8192字节，而不是剩余的字节。

有什么建议吗？

Answer 1

首先，看起来inputSTreamToByteArray（）读取整个流直到文件结束，而不仅仅是前8192个字节。您可能希望首先分别读取前8192个字节。

其次，您想再次重新读取这些字节吗？如果没有，我不确定您是否需要标记/重置。（尤其是因为您还有字节数组）

下面的代码首先读取前8192个字节，然后决定做什么：

byte[] header = new byte[8192];
//reads entire array or until EOF whichever is first
bis.read(header);

if(isFileType(header)) {
  byte[] finalBytes = inputStreamToByteArray(bis);
  methodForFinalBytes(finalBytes);
} else {
  // Do other stuff;
}