While循环在文件下载后不退出

Question

我有以下代码来下载通过TCP传输的文件：

        try (OutputStream out = new FileOutputStream(path); InputStream is = socket.getInputStream();) {
            byte[] bytes = new byte[1024];
            int count, xp = 0;
            while ((count = is.read(bytes)) > 0) {      // TODO after upload the service doesn't leave while loop
                out.write(bytes, 0, count);
            }
            System.out.println("hello");
         ...

if (ready.equalsIgnoreCase(CdnResponse.READY.getContext())){
    int read = 0;
    byte[] bytes = new byte[1024];
    while ((read = inputStream.read(bytes)) != -1) {
        out.write(bytes, 0, read);
    }

}

一旦所有的字节都被处理了（它们总是被处理的，它像正常的一样工作，只是不退出循环），文件就被创建了，没有一个问题，但是，循环没有退出。

Answer 1

TCP/IP连接被设计为长期流式连接（构建在无序、无保证、基于数据包的IP协议之上）。

这意味着is.read(bytes)完全按照规范的要求执行：它将等待至少一个字节可用，或者“流结束”信号进入。只要这两种情况都不发生（没有字节到达，但流没有关闭），它就会自动阻塞。如果有必要的话永远。

解决方法是[A]预先发送文件的大小，然后调整循环，使其只在收到该数量的字节后退出，或者[B]关闭流。

// upload side
void upload() {
  Path p = Paths.get("/path/to/file/you/want/to/send");
  long fileSize = Files.size(p);
  out.write(longToBytes(fileSize);
  try (InputStream in = Files.newInputStream(p)) {
    in.transferTo(out);
  }
}

public static byte[] longToBytes(long x) {
    ByteBuffer buffer = ByteBuffer.allocate(Long.BYTES);
    buffer.putLong(x);
    return buffer.array();
}

此代码具有以下属性：

• 它首先发送8个字节，按大字节顺序发送，这是将要发送的数据的大小。
• 它使用新的java.nio.fileAPI。
• 它在InputStream中使用了新的TransferTo方法，这避免了必须声明一个字节数组作为缓冲区的rigamarole和一个while循环。

然后在下载端：

void download() {
  long size = bytesToLong(in.readNBytes(8));
  Path p = Paths.get("/file/to/store/data/at");
  // Generally network transfer sizes are a bit higher than 1k;
  // up to 64k in fact. Best to declare a larger buffer!
  byte[] bb = new byte[65536];
  try (OutputStream out = Files.newOutputStream(p)) {
    while (size > 0) {
      int count = in.read(bb);
      if (count == -1) throw new IOException("Unexpected end of stream");
      out.write(bb, 0, count);
      size -= count;
    }
  }
}

public static long bytesToLong(byte[] bytes) {
    ByteBuffer buffer = ByteBuffer.allocate(Long.BYTES);
    buffer.put(bytes);
    buffer.flip();//need flip 
    return buffer.getLong();
}

null

• 发送一个大小，以2字节的形式，按大字节顺序发送，不带符号。随后是许多字节，然后发送另一个大小，无限大。
• 当流完成时，将发送大小为0的大小（因此，值为02个字节），这表示文件已完成。

如果您需要的话，我将把它作为一个练习，让您来实现上行和下载端。