TCP实现文件上传

服务器端

public class TcpServerDemo02 {
    public static void main(String[] args) throws Exception{
        //1. 创建服务
        ServerSocket serverSocket = new ServerSocket(9000);
        //2. 监听客户端的连接
        Socket socket = serverSocket.accept();//阻塞式监听，会一直等待客户端连接

        //3. 获取输入流
        InputStream is = socket.getInputStream();

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

        //通知客户端我接收完毕了
        OutputStream os = socket.getOutputStream();
        os.write("我已经接收完毕了，可以关闭连接".getBytes());


        //5. 关闭资源
        os.close();
        fos.close();
        is.close();
        socket.close();
        serverSocket.close();

    }
}

客户端

public class TcpClientDemo02 {
    public static void main(String[] args) throws Exception{
        //1. 服务器地址,并创建连接
        Socket socket = new Socket(InetAddress.getByName("127.0.0.1"),9000);
        //2. 创建输出流
        OutputStream os = socket.getOutputStream();


        //3. 文件流
        FileInputStream fis = new FileInputStream(new File("timg.jpg"));

        //4. 写出
        byte[] buffer = new byte[1024];
        int len;
        while((len=fis.read(buffer))!=-1){
            os.write(buffer,0,len);
        }

        //我已经传输完了
        socket.shutdownOutput();

        //确定服务器接收完毕，才能断开连接
        InputStream is = socket.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] buffer2 = new byte[1024];
        while((len=is.read(buffer2))!=-1){
            baos.write(buffer2,0,len);
        }

        System.out.println(baos.toString());

        //5. 关闭资源
        baos.close();
        is.close();
        fis.close();
        os.close();
        socket.close();


    }
}