Java DataInputStream长度

Question

我正在为学校作业创建文件服务器应用程序。我目前拥有的是一个简单的Client类，它通过TCP发送一个图像，一个Server类接收它并将其写入文件。

这是我的客户端代码

import java.io.*; 
import java.net.*; 

class Client { 
    public static void main(String args[]) throws Exception { 
     long start = System.currentTimeMillis(); 
     Socket clientSocket = new Socket("127.0.0.1", 6789); 
     DataOutputStream outToServer = new DataOutputStream(clientSocket.getOutputStream()); 

     File file = new File("hot.jpg"); 
     FileInputStream fin = new FileInputStream(file); 
     byte sendData[] = new byte[(int)file.length()]; 
     fin.read(sendData); 

     outToServer.write(sendData, 0, sendData.length); 
     clientSocket.close(); 

     long end = System.currentTimeMillis(); 
     System.out.println("Took " + (end - start) + "ms"); 
    } 
} 

，这是我的服务器代码。

import java.io.*; 
import java.net.*; 

class Server { 
    public static void main(String args[]) throws Exception { 
     ServerSocket serverSocket = new ServerSocket(6789); 
     Socket connectionSocket = serverSocket.accept(); 
     DataInputStream dis = new DataInputStream(connectionSocket.getInputStream()); 

     byte[] receivedData = new byte[61500]; // <- THIS NUMBER 

     for(int i = 0; i < receivedData.length; i++) 
      receivedData[i] = dis.readByte(); 

     connectionSocket.close(); 
     serverSocket.close(); 

     FileOutputStream fos = new FileOutputStream("received.jpg"); 
     fos.write(receivedData); 
     fos.close(); 
    } 
} 

我的问题是如何获取正在发送的文件的大小。如果您检查Server的代码，您会看到我现在已经对数字进行了硬编码，即61500。我如何动态检索这个号码？

或者，我这样做是错误的方式？什么是替代解决方案？

Answer 1

在发送文件之前添加一个“长度字段”。 （注意，因为你读的文件到内存中的文件的最大尺寸可以〜2GB）

---

发送文件写入文件的长度之前：

outToServer.writeInt(sendData.length); 

而且接收读出的第一长度，并用它作为一个长度时：

int dataLength = dis.readInt() 
    byte[] receivedData = new byte[dataLength]; 

---

更好的方法是而不是先将文件读入内存，但直接从FileInputStream传输 - 那么您可以传输更大的文件！

Answer 2

如果知道长度，使用readFully（）比一次读取一个字节效率更高。

在这种情况下，您不需要知道长度，您可以编写循环来读取/写入尽可能多的数据。

InputStream is = connectionSocket.getInputStream(); 

byte[] bytes = new byte[8192]; 
int len; 
while((len = is.read(bytes)) > 0) 
    fos.write(bytes, 0, len); 

可避免当您阅读它复制数据读取整个文件到内存中。

FileInputStream fis = new FileInputStream(filename); 
OutputStream os = socket.getOutputStream(); 

byte[] bytes = new byte[8192]; 
int len; 
while((len = fis.read(bytes)) > 0) 
    os.write(bytes, 0, len); 

您可以使用Apache IOUtils .copy（），如果要执行从一个流复制到另一个。

此方法的优点是文件可以是任何大小（大于2 GB）。使用数组限制为2 GB（并使用更多内存）