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