Java TCP 实现文件传输

Java TCP 实现文件传输

环境

CentOS Linux release 8.1.1911 (Core)
java version "14.0.2" 2020-07-14

实现代码

ReceiveFileServer

其中，<port>为接收文件进程使用的端口号

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
public class ReceiveFileServer {private static final int PORT = <port>;private static final int BUFFER_SIZE = 1000;public static void main(String[] args) {try {while (true) {ServerSocket serverSocket = new ServerSocket(PORT);Socket socket = serverSocket.accept();InputStream inputStream = socket.getInputStream();BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(socket.getInputStream(), "utf-8"));String fileName = bufferedReader.readLine().toString();FileOutputStream fileOutputStream = new FileOutputStream(fileName);socket = serverSocket.accept();inputStream = socket.getInputStream();byte[] buffer = new byte[BUFFER_SIZE];int length = 0;while ((length = inputStream.read(buffer)) != -1) {fileOutputStream.write(buffer, 0, length);}fileOutputStream.flush();System.out.println("Receive file [" + fileName + "]");socket.close();serverSocket.close();fileOutputStream.close();}} catch(IOException e) {e.printStackTrace();}}
}

SendFileClient

其中<hostname>为服务器IP地址，<port>为接收文件进程使用的端口号，与上述服务器端代码中的端口号一致

sendMessage方法中的message变量为在服务器中保存的路径

package com.demo.filetransmission;import java.io.*;
import java.net.Socket;
import java.util.Arrays;public class SendFileClient {private static final String IP = "<hostname>";private static final int PORT = <port>;public void sendMessage(String message) {Socket socket = null;OutputStream outputStream = null;try {socket = new Socket(IP, PORT);outputStream = socket.getOutputStream();DataOutputStream dataOutputStream = new DataOutputStream(outputStream);System.out.println("Sending...");System.out.println(message);dataOutputStream.write(message.getBytes("utf-8"));dataOutputStream.flush();System.out.println("Send Successfully!");socket.shutdownOutput();} catch (IOException e) {e.printStackTrace();} catch (UnknownError e) {e.printStackTrace();} finally {try {socket.close();outputStream.close();} catch (IOException e) {e.printStackTrace();}}}public void uploadFile(String fileName) {Socket socket = null;FileInputStream fileInputStream = null;OutputStream outputStream = null;OutputStream fileNameOutputStream = null;DataOutputStream dataOutputStream;try {socket = new Socket(IP, PORT);fileInputStream = new FileInputStream(fileName);outputStream = socket.getOutputStream();byte[] buffer = new byte[1024];int length = 0;while ((length = fileInputStream.read(buffer)) != -1) {outputStream.write(buffer, 0, length);}socket.shutdownOutput();} catch (IOException e) {e.printStackTrace();} catch (UnknownError e) {e.printStackTrace();} finally {try {fileInputStream.close();socket.close();outputStream.close();} catch (IOException e) {e.printStackTrace();}}}public static void main(String[] args) {SendFileClient client = new SendFileClient();client.sendMessage("./res/file.mp4");client.uploadFile("./res/file.mp4");}
}

最后

  • 由于博主水平有限，不免有疏漏之处，欢迎读者随时批评指正，以免造成不必要的误解！