Java网络编程Demo（二）TCP文件上传

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 [])方法也是一个阻塞方法，这样服务器端和客户端进入了一个“死锁”状态，两个程序都不能结束。