TCP通信的文件上传案例

TCP通信的文件上传案例

原理：客户端读取本地的文件，把文件上传到服务器，服务器再把上传的文件保存到服务器硬盘上

1.客户端使用 本地字节输入流 ， 读取要上传的内容
2.客户端使用 网络字节输出流 ，把读取到的文件上传到服务器
3.服务器使用 网络字节输入流 ， 读取客户端上传的文件
4.服务器使用 本地字节输出流 ， 把读取的文件保存到服务器硬盘上面去
5.服务器使用 网络字节输出流 ， 给客户端回写一个 “上传成功”
6.客户端使用 网络字节输入流 ，读取服务器回写的数据
7.释放资源

注意：
	客户端和服务器和本地硬盘进行读写，需要使用自己创建的字节流对象（本地流） 
	客户端和服务器之间进行读写，必须使用Socket中提供的字节流对象（网络流）

客户端代码：

/**

/**
 * 1.创建一个本地字节输入流FileInputStream对象，构造方法中绑定要读取的数据源
 * 2.创建一个客户端Socket对象，构造方法中绑定服务器的IP地址和端口号
 * 3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
 * 4.使用本地字节输入流FileInputStream对象中的方法read , 读取本地文件
 * 5.使用网络字节输出流OutputSteam对象中的方法write，吧读取的文件上传到服务器
 * 6.使用Socket中的方法getInputStream ，获取网络字节输入流InputStream对象
 * 7.使用网络字节输入流InputStream对象中的方法read读取服务回写的数据
 * 8.释放资源（FileInputStream , Socket）
 */
public class TCPSocketFile {


    public static void main(String[] args) throws IOException {
        //1.创建一个本地字节输入流FileInputStream对象，构造方法中绑定要读取的数据源
        FileInputStream fis = new FileInputStream("E:\\OtherSoft\\bg\\4.JPG");
        //2.创建一个客户端Socket对象，构造方法中绑定服务器的IP地址和端口号
        Socket sc = new Socket("127.0.0.1",8888);
        //3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
        OutputStream os = sc.getOutputStream();
        byte[] bytes = new byte[1024];
        int len = 0;
        //4.使用本地字节输入流FileInputStream对象中的方法read , 读取本地文件
        while ((len = fis.read(bytes)) != -1){
            //5.使用网络字节输出流OutputSteam对象中的方法write，吧读取的文件上传到服务器
            os.write(bytes,0,len);
        }

        /**
         * 注意 ： 这里会发生阻塞的问题
         *          解决：
         *              上传完文件，给服务器一个结束的标记
         *              void shutdownOutput() 禁止此套接字的输出流
         *              对于 TCP 套接字 ， 任何以前写入的数据都将被发送 ， 并且后跟 TCP 的正常链接终止序列。
         *
         */
        sc.shutdownOutput();

        //6.使用Socket中的方法getInputStream ，获取网络字节输入流InputStream对象
        InputStream is = sc.getInputStream();
        //7.使用网络字节输入流InputStream对象中的方法read读取服务回写的数据
        while ((len = is.read(bytes)) != -1){
            System.out.println(new String(bytes,0,len));
        }

        // 8.释放资源（FileInputStream , Socket）
        fis.close();
        sc.close();
    }

服务器端代码:

/**
 *
 * 1.创建一个服务器ServerSocket对象，和系统要制定的端口号
 * 2.使用ServerSocket对象中的方法accept,获取请求的客户端Socket对象
 * 3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
 * 4.判断 磁盘路径 是否存在，不存在则创建
 * 5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
 * 6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
 * 7.使用本地字节输出流FileOutputStream对象中的方法write,把读取的文件保存到雾服务器的硬盘上
 * 8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
 * 9.使用网络字节输出流OutputStream对象中的write方法，给客户端回写“上传成功”
 * 10.释放资源
 */
public class TCPServer {


    public static void main(String[] args) throws IOException {


        //1.创建一个服务器ServerSocket对象，和系统要制定的端口号
        ServerSocket server = new ServerSocket(8888);
        //2.使用ServerSocket对象中的方法accept,获取请求的客户端Socket对象
        Socket sc = server.accept();
        // 3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
        InputStream is = sc.getInputStream();
        // 4.判断 磁盘路径 是否存在，不存在则创建
        File file = new File("E:\\abc");
        if(!file.exists()){
            file.mkdir();
        }
        // 5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
        FileOutputStream fos = new FileOutputStream(file+"\\4.jpg");

        byte[] bytes = new byte[1024];
        int len = 0;
        //6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
        while ((len = is.read(bytes)) != -1){
            //7.使用本地字节输出流FileOutputStream对象中的方法write,把读取的文件保存到雾服务器的硬盘上
            fos.write(bytes,0,len);
        }
        //8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
        OutputStream os = sc.getOutputStream();
        //9.使用网络字节输出流OutputStream对象中的write方法，给客户端回写“上传成功”
        os.write("上传成功".getBytes());
        //10.释放资源
        fos.close();
        server.close();
        sc.close();
    }

注意：还是存在很多问题，比如说，文件上传的效率问题，文件命名规则问题，服务器一次性使用，程序阻塞等等问题，所以我们要对代码进行优化

---

客户端代码优化的地方：

//4.使用本地字节输入流FileInputStream对象中的方法read , 读取本地文件
while ((len = fis.read(bytes)) != -1){
//5.使用网络字节输出流OutputSteam对象中的方法write，吧读取的文件上传到服务器
os.write(bytes,0,len);
}
/**
* 注意 ： 这里会发生阻塞的问题
* 解决：
* 上传完文件，给服务器一个结束的标记
* void shutdownOutput() 禁止此套接字的输出流
* 对于 TCP 套接字 ， 任何以前写入的数据都将被发送 ， 并且后跟 TCP 的正常链接终止序列。
*
*/
sc.shutdownOutput();
//6.使用Socket中的方法getInputStream ，获取网络字节输入流InputStream对象
InputStream is = sc.getInputStream();

服务器优化的地方：

/**
* 优化2 ：
* 服务器一直处于启动状态 所以这里给个死循环
/
while(true){
/*
* 优化3：
* 使用多线程技术，提高上传文件效率
*/
//每上传一个文件就创建一个线程
Socket sc = server.accept();

        new Thread(new Runnable() {
            @Override
            public void run() {
                //注意捕获异常
                try {
                    File file = new File("E:\\abc");
                    if(!file.exists()){
                        file.mkdir();
                    }

                    /**
                     * 优化1：
                     * 自定义命名规则
                     *      域名+毫秒值+随机数+后缀名
                     */

                    String filename = "itcast"+System.currentTimeMillis()+ new Random(9999999) + ".jpg";

                    //创建本地输出流对象
                    FileOutputStream fos = new FileOutputStream(file+"\\"+filename);
                    //获取Socket对象\
                    //使用网络输入流
                    InputStream is = sc.getInputStream();
                    byte[] bytes = new byte[1024];
                    int len = 0;
                    while ((len = is.read(bytes)) != -1){
                        //获取到的 网络传过来的
                        fos.write(bytes,0,len);
                    }

                    OutputStream os = sc.getOutputStream();
                    os.write("上传成功".getBytes());
                    fos.close();
                    sc.close();
                }catch (IOException e){
                    e.printStackTrace();
                }
            }
        }).start();

    }

好了，这里就是TCP通信的文件上传的简单案例，希望能够帮助到大家