2017-03-31 152 views
1

TL; DR:如何發送(使用單個連接)文件,文件的大小和名稱。互聯網上的所有例子都單獨發送一個文件。通過套接字發送文件 - 緩衝區大小

服務器:

public class Server { 
    private static int PORT = 6667; 
    private ServerSocket serverSocket; 

    public void run() throws IOException { 
     System.out.println("Opening server"); 
     serverSocket = new ServerSocket(PORT); 

     while(true) { 
      try(Socket incomingSocket = serverSocket.accept()) { 
       System.out.println("Accepted connection: " + incomingSocket); 
       incomingSocket.setSoTimeout(2000); // Don't let scanner block the thread. 

       InputStream inputStream = incomingSocket.getInputStream(); 
       Scanner scanner = new Scanner(inputStream); 

       String command = ""; 
       if(scanner.hasNextLine()) 
        command = scanner.nextLine(); 

       if(command.equals("update")) { 
        File file = new File("abc.txt"); 
        sendFile(incomingSocket, file); 
       } 
       else { 
        // ... 
        System.out.println("Another command"); 
       } 
      } 
     } 
    } 

    private void sendFile(Socket socket, File file) throws IOException { 
     byte[] bytes = new byte[(int)file.length()]; 
     FileInputStream fileInputStream = new FileInputStream(file); 
     BufferedInputStream bufferedInputStream = new BufferedInputStream(fileInputStream); 
     bufferedInputStream.read(bytes, 0, bytes.length); 
     OutputStream outputStream = socket.getOutputStream(); 

     PrintWriter writer = new PrintWriter(outputStream, true); 
     writer.println(file.length()); 
     writer.println(file.getName()); 

     System.out.println("Sending " + file.getName() + "(" + bytes.length + " bytes) to " + socket); 
     outputStream.write(bytes, 0, bytes.length); 
     outputStream.flush(); 
     System.out.println("File sent"); 
    } 

    public void stopRunning() { 
     try { 
      serverSocket.close(); 
     } catch(IOException e) { 
      e.printStackTrace(); 
     } 
    } 
} 

客戶:

public class Client { 
    private static String HOST = "localhost"; 
    private static int PORT = 6667; 

    public void run() throws IOException { 
     Socket socket = new Socket(HOST, PORT); 
     System.out.println("Connecting..."); 

     OutputStream outputStream = socket.getOutputStream(); 
     PrintWriter writer = new PrintWriter(outputStream, true); 
     writer.println("update");  // Example command which will determine what server sends back 

     receiveFile(socket); 

     socket.close(); 
    } 

    private void receiveFile(Socket socket) throws IOException { 
     InputStream inputStream = socket.getInputStream(); 

     int size = 16384; 
     String name = "example.txt"; 

     Scanner scanner = new Scanner(inputStream); 
     size = Integer.parseInt(scanner.next()); 
     name = scanner.next(); 

     FileOutputStream fileOutputStream = new FileOutputStream(name); 
     BufferedOutputStream bufferedOutputStream = new BufferedOutputStream(fileOutputStream); 

     byte[] buffer = new byte[size]; 
     int bytesRead, totalRead = 0; 
     while ((bytesRead = inputStream.read(buffer, 0, buffer.length)) != -1) { 
      totalRead += bytesRead; 
      bufferedOutputStream.write(buffer, 0, bytesRead); 
     } 
     bufferedOutputStream.flush(); 

     System.out.println("File " + name + " received. " + totalRead + " bytes read"); 

     bufferedOutputStream.close(); 
     fileOutputStream.close(); 
    } 

我想我的服務器將文件發送到客戶端。它還應該包含文件的名稱和大小。名稱,因爲它非常重要,因爲我不想製作大尺寸的硬編碼緩衝區。

試用上面的代碼。客戶端的「掃描儀部分」

Scanner scanner = new Scanner(inputStream); 
size = Integer.parseInt(scanner.next()); 
name = scanner.next(); 

工作正常,但文件沒有收到。 inputStream.read(buffer,0,buffer.length)從不讀取流中的剩餘字節。

如果我註釋掉掃描儀部分,字節被正確讀取(大小和名稱信息+文件本身)

所以,問題是,我怎麼用一個連接發送它?或者我應該做兩個單獨的連接,第一個請求大小和文件名,並在第二個文件中發送文件?

回答

0

Scanner適合基於文本的工作。

做你想做的事情的一種方法是使用DataInputStreamDataOutputStream。只需要一個連接:

public void send(File file, OutputStream os) throws IOException { 
    DataOutputStream dos = new DataOutputStream(os); 

    // writing name 
    dos.writeUTF(file.getName()); 
    // writing length 
    dos.writeLong(file.length()); 

    // writing file content 
    ... your write loop, write to dos 

    dos.flush(); 
} 

public void receive(InputStream is) throws IOException { 
    DataInputStream dis = new DataInputStream(is); 

    String fileName = dis.readUTF(); 
    long fileSize = dis.readLong(); 

    // reading file content 
    ... your read loop, read from dis 
} 
+0

太棒了,非常感謝。我需要閱讀一些關於正確處理流的信息:) – MrGrzyboo