Java 基于TCP Socket 实现文件上传

Z时代
2024-01-10
分类：综合

文件上传过程一个单向Socket通信过程。客户端通过文件输入流读取文件，然后从Socket获取输出流写入数据。服务端从Socket中获得输入流，然后写入文件输出流，写入数据完成则上传完成。

服务端UploadServer：


public class UplaodServer {
  public static void main(String []args){
    try(
      // 创建一个ServerSocket监听8080端口的请求
      // ServerSocket 实现了 AutoCloseable接口, 可以自动释放资源
    	ServerSocket server = new ServerSocket(8080);
      // accept()会阻塞当前线程, 等待客户端请求 
      Socket socket = server.accept();
      // 获取输入流
      BufferedInputStream in = new BufferedInputStream(socket.getInputStream());
      // 由文件输出流创建缓冲输出流, 写入本地upload.jpg 文件
      FileOutputStream outputStream = new FileOutputStream("./upload.jpg")
    ){
      // 缓冲区 
      byte [] buffer = new byte[1024];
      int len = in.read(buffer);
      while(len != -1){
        // 写入数据到文件
        outputStream.write(buffer, 0, len);
        // 再次从Socket中读取数据
        len = in.read(buffer);
      }
      System.out.println("文件已上传");
    }catch (IOException e) {
      e.printStackTrace();
    }
  }
}

客户端UploadClient：


public class UploadClient {
  public static void main(String []args){
    try(
    	// Socket 实现了 AutoCloseable接口 可自动关闭资源
      Socket socket = new Socket("127.0.0.1", 8080);
       // 又Socket获得输出流， 并创建缓冲输出流
      BufferedOutputStream out = new BufferedOutputStream(socket.getOutputStream());
      FileInputStream fileInput = new FileInputStream("F:\file\oxx6j5.jpg");
      // 由文件输入流创建缓冲输入流
      BufferedInputStream in = new BufferedInputStream(fileInputStream);
    ){
       // 准备缓冲区
      byte[] buffer = new byte[1024];
      int len = in.read(buffer);
      while (len != -1) {
        // 写入Socket
        out.write(buffer, 0, len);
        // 再次读取文件
        len = in.read(buffer);
      }
      System.out.println("上传成功！");
    }catch (ConnectException e) {
      System.out.println("服务器未启动");
      e.printStackTrace();
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
}