文件上传案例阻塞问题和文件上传案例优化

文件上传案例阻塞问题

 

 

 解决：

void shutdownOutput() 禁用此套接字的输出流

对于TCP套接字,任何以前写入的数据都将被发送, 并且后跟TCP的正常连接终止序列。

 

文件上传案例优化

/*
文件上传客户端
 */
public class TCPFileClient {
    public static void main(String[] args) throws IOException {
        //创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源
        FileInputStream fis = new FileInputStream("F:\\1.jpg");
        //创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
        Socket socket = new Socket("127.0.0.1", 8888);
        //使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
        OutputStream os = socket.getOutputStream();
        //使用本地字节输入流FileInputStream对象中的方法read,读取本地文件
        int len = 0;
        byte[] bytes = new byte[1024];
        while ((len=fis.read(bytes))!=-1){
            os.write(bytes,0,len);
        }
        socket.shutdownOutput();
        //使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器
        InputStream is = socket.getInputStream();
        //使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象
        while ((len=is.read(bytes))!=-1){
            System.out.println(new String(bytes,0,len));
        }
        //使用网络字节输入流InputStream对象中的方法read读取服务回写的数据
        fis.close();
        socket.close();
    }
}

可以使用多线程和循环让服务器一直监听

fileName 自定义名称

/*
文件上传服务器端
 */
public class TCPFileServer {
    public static void main(String[] args) throws IOException {
        //创建一个服务器ServerSocket对象，和系统要指定的端口号
        ServerSocket server = new ServerSocket(8888);

        while (true){
            //使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
            Socket socket = server.accept();
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        //使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                        InputStream is = socket.getInputStream();
                        //判断文件夹是否存在,不存在则创建
                        File file = new File("F:\\a\\upload");
                        if (!file.exists()){
                            file.mkdirs();
                        }

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

                        //创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                        //FileOutputStream fos = new FileOutputStream(file + "\\1.jpg");
                        FileOutputStream fos = new FileOutputStream(file + "\\"+fileName);
                        //使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
                        int len = 0;
                        byte[] bytes = new byte[1024];
                        while ((len=is.read(bytes))!=-1){
                            //使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
                            fos.write(bytes,0,len);
                        }
                        //使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
                        //使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
                        socket.getOutputStream().write("上传成功".getBytes());
                        fos.close();
                        socket.close();
                    }catch (IOException e){
                        System.out.println(e);
                    }
                }
            }).start();
        }
//        server.close();
    }
}

 

 

 

搜索

复制