TCP文件上传实现

服务器端

package com.sean.base.netStudy;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;

/**
 *
 * @create 2021-03-01 9:15
 */
public class TcpServerDemo02 {
    public static void main(String[] args) throws Exception{
        //1创建服务
        ServerSocket serverSocket = new ServerSocket(9000);
        //2监听客户端的链接
        Socket socket=serverSocket.accept();//阻塞式监听，会一直等待客户端连接
        //3获取输入流
        InputStream ips= socket.getInputStream();

        //4文件输出
        FileOutputStream fos = new FileOutputStream(new File("receive.png"));
        byte[] buffer = new byte[1024];
        int len;
        while ((len=ips.read(buffer))!=-1){
            fos.write(buffer,0,len);
        }

        //通知客户端我接收完毕了
        OutputStream ops2=socket.getOutputStream();
        ops2.write("我接收完毕了，你可以断开了".getBytes(StandardCharsets.UTF_8));

        //关闭资源
        fos.close();
        ips.close();
        socket.close();
        serverSocket.close();
    }
}

客户端

package com.sean.base.netStudy;

import java.io.*;
import java.net.InetAddress;
import java.net.Socket;

/**
 *
 * @create 2021-03-01 9:10
 */
public class TcpClientDemo02 {
    public static void main(String[] args) throws Exception{
        //1.创建一个Socket连接
        Socket socket = new Socket(InetAddress.getByName("127.0.0.1"),9000);
        //2.创建一个输出流
        OutputStream ops=socket.getOutputStream();;

        //3。读取文件
        FileInputStream fis=new FileInputStream(new File("question.png"));
        //写出文件
        byte[] buffer = new byte[1024];
        int len;
        while ((len=fis.read(buffer))!=-1){
            ops.write(buffer,0,len);
        }

        //通知服务器，我已经结束了
        socket.shutdownOutput();//我已经传输完了

        //确定服务器接收完毕，才能断开链接
        InputStream ips2=socket.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        byte[] buffer2 = new byte[1024];
        int len2;
        while ((len2=ips2.read(buffer2))!=-1){
            baos.write(buffer2,0,len2);
        }
        System.out.println(baos.toString());


        //5.关闭资源
        baos.close();
        ips2.close();
        fis.close();
        ops.close();
        socket.close();

    }
}