Java实现TCP通信的文件上传

package com.zhanshen.demo_file_upload;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
/**
 * 客户端读取本地文件,上传到服务器,读取服务器回写的数据
 */
public class TCPClient {
    public static void main(String[] args) throws IOException {
        //1. 创建一个本地字节输入流fileInputStream, 构造方法中绑定要上传的文件
        FileInputStream fileInputStream = new FileInputStream("/home/zhanshen/IdeaProjects/Net/src/com/zhanshen/demo_file_upload/xv.png");
        //2. 创建客户端Socket对象,绑定要上传到服务器的IP地址和端口号
        Socket socket = new Socket("127.0.0.1", 8888);
        //3. 获取网络字节输出流OutputStream
        OutputStream outputStream = socket.getOutputStream();
        //4. 使用FIleInputStream 读取本地文件
        /**
         * 这一次我们的读取是循环读取,因为一个文件会很大,所以我们循环读取每次读取1024
         */
        int len = 0;
        byte[] bytes = new byte[1024];
        while( (len = fileInputStream.read(bytes) ) != -1) {
            //5. 上传到服务器
            outputStream.write(bytes, 0, len);
        }
        socket.shutdownOutput();  //当上传文件结束高速服务器结束了,防止阻塞
        //6. 获取InputStream
        InputStream inputStream = socket.getInputStream();
        //7. 读取服务器返回的数据
        while ( (len = inputStream.read(bytes) ) != -1) {
            System.out.println(new String(bytes, 0, len));
        }
        //8. 释放资源
        fileInputStream.close();
        socket.close();
    }
}

    package com.zhanshen.demo_file_upload;
    import java.io.*;
    import java.net.ServerSocket;
    import java.net.Socket;
    /**
     * 读取客户端上传的文件,把文件保存到本地  然后回写保存成功等提示消息
     */
    public class TCPServer {
        public static void main(String[] args) throws IOException {
            //1. 创建ServerSocket,指定端口号8888
            ServerSocket serverSocket = new ServerSocket(8888);
            //2. 使用ServerSocket的accept来获得客户端请求的socket对象
            Socket socket = serverSocket.accept();
            //3. 获取socket的输入流
            InputStream inputStream = socket.getInputStream();
            //4. 判断是否有要保存文件的文件夹,若没有就创建一个
            File file = new File("/home/zhanshen/IdeaProjects/Net/src/com/zhanshen/demo_file_upload/upload");
            if(!file.exists()) {   //如果这个文件夹不存在
                file.mkdirs();   //若没有,就创建他
            }
            //5. 创建本地输出流,写入到本地硬盘的位置
            FileOutputStream fileOutputStream = new FileOutputStream(file+"/1.png");
            //6. 使用InputStream 来读取客户端传过来的数据
            int len = 0;
            byte[] bytes = new byte[1024];
            while( (len = inputStream.read(bytes) ) != -1) {
                //7. 写入到硬盘
                fileOutputStream.write(bytes, 0, len);
            }
            //8. 获取socket的OutputStream来给客户端回显数据
            socket.getOutputStream().write("上传成功!".getBytes());
            //9. 释放资源
            fileOutputStream.close();
            socket.close();
            serverSocket.close();
        }
    }