TCP上传图片
继续做一个TCP的相关案例,在客户端上传一张图片到服务端,服务端收到图片后,给客户端回馈上传成功信息。
实现细节:
- 对于一些文件的读取,若文件偏大,可以始用缓冲区去读取和写入,可以提高效率
- 服务端面向的是很多用户,所以必须要开线程来接收
- 如果同一个用户上传多张图片则有可能文件名重复,所以要对文件名做处理,具体在代码处有注释
- 客户端图片上传完成后需要给服务端发送结束信息,s.shutdownOutput();
就像是写入了一个终止符一样
客户端
需要在上传结束的时候写入终止符告诉服务端上传结束,可以双方约定一个终止的符号,也可以用shutdownOutput()方法来传递结束信息
import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.Socket;
public class UploadPicClient {
public static void main(String[] args) throws IOException {
System.out.println("上传图片客户端启动...");
//1、创建socket
Socket s = new Socket("127.233.24.3", 10018);
//2、读取源图片
File picFile = new File("D:\\forio\\download.jpg");
BufferedInputStream fis = new BufferedInputStream(new FileInputStream(picFile));
//3、目的:socket输出流,发送到服务端
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();
//读取上传成功字样
BufferedReader in = new BufferedReader(new InputStreamReader(s.getInputStream()));
String info = in.readLine();
System.out.println(info);
//关闭
fis.close();
s.close();
}
}
服务端
服务端需要多线程接收多个客户的上传信息,对于有可能存在的文件名重复,添加(1)(2)(3)...加以区分
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.net.ServerSocket;
import java.net.Socket;
public class UploadPicServer {
public static void main(String[] args) throws IOException {
System.out.println("上传图片服务端启动...");
// 创建ServerSocket
ServerSocket ss = new ServerSocket(10018);
while (true) {
// 获取客户端
Socket s = ss.accept();
// 实现多个客户端并发上传,服务器端必须启动做个线程来完成
new Thread(new UploadPic(s)).start();
}
}
}
class UploadPic implements Runnable {
private Socket s;
public UploadPic(Socket s) {
super();
this.s = s;
}
@Override
public void run() {
try {
String ip = s.getInetAddress().getHostAddress();
System.out.println(ip + "connected...");
// 写图片数据到文件的输出流
File dir = new File("D:\\forio\\photo");
// 为了避免重名覆盖,通过给重名文件进行编号
int count = 1;
File picFile = new File(dir, ip + "(" + count + ")" + ".jpg");
while (picFile.exists()) {
count++;
picFile = new File(dir, ip + "(" + count + ")" + ".jpg");
}
FileOutputStream fos = new FileOutputStream(picFile);
// 读取客户端发来的数据
InputStream in = s.getInputStream();
byte[] buf = new byte[1024];
int len = 0;
while ((len = in.read(buf)) != -1) {
fos.write(buf, 0, len);
}
// 给客户端一个回馈信息
BufferedWriter out = new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
out.write("上传成功!"); // 写完需要刷新一下,或者关闭流
// 关闭资源
out.close();
fos.close();
s.close();
} catch (Exception e) {
}
}
}