TCP文件上传案例

客户端

package com.wybing.demo1.FileUpload;

/**
 * ClassName:    TCPClient
 * Package:    com.wybing.demo1.FileUpload
 * Description:  文件上传案例客户端
 * Datetime:    2021/7/5   10:51
 * Author:   wybing(wybingcom@126.com)
 */

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

/**
 * 文件上传案例的客户端:读取本地文件,上传到服务器,读取服务器回写的数据
 *
 * 明确:
 *      数据源:
 *      目的地:
 * 实现步骤:
 *      1.创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源
 *      2.创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
 *      3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
 *      4.使用本地字节输入流FileInputStream对象中的方法read,读取本地文件
 *      5.使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器
 *      6.使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象
 *      7.使用网络字节输入流InputStream对象中的方法read读取服务回写数据
 *      8.释放资源(FileInputStream, Socket)
 *
 */

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

        /**
         * 存在问题:阻塞
         * 解决办法:
         *          上传完文件,给服务器写一个结束标记
         *          void shutdownOutput() 禁用套接字的输出流
         *          对于TCP套接字,任何以前写入的数据都将被发送,并且后跟TCP的正常连接终止序列
         */
        socket.shutdownOutput();

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

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

服务端

package com.wybing.demo1.FileUpload;

/**
 * ClassName:    TCPServer
 * Package:    com.wybing.demo1.FileUpload
 * Description:  文件上传案例-服务器端
 * Datetime:    2021/7/5   11:09
 * Author:   wybing(wybingcom@126.com)
 */

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

/**
 * 文件上传案例-服务器端:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写”上传成功“
 * 明确:
 *      数据源:
 *      目的地:
 *
 * 实现步骤:
 *      1.创建一个服务器ServerSocket对象,和系统要指定的端口号
 *      2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
 *      3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
 *      4.判断 E:\\upload文件夹是否存在,不存在则创建
 *      5.创建一个本地字节输出流FileInputStream对象,构造方法中绑定要输出的目的地
 *      6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
 *      7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
 *      8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
 *      9.使用网络字节输出流OutputStream对象中的write,给客户端回写”上传成功“
 *      10.释放资源(FileOutputStream, Socket, ServerSocket)
 *
 */

public class TCPServer  {
    public static void main(String[] args) throws IOException {
        // 1.创建一个服务器ServerSocket对象,和系统要指定的端口号
        ServerSocket server = new ServerSocket(8887);

        /**
         * 让服务器一直处于监听状态(死循环accept方法)
         * 有一个客户端上传文件,就保存一个文件
         *
         */

        while(true){
                // 2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
                Socket socket = server.accept();

                /**
                 * 使用多线程技术,提高程序的效率
                 * 有一个客户端上传文件,就开启一个线程,完成文件上传
                 */

                 new Thread(new Runnable() {
                     @Override
                     public void run() {
                        try{

                            // 3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                            InputStream is = socket.getInputStream();
                            // 4.判断 E:\\upload文件夹是否存在,不存在则创建
                            File file = new File("E:\\temp\\upload");
                            if(!file.exists()){
                                file.mkdir();
                            }

                            /**
                             * 自定义一个文件的命令规则:防止同名文件被覆盖
                             * 规则:域名+毫秒值+随机数
                             *
                             */

                            String fileName = "wybing" + System.currentTimeMillis() + new Random().nextInt(99999) + ".png";

                            // 5.创建一个本地字节输出流FileInputStream对象,构造方法中绑定要输出的目的地
                            FileOutputStream fos = new FileOutputStream(file + "\\" + fileName);
                            // 6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
                            int len = 0;
                            byte[] bytes = new byte[1024];
                            while((len = is.read(bytes)) != -1){
                                // 7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
                                fos.write(bytes, 0, len);
                            }

                            // 8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
                            // 9.使用网络字节输出流OutputStream对象中的write,给客户端回写”上传成功“
                            socket.getOutputStream().write("上传成功" .getBytes());
                            // 10.释放资源(FileOutputStream, Socket, ServerSocket)
                            fos.close();
                            socket.close();

                        }catch (IOException e){
                            System.out.println(e);
                        }
                     }
                 }).start();

        }

        // 服务器不用关闭
        // server.close();
    }
}

测试类

package com.wybing.demo1.FileUpload;

import java.io.IOException;

/**
 * ClassName:    Test
 * Package:    com.wybing.demo1.FileUpload
 * Description:  测试类,调用TCPClient 的main ()方法
 * Datetime:    2021/7/5   12:08
 * Author:   wybing(wybingcom@126.com)
 */

public class Test {
    public static void main(String[] args) throws IOException {
        long before = System.currentTimeMillis();

        for (int i = 0; i < 5000; i++) {
            TCPClient.main(args);
        }

        long now = System.currentTimeMillis();

        System.out.println(now-before);
    }
}

posted @ 2022-01-11 11:49  乌药ice  阅读(43)  评论(0编辑  收藏  举报