Java基础-网络编程(TCP-上传图片)

/*

需求：上传图片

*/

/*

客户端：

1，服务端点

2，读取客户端已有的图片数据

3，通过socket输出流将数据发给服务端

4，读取服务端反馈信息

5，关闭

*/import java.io.*;
import java.net.*;

class PicClient
{
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;
}

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.0.103",10007);

FileInputStream fis = new FileInputStream(file);

OutputStream out = s.getOutputStream();

byte[] buf = new byte[1024];

int len = 0;

while((len=fis.read(buf))!=-1)
{
out.write(buf,0,len);
}

//告诉服务端数据已写完
s.shutdownOutput();

InputStream in = s.getInputStream();

byte[] bufIn = new byte[1024];

int num = in.read(bufIn);

System.out.println(new String(bufIn,0,num));

fis.close();
s.close();
}
}

/*
服务端

这个服务端有个局限性。当A客户端连接上以后。被服务端获取到。服务端执行具体流程
这时B客户端连接，只能等待。
因为服务端还没有处理完A客户端的请求，还有循环回来执行下次accept方法，所以
暂时获取不到B客户端对象。

那么为了可以让多个客户端同时并发访问服务端
那么服务端最好就是将每个客户端封装到一个单独的线程中，这样，就可以同时处理多个客户端请求。

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

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

public void run()
{
int count = 0;
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);
// FileOutputStream fos = new FileOutputStream("a-server.jpg");

byte[] buf = new byte[1024];

int len = 0;
while((len=in.read(buf))!=-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 PicServer
{
public static void main(String[] args) throws Exception
{
ServerSocket ss = new ServerSocket(10007);

while(true)
{
Socket s = ss.accept();

new Thread(new PicThread(s)).start();
}

// ss.close();
}
}