Java网络编程之TCP文件上传

服务器类-TcpServerDemo02

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

//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 inputStream=socket.getInputStream();
        //4.文件输出
        FileOutputStream fileOutputStream=new FileOutputStream(new File("receiveFile.png"));
        byte[] buffer=new byte[1024];
        int len;
        while ((len=inputStream.read(buffer))!=-1)
        {
            fileOutputStream.write(buffer,0,len);
        }

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

        //6.关闭资源
        fileOutputStream.close();
        inputStream.close();
        socket.close();
        serverSocket.close();
    }
}

服务器类-TcpClientDemo02

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

//TCP文件上传实现-客户端
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 os=socket.getOutputStream();
        //3.读取文件
        FileInputStream fileInputStream=new FileInputStream(new File("example.png"));
        //4.写出文件
        byte buffer[]=new byte[1024];
        int len;
        while ((len=fileInputStream.read(buffer))!=-1)
        {
            os.write(buffer,0,len);
        }

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

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

        //7.关闭资源
        baos.close();
        inputStream.close();
        fileInputStream.close();
        os.close();
        socket.close();
    }
}

运行结果

客户端：

服务器端：