day24-02-网络编程(TCP-客户端并发上传图片)

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

class PicDemo2
{
    public static void main(String[] args) throws Exception
    {
//        传图片路径就可以，一个参数
        if(args.length!=1)
        {
            System.out.println("请选择一个jpg格式的图片");
            return;
        }
        File file = new File(args[0]);
//        判断文件不存在或者不是文件
        if (file.exists() && file.isFile())
        {
            System.out.println("该文件有问题，要么不存在，要么不是文件");
            return;
        }
//      jpg图片
        if(file.getName().endsWith(".jpg"))
        {
            System.out.println("图片格式错误，请重新选择");
            return;
        }
//      文件大小
        if (file.length()>1024*1024*5)
        {
            System.out.println("文件过大，没安好心");
            return;
        }
        
        Socket s = new Socket("192.168.1.13",1007);

        FileInputStream fis = new FileInputStream(file);

        OutputStream out = s.getOutputStream();

        byte[] buf = new byte[1024];

        int len = 0;

        while ((len=fis.read())!=-1)
        {
            out.write(buf,0,len);
        }
        //        告诉服务端数据已写完。
        s.shutdownOutput();
        InputStream in = s.getInputStream();
        byte[] b = new byte[1024];
        int len1 = in.read(b);
        System.out.println(new String(b,0,len1));

        fis.close();
        s.close();
    }
}
/*
服务端、

这个服务端有个局限性，当A客户端连接上以后，被服务端获取到，服务端执行具体流程。
这时B客户端连接，只有等待。
因为服务端还没处理完A客户端的请求，还有偶循环回来执行下次accpet方法。所以暂时获取不到B客户端对象。
那么为了可以让多个客户端同时并发访问服务端。
那么服务端最好就是将每个客户端封装到一个单独的线程中，这样就可以同时处理多个客户端请求。

如何定义线程呢？
只要明确了每一个客户端在服务端执行的代码即可。将该代码存入run方法当中。
 */
class PicThread implements Runnable
{

    private Socket s;
    PicThread(Socket s)
    {
        this.s =s;
    }

    @Override
    public void run()
    {
        int count = 1;
        String ip= s.getInetAddress().getHostAddress();
        try
        {
            System.out.println(ip+"...connected");

            InputStream in = s.getInputStream();

            File file = new File(ip+"("+(count)+")"+".jpg");
            while (file.exists())
            {
                file = new File(ip+"("+(count++)+")"+".jpg");
            }

            FileOutputStream fos = new FileOutputStream(file);

            byte[] buf = new byte[1024];
            int len = 0;
            while ((len = in.read()) != -1) {
                fos.write(buf, 0, len);
            }

            OutputStream out = s.getOutputStream();
            out.write("图片上传成功".getBytes());

            fos.close();
            s.close();
        }catch (Exception e)
        {
            throw new RuntimeException(ip+"上传失败");
        }
    }
}


class PicServer2
{
    public static void main(String[] args) throws Exception
    {
        ServerSocket ss = new ServerSocket(1007);
        while (true)
        {

            Socket s = ss.accept();//有阻塞式机制，不会死循环
//            客户端创建后，要建立线程
            new Thread(new PicThread(s)).start();


        }

    }
}