Demo_03.java
package TCP;
import java.io.*;
import java.net.*;
public class Demo_03 {
public static void main(String[] args) throws Exception {
ServerSocket serverSocket = new ServerSocket(10001);
while (true) {
Socket socket = serverSocket.accept();
new Thread(new ServerThread(socket)).start();
}
}
}
// TCP服务端
class ServerThread implements Runnable {
private Socket socket;
public ServerThread(Socket socket) {
this.socket = socket;
}
public void run() {
String name = socket.getInetAddress().getHostName(); // 获取客户端的主机名
int count = 1; // 上传文件个数
try {
InputStream in = socket.getInputStream();
File parentFile = new File("D:\\Test\\"); // 创建上传文件目录的File对象
// 如果不存在, 就创建这个目录
if (!parentFile.exists()) {
parentFile.mkdir();
}
File file = new File(parentFile, name + "(" + count + ").jpg");
while (file.exists()) {
file = new File(parentFile, name + "(" + (count++) + ").jpg");
}
FileOutputStream fos = new FileOutputStream(file);
byte[] buf = new byte[1024];
int len = 0;
while ((len = in.read(buf)) != -1) {
fos.write(buf, 0, len);
}
OutputStream out = socket.getOutputStream();
out.write("上传文件成功!".getBytes()); // 上传文件成功后向客户端写出"上传文件成功!"
fos.close();
socket.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}
Demo_04.java
package TCP;
import java.io.*;
import java.net.*;
// TCP客户端
public class Demo_04 {
public static void main(String[] args) throws Exception {
Socket socket = new Socket("192.168.1.100", 10001); // 创建客户端Socket
socket.setSoTimeout(10000); // 设置超时
OutputStream out = socket.getOutputStream();
FileInputStream fis = new FileInputStream("D:\\Test.jpg"); // 要上传的文件
byte[] buf = new byte[1024];
int len;
while ((len = fis.read(buf)) != -1) {
out.write(buf, 0, len);
}
socket.shutdownOutput(); // 关闭客户端输出流
InputStream in = socket.getInputStream();
byte[]bufMsg = new byte[1024];
int num = in.read(bufMsg); // 接收服务端的信息
System.out.println(new String(bufMsg, 0, num));
fis.close();
socket.close();
}
}
运行结果
- Demo_03.java
依次开启3个客户端,分别上传文件到服务器端。为了验证图片是否上传成功,进入D:\Test目录下,在该目录下可以看见三张以name+count编号命名的图片,说明文件上传成功。
- Demo_04.java
客户端上传文件成功后,会读取服务器端发送的“上传文件成功!”信息。
总结
shutdownOutput()方法非常重要,因为服务器端程序(Demo_03.java)在while循环中读取客户端(Demo_04.java)发送的数据,当读取到-1时才会结束循环。如果在客户端不调用shutdownOutput()方法关闭客户端的输出流,服务器端就不会读到-1,而会一直执行while循环,同时客户端读取服务器端数据的read(byte [])方法也是一个阻塞方法,这样服务器端和客户端进入了一个“死锁”状态,两个程序都不能结束。