这是一个简单的包含发送端和接收端的例子。发送端向接收端发送文件名和文件内容,接收端将收到的文件保存在磁盘上。接收端可以同时接收多个发送端传来的文件,但没有处理文件同名的情况。
这个例子中设计了一个简单的协议。发送的内容是这样的:
文件名长度(4字节)—文件名—文件内容长度(4字节)—文件内容。
这个例子中设计了一个简单的协议。发送的内容是这样的:
文件名长度(4字节)—文件名—文件内容长度(4字节)—文件内容。
接收端也按照这个结构进行解析。建议先看 Client 类,再看 Server 类。
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
/**
* 简单的文件发送与接收示例
*/
public class FileTrasmission {
// 程序入口
public static void main(String[] args) throws Exception {
int port = 7788;
new Server(port, "c:\\save\\").start();
new Client().sendFile("127.0.0.1", port, "c:\\迷失在康熙末年.txt");
}
}
/**
* 接收端。可同时接收多个发送端发来的文件。但如果发来的文件是同名的话那就乱了。
*/
class Server {
private int listenPort;
private String savePath;
/**
* 构造方法
*
* @param listenPort
* 侦听端口
* @param savePath
* 接收的文件要保存的路径
*
* @throws IOException
* 如果创建保存路径失败
*/
Server(int listenPort, String savePath) throws IOException {
this.listenPort = listenPort;
this.savePath = savePath;
File file = new File(savePath);
if (!file.exists() && !file.mkdirs()) {
throw new IOException("无法创建文件夹 " + savePath);
}
}
// 开始侦听
public void start() {
new ListenThread().start();
}
// 网上抄来的,将字节转成 int。b 长度不得小于 4,且只会取前 4 位。
public static int b2i(byte[] b) {
int value = 0;
for (int i = 0; i < 4; i++) {
int shift = (4 - 1 - i) * 8;
value += (b[i] & 0x000000FF) << shift;
}
return value;
}
/**
* 侦听线程
*/
private class ListenThread extends Thread {
@Override
public void run() {
try {
ServerSocket server = new ServerSocket(listenPort);
// 开始循环
while (true) {
Socket socket = server.accept();
new HandleThread(socket).start();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* 读取流并保存文件的线程
*/
private class HandleThread extends Thread {
private Socket socket;
private HandleThread(Socket socket) {
this.socket = socket;
}
@Override
public void run() {
try {
InputStream is = socket.getInputStream();
readAndSave(is);
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
socket.close();
} catch (IOException e) {
// nothing to do
}
}
}
// 从流中读取内容并保存
private void readAndSave(InputStream is) throws IOException {
String filename = getFileName(is);
int file_len = readInteger(is);
System.out.println("接收文件:" + filename + ",长度:" + file_len);
readAndSave0(is, savePath + filename, file_len);
System.out.println("文件保存成功(" + file_len + "字节)。");
}
private void readAndSave0(InputStream is, String path, int file_len)
throws IOException {
FileOutputStream os = getFileOS(path);
readAndWrite(is, os, file_len);
os.close();
}
// 边读边写,直到读取 size 个字节
private void readAndWrite(InputStream is, FileOutputStream os, int size)
throws IOException {
byte[] buffer = new byte[4096];
int count = 0;
while (count < size) {
int n = is.read(buffer);
// 这里没有考虑 n = -1 的情况
os.write(buffer, 0, n);
count += n;
}
}
// 读取文件名
private String getFileName(InputStream is) throws IOException {
int name_len = readInteger(is);
byte[] result = new byte[name_len];
is.read(result);
return new String(result);
}
// 读取一个数字
private int readInteger(InputStream is) throws IOException {
byte[] bytes = new byte[4];
is.read(bytes);
return b2i(bytes);
}
// 创建文件并返回输出流
private FileOutputStream getFileOS(String path) throws IOException {
File file = new File(path);
if (!file.exists()) {
file.createNewFile();
}
return new FileOutputStream(file);
}
}
}
/**
* 发送端
*/
class Client {
// 网上抄来的,将 int 转成字节
public static byte[] i2b(int i) {
return new byte[] { (byte) ((i >> 24) & 0xFF),
(byte) ((i >> 16) & 0xFF), (byte) ((i >> 8) & 0xFF),
(byte) (i & 0xFF) };
}
/**
* 发送文件。文件大小不能大于 {@link Integer#MAX_VALUE}
*
* @param hostname
* 接收端主机名或 IP 地址
* @param port
* 接收端端口号
* @param filepath
* 文件路径
*
* @throws IOException
* 如果读取文件或发送失败
*/
public void sendFile(String hostname, int port, String filepath)
throws IOException {
File file = new File(filepath);
FileInputStream is = new FileInputStream(filepath);
Socket socket = new Socket(hostname, port);
OutputStream os = socket.getOutputStream();
try {
int length = (int) file.length();
System.out.println("发送文件:" + file.getName() + ",长度:" + length);
// 发送文件名和文件内容
writeFileName(file, os);
writeFileContent(is, os, length);
} finally {
os.close();
is.close();
}
}
// 输出文件内容
private void writeFileContent(InputStream is, OutputStream os, int length)
throws IOException {
// 输出文件长度
os.write(i2b(length));
// 输出文件内容
byte[] buffer = new byte[4096];
int size;
while ((size = is.read(buffer)) != -1) {
os.write(buffer, 0, size);
}
}
// 输出文件名
private void writeFileName(File file, OutputStream os) throws IOException {
byte[] fn_bytes = file.getName().getBytes();
os.write(i2b(fn_bytes.length)); // 输出文件名长度
os.write(fn_bytes); // 输出文件名
}
}