TCP文件上传

TCP文件上传

TCP文件上传主要通过socket和io流进行实现的，客户端将本地的文件通过网络流发送到服务器，服务器接收后保存到服务器本地，从而实现文件上传。
下图是文件上传的具体实现。要注意的是，在客户端上传完成之后，要通过socket调用shutdownOutput方法关闭输出流，告诉服务器文件已全部发送，否则程序会被卡住。

服务器

public static void main(String[] args) throws IOException {
        //创建一个端口号为9999的服务器
        ServerSocket serverSocket = new ServerSocket(9999);
        //阻塞式监听(只有收到客户端的请求才会往下执行),获取与客户端配对的socket
        Socket socket = serverSocket.accept();
        //获取输入流
        InputStream is = socket.getInputStream();
        //创建本地输出流，用于保存客户端发送的文件
        FileOutputStream fos = new FileOutputStream(new File("uploading.jpg"));
        byte[] bytes = new byte[1024];
        int len = 0;
        while ((len = is.read(bytes)) != -1) {
            //保存到本地
            fos.write(bytes, 0, len);
        }
        //获取输出流，向客户端发送消息
        OutputStream os = socket.getOutputStream();
        //发送消息
        os.write("服务器接收完毕".getBytes());
        //关闭资源
        fos.close();
        is.close();
        socket.close();
        serverSocket.close();
    }

客户端

 public static void main(String[] args) throws IOException {
        //连接ip为"127.0.0.1"端口号为9999的服务器
        Socket socket = new Socket("localhost", 9999);
        //获取输出流
        OutputStream os = socket.getOutputStream();
        //创建输入流读取本地图片
        FileInputStream fis = new FileInputStream(new File("1.jpg"));
        //一次读取1024个字节
        byte[] bytes = new byte[1024];
        int len = 0;
        while ((len = fis.read(bytes)) != -1) {
            os.write(bytes, 0, len);
        }
        //通知服务器上传完毕
        socket.shutdownOutput();
        //创建输入流读取服务器发送的文字
        InputStream is = socket.getInputStream();
        while ((len = is.read(bytes)) != -1) {
            //将字节转化为字符串并打印
            String str = new String(bytes, 0, len);
            System.out.println(str);
        }
        //关闭资源
        fis.close();
        os.close();
        socket.close();

    }