读取InputStream并写入文件

Question

这与我之前的问题-DataInputStream提供java.io.EOFException有关

在该客户端-服务器应用程序中，有一种方法可以检索从服务器发送的文件并将其保存到文件中。

Client.java-

 public void receiveFile(InputStream is, String fileName) throws Exception {
        int filesize = 6022386;
        int bytesRead;
        int current = 0;
        byte[] mybytearray = new byte[filesize];

        System.out.println("Receving File!");
        FileOutputStream fos = new FileOutputStream("RECEIVED_"+fileName);
        BufferedOutputStream bos = new BufferedOutputStream(fos);
        bytesRead = is.read(mybytearray, 0, mybytearray.length);
        current = bytesRead; 
        System.out.println(bytesRead);
        do {
            bytesRead = is.read(mybytearray, current,
                    (mybytearray.length - current));
            System.out.println(bytesRead);
            if (bytesRead >= 0)
                current += bytesRead;
        } while (bytesRead > -1);
        System.out.println("Loop done");
        bos.write(mybytearray, 0, current);
        bos.flush();
        bos.close();
    }
}

Server.Java

public void sendFile(OutputStream os, String fileName) throws Exception {
    File myFile = new File(fileName);
    byte[] mybytearray = new byte[(int) myFile.length() + 1];
    FileInputStream fis = new FileInputStream(myFile);
    BufferedInputStream bis = new BufferedInputStream(fis);
    bis.read(mybytearray, 0, mybytearray.length);
    System.out.println("Sending File!");
    os.write(mybytearray, 0, mybytearray.length);
    os.flush();
    bis.close();
}

如您所见，客户端的receiveFile方法中存在多个receiveFile 。 这是我收到的输出。

问题在于该方法无法完成其任务，并且永远不会到达System.out.println("Loop done");

有什么问题？

Answer 1

稍后我将介绍循环错误。

源头有一个小字节太多：

public void sendFile(OutputStream os, String fileName) throws Exception {
    File myFile = new File(fileName);
    if (myFile.length() > Integer.MAX_VALUE) {
        throw new IllegalStateException();
    }

或

    byte[] mybytearray = Files.readAllBytes(myFile.toPath());

要么

    byte[] mybytearray = new byte[(int) myFile.length()]; // No +1.
    // BufferedInputStream here not needed.
    try (BufferedInputStream bis = new BufferedInputStream(
            new FileInputStream(myFile))) {
        bis.read(mybytearray);
    } // Always closed.

接着

    System.out.println("Sending File!");
    os.write(mybytearray);
    os.flush();
}

另外，我添加了java 7 Files.readAllBytes 。 使用Files.copy甚至可能更简单。

我只是看到已经提到了主要错误。 基本上有一个误解：您可能会完整读取一个字节数组。 它将阻塞直到读完结尾。 如果要读取的内容更少（达到“文件尾”），则返回字节数。 因此，出于所有目的，您可能会完整阅读它。

人们经常看到类似的代码来读取固定大小（为2的幂）的块（例如4096）重复并写入输出流。

同样，java 7 Files简化了所有操作：

    Files.copy(is, Paths.get("RECEIVED_" + fileName));

---

简而言之：

public void receiveFile(InputStream is, String fileName) throws Exception {
    Files.copy(is, Paths.get("RECEIVED_" + fileName),
        StandardCopyOption.REPLACE_EXISTING);
}

public void sendFile(OutputStream os, String fileName) throws Exception {
    Files.copy(Paths.get(fileName), os);
}

Answer 2

您可能希望您的条件显示为：

} while (bytesRead > -1 && current < mybytearray.length);
//                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

目前，当您读取了整个文件时，您的read语句将继续读取零字节（ mybytearray.length - current ），该字节成功并返回零，因此循环永远不会终止。

Answer 3

不保持附加到输入缓冲区会更简单

public void receiveFile(InputStream istream, String fileName) throws Exception {
  System.err.println("Receving File!");
  try (BufferedOutputStream ostream = new BufferedOutputStream(
      new FileOutputStream("RECEIVED_" + fileName))) {
    int bytesRead;
    byte[] mybytearray = new byte[1024];

    do { 
      bytesRead = istream.read(mybytearray);
      if (bytesRead > 0) {
        ostream.write(mybytearray, 0, bytesRead);
      }
    } while (bytesRead != -1);
    System.err.println("Loop done");
    ostream.flush();
  }
}

如果可以使用Guava，则可以使用ByteStreams.copy(java.io.InputStream, java.io.OutputStream) （ Javadoc ）