TCP实现文件上传

TCP实现文件上传

客户端代码

package TCstudy;
​
import java.io.*;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
​
public class TcpClient {
    public static void main(String[] args) throws Exception {
        //连接服务器的socket
        Socket socket =new Socket(InetAddress.getByName("127.0.0.1"),9000);
        //得到socket的流对象
        OutputStream os = socket.getOutputStream();
        //图片提前放到根目录下了，使用File可以省略路径，因为是传输文件，所以使用FileInputStream
        //文件名记得加后缀
        FileInputStream fis=new FileInputStream(new File("QQ图片20210622225717.jpg"));
        //fis读取文件数据到buffer里，再写入到socket的流对象os里，在服务器端可以通过socket的inputStream对象读取内容
        
        
        byte[] buffer=new byte[1024];
        int len=0;
        while((len=fis.read(buffer))!=-1) 
            os.write(buffer,0,len);
​
        //这一步必须要有，如果没有这一步，客户端上面os写入的数据会被下面的inputStream对象接收，相当于没有传输到服务器
        socket.shutdownOutput();
​
        //确认服务器已经接收
        InputStream inputStream = socket.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        //byte类用byte管道流
        byte[] buffer2=new byte[2014];
        int len2;
        while((len2=inputStream.read(buffer2))!=-1)
            baos.write(buffer2,0,len2);
        System.out.println(baos.toString());
​
        baos.close();
        inputStream.close();
        fis.close();
        os.close();
        socket.close();
    }
​
​
​
}
​

 

服务器端代码

package TCstudy;
​
import jdk.internal.util.xml.impl.Input;
​
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
​
public class TcpServer {
    public static void main(String[] args) throws Exception {
        //注册端口
        ServerSocket server=new ServerSocket(9000);
        //阻塞式监听，会一直等待客户端传输数据
         Socket socket=server.accept();
         //获取输入流
        InputStream is = socket.getInputStream();
        //文件输出
        FileOutputStream fos = new FileOutputStream(new File("recieve.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());
​
​
​
        //关闭资源
        fos.close();
        is.close();
        socket.close();
        server.close();
    }
}
​