TCP实现文件上传

TCP实现文件上传

• //服务端
  public static void main(String[] args) throws Exception{
      // 1.创建服务
      ServerSocket serverSocket = new ServerSocket(9000);
      //2.监听客户端的
      Socket socket = serverSocket.accept();//阻塞式监听，会一直等待客户端连接
      // 3.获取输入流
      InputStream inputStream = socket.getInputStream();
      //4.文件输出
      FileOutputStream fileOutputStream = new FileOutputStream(new File("receive.jpg"));
      byte[] bytes = new byte[1024];
      int len;
      while ((len = inputStream.read(bytes)) != -1){
          fileOutputStream.write(bytes,0,len);
      }
      //通知客户端我接受完毕
      OutputStream outputStream = socket.getOutputStream();
      outputStream.write("接受完毕，可以断开".getBytes());
  
      //关闭资源
      fileOutputStream.close();
      inputStream.close();
      socket.close();
      serverSocket.close();
  }
  
  
  //客户端
      public static void main(String[] args) throws Exception {
          //1.创建一个Socket连接
          Socket socket = new Socket(InetAddress.getByName("127.0.0.1"), 9000);
          //2.创建一个输出流
          OutputStream outputStream = socket.getOutputStream();
          //3.读取文件
          FileInputStream fileInputStream = new FileInputStream(new File("2.jpg"));
          //4.写出文件
          byte[] bytes = new byte[1024];
          int len;
          while ((len = fileInputStream.read(bytes))!= -1){
              outputStream.write(bytes,0,len);
          }
          //通知服务器传输完毕
          socket.shutdownOutput();
          //确定服务器接受完毕，才能断开连接
          InputStream is = socket.getInputStream();
          ByteArrayOutputStream baos = new ByteArrayOutputStream();
          byte[] buffer2 = new byte[1024];
          int len2;
          while ((len2=is.read(buffer2))!=-1){
              baos.write(buffer2,0,len2);
          }
          System.out.println(baos.toString() );
  
          //5.关闭资源
          baos.close();
          is.close();
          fileInputStream.close();
          outputStream.close();
          socket.close();
      }