TCP-多线程并发上传图片

//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();
    }
}