TCP-多线程并发上传图片

来源:互联网 发布:装修全包 半包 知乎 编辑:程序博客网 时间:2024/06/06 11:49
//TCP-多线程并发上传图片import java.io.*;import java.net.*;/*思路客户端1,服务端点.2,读取客户端已有的图片数据.3,通过socket 输出流将数据发给服务端.4,读取服务端反馈信息.5,关闭*/class  PicClient{    public static void main(String[] args) throws Exception    {           if (args.length!=1)//意思是args只能传1个参数,不是1个参数都是错的        {            System.out.println("请选择一个JPG格式的图片");            return;        }        File file = new File(args[0]);//通过主函数传值的形式为file指定路径        if (!(file.exist() && file.isFile()))        {            System.out.out.println("该对象文件不存在,或者不是文件");            return;        }        if (!file.getName().endsWith(".jpg"))//文件名称末尾.jpg        {            System.out.out.println("该文件不是jpg文件,请重新选择jpg文件");        }        if (file.length()>1024*1024*5)//文件超过5M        {            System.out.out.println("文件超过5M,拒绝上传");        }        Socket s = new Socket("192.168.1.254",10007);        FileInputStream fis = new FileInputStream(file);//文件字节流        OutputStream out = s.getOutputStream();        byte[] buf = new byte[1024];//字节数组缓冲区        int len = 0;//长度        while ((len=fis.read(buf))!=-1)//读取文件字节流fis并写入buf,-1为结束        {            out.write(buf,0,len);        }               s.shutdoenOutput();//结束标记 发送给服务端 表示数据已写完        InputStream in = s.getInputStream();//读取服务端信息        byte[] bufIn = new byte[1024];//字节数组缓冲区        int num = in.read(bufIn);//长度        System.out.println(new String(bufIn,0,num));//new String(byte[],offset,length)转化字节数组byte[]为String        fis.close();//关io流        s.close();//关Socket    }}/*服务端这个服务端有个局限性.当A客户端连接上以后.被服务端获取到.服务端执行具体流程.这时B客户端连接,只有等待.因为服务端还没有处理完A客户端的请求,还没有循环回来执行下次accept方法.所以暂时获取不到B客户端对象.那么为了可以让多个客户端同时并发访问服务端.那么服务端最好就是将每个客户端封装到一个单独的线程中,这样,就可以同时处理多个客户端请求.这就是多线程实现.如何定义线程呢?只要明确了每一个客户端要在服务端执行的代码即可,将代码存入run方法*/class PicThread implements Runnable{    private Socket s;//因为要封装客户端Socket s,所以必须先定义成员变量    PicThread(Socket s)//构造器    {        this.s = s;    }    public void run()    {        int count = 1;//定义一个计数器count,命名文件用        String ip = s.getInetAddress().getHostAddress();//IP放try外面,catch里面才能输出IP        try        {                       System.out.println(ip+"...connected");            InputStream in = s.getInputStream();//读取Socket流            File file = new File(ip++"("+(count)+")"+".jpg");//防止上传的文件覆盖掉,写个命名规则,封装成File对象file            while (file.exists())//如果file存在,exists()方法 判断file对象是否存在//先封装count为(1),再判断(1)是否存在,是,则count++                file = new File(ip++"("+(count++)+")"+".jpg");//防止上传的文件覆盖掉,写个命名规则,封装成File对象file            FileOutputStream fos = new FileOutputStream(file);//写文件            byte[] buf = new tyte[1024];//缓冲区            int len = 0;            while ((len=in.read(buf))!=-1)//写缓冲区            {                fos.write(buf,0,len);//写文件            }                   //s.shutdownOutput();//s调用shutdownOutput() //告诉服务端数据已写完,不然会阻塞accept()            OutputStream out = s.getOutputStream();//给Socket写返回信息            out.write("上传成功!".getByte());//getByte()方法,String转成字节数组byte[]来发送            fos.close();            s.close();        }        catch (Exception e)        {            throw new RuntimeException(ip+"上传失败");//IP放外面,catch里面才能输出IP        }    }}class  PicServer{    public static void main(String[] args) throws Exception    {        ServerSocket ss = new ServerSocket(10007);        while (true)        {               Socket s = ss.accept();//拿Socket对象        new Thread(new PicThread(s)).start();//开启线程 //Thread()参数里面放要开的线程PicThread        //构造器里面参数放(s),是因为getInputStream()要用到socket s        }        //ss.close();    }}
0 0