TCP实现文件上传

文件上传分析

    

 一、基本实现

1、服务端

public class FileUpload_Server {
    public static void main(String[] args) throws IOException {
        System.out.println("服务器 启动.....  ");
        // 1. 创建服务端ServerSocket
          ServerSocket serverSocket = new ServerSocket(8888);
          // 2. 建立连接 
        Socket accept = serverSocket.accept();
          // 3. 创建流对象
          // 3.1 获取输入流,读取文件数据
        BufferedInputStream bis = new BufferedInputStream(accept.getInputStream());
        // 3.2 创建输出流,保存到本地 .
        BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream("copy.jpg"));
        // 4. 读写数据
        byte[] b = new byte[1024 * 8];
        int len;
        while ((len = bis.read(b)) != -1) {
            bos.write(b, 0, len);
        }
        //5. 关闭 资源
        bos.close();
        bis.close();
        accept.close();
        System.out.println("文件上传已保存");
    }
}

2、客户端

public class FileUPload_Client {
    public static void main(String[] args) throws IOException {
        // 1.创建流对象
        // 1.1 创建输入流,读取本地文件  
        BufferedInputStream bis  = new BufferedInputStream(new FileInputStream("test.jpg"));
        // 1.2 创建输出流,写到服务端 
        Socket socket = new Socket("localhost", 8888);
        BufferedOutputStream   bos   = new BufferedOutputStream(socket.getOutputStream());

        //2.写出数据. 
        byte[] b  = new byte[1024 * 8 ];
        int len ; 
        while (( len  = bis.read(b))!=-1) {
            bos.write(b, 0, len);
            bos.flush();
        }
        System.out.println("文件发送完毕");
        // 3.释放资源

        bos.close(); 
        socket.close();
        bis.close(); 
        System.out.println("文件上传完毕 ");
    }
}

- 存在问题:

  服务端和客户端都会陷入阻塞状态,原因是客户端的read()方法引起的。

  客户端的本地输入流bis.read(b))一直阻塞,读取不到-1,其网络输出流也就输出不了-1;这样服务端的网络输入流也就读不到-1,进入阻塞,一直死循环等待结束标记。

- 解决办法:

  客户端上传完文件,给服务器写一个结束标记。

    /**
         * 实现从硬盘文件读取数据
         * 将数据写出到服务端
         */
        byte[] b = new byte[1024*8];
        int len = 0 ;
        while((len = fis.read(b)) != -1){   //读取到-1结束,但是while不会读取到-1,也不会把结束标记写到服务器。
            os.write(b);      //输出流
        }

        //-----------  解决read()阻塞 ------------------
        client.shutdownOutput();

二、文件上传优化分析

1、文件名称写死的问题

  服务端,保存文件的名称如果写死,那么最终导致服务器硬盘,只会保留一个文件,建议使用系统时间优化,保证文件名称唯一,代码如下:

/*
      自定义一个文件的命名规则:防止同名的文件被覆盖
      规则:域名+毫秒值+随机数
*/
       String fileName = System.currentTimeMillis()+new Random().nextInt(999999)+".jpg";

       //5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
       //FileOutputStream fos = new FileOutputStream(file+"\1.jpg");
       FileOutputStream fos = new FileOutputStream(file+"\"+fileName);

2、循环接收的问题

  服务端,指保存一个文件就关闭了,之后的用户无法再上传,这是不符合实际的,使用循环改进,可以不断的接收不同用户的文件,代码如下:

// 每次接收新的连接,创建一个Socket
whiletrue){
    Socket accept = serverSocket.accept();
    ......
//        server.close();     //不能关闭server
}

3、效率问题

  服务端,在接收大文件时,可能耗费几秒钟的时间,此时不能接收其他用户上传,所以,使用多线程技术优化,代码如下:

whiletrue){
    Socket accept = serverSocket.accept();
    // accept 交给子线程处理.
    new Thread(() -> {
          ......
        InputStream bis = accept.getInputStream();
          ......
    }).start();
}

4、服务端优化实现

import java.net.ServerSocket;
import java.net.Socket;
import java.util.Random;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

/**
    文件上传案例服务器端:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写"上传成功"

    明确:
        数据源:客户端上传的文件
        目的地:服务器的硬盘 
 */
public class FileServer {
    public static void main(String[] args) throws IOException {
        //1.创建一个服务器ServerSocket对象,和系统要指定的端口号
        ServerSocket server = new ServerSocket(8888);

        //2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象

        /*
            让服务器一直处于监听状态(死循环accept方法)
            有一个客户端上传文件,就保存一个文件
         */
        while(true){
            Socket socket = server.accept();
            FileOutputStream fos = null;
            /*
                使用多线程技术,提高程序的效率
                有一个客户端上传文件,就开启一个线程,完成文件的上传
             */
            new Thread(new Runnable() {
                //完成文件的上传
                @Override
                public void run() {
                    try {               //由于Runnable接口中run()方法没有throws异常,所以子类也不能
                        //3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                        InputStream is = socket.getInputStream();
                        //4.判断d:\upload文件夹是否存在,不存在则创建
                        File file =  new File("F:\IntelliJ IDEA 14.1.7\Project\HelloIdea\src\upload");
                        if(!file.exists()){
                            file.mkdirs();
                        }

                    /*
                        自定义一个文件的命名规则:防止同名的文件被覆盖
                        规则:域名+毫秒值+随机数
                     */
                        String fileName = "itcast"+System.currentTimeMillis()+new Random().nextInt(999999)+".jpg";

                        //5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                        //FileOutputStream fos = new FileOutputStream(file+"\1.jpg");
                        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());

                    }catch (IOException e){
                        System.out.println(e);
                    }finally {
                        //10.释放资源(FileOutputStream,Socket,ServerSocket)
                        try {
                            fos.close();
                            socket.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }
            }).start();
        }

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

5、客户端实现

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

/**
 * 文件上传案例的客户端:读取本地文件,上传到服务器,读取服务器回写的数据
    明确:
    数据源: 本地硬盘文件
    目的地:服务器
 */

public class FileClient {
    public static void main(String[] args) throws IOException {
        //1.创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源
        FileInputStream fis = new FileInputStream("F:\IntelliJ IDEA 14.1.7\Project\HelloIdea\src\img.jpg");
        //2.创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
        Socket socket = new Socket("127.0.0.1",8888);
        //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();
    }
}
原文地址:https://www.cnblogs.com/timetellu/p/11625924.html