基于Socket的简易客户端-服务端程序文件传输功能实现问询

Hey there! 既然你已经搞定了Socket的文本消息互传，那扩展文件传输其实是在现有基础上的自然延伸——核心难点是区分普通文本消息和文件传输指令，毕竟你之前用的BufferedReader/PrintWriter是字符流，只适合处理文本，而文件（尤其是二进制文件，比如图片、压缩包）必须用字节流来传输。下面给你一步步拆解实现思路和代码示例：

第一步：约定通信协议 #

首先得和服务端/客户端约定好指令格式，让双方知道什么时候要切换到文件传输模式。比如我们可以定义：

• 普通文本消息：直接发送字符串即可
• 文件传输指令：SEND_FILE|文件名|文件大小（用|分隔参数，避免和文件名里的空格冲突）
• 传输结束标记：FILE_TRANSFER_COMPLETE
• 错误反馈：ERROR|错误描述

第二步：客户端实现文件发送 #

客户端需要先发送指令头告知服务端要传文件，再用字节流传输文件内容，最后发送结束标记：

// 客户端发送文件的方法
public void sendFile(String filePath) throws IOException {
    File targetFile = new File(filePath);
    // 先检查文件是否存在
    if (!targetFile.exists() || !targetFile.isFile()) {
        printWriter.println("ERROR|指定文件不存在或不是有效文件");
        printWriter.flush(); // 必须flush，确保消息立即发送
        return;
    }

    // 1. 发送文件传输指令头
    printWriter.println("SEND_FILE|" + targetFile.getName() + "|" + targetFile.length());
    printWriter.flush(); // 这里一定要flush，不然字符流的缓冲区会把指令卡住

    // 2. 用字节流传输文件内容
    FileInputStream fileIn = new FileInputStream(targetFile);
    OutputStream socketOut = socket.getOutputStream();
    byte[] buffer = new byte[4096]; // 4KB缓冲区，平衡速度和内存占用
    int bytesRead;

    while ((bytesRead = fileIn.read(buffer)) != -1) {
        socketOut.write(buffer, 0, bytesRead);
        socketOut.flush();
    }
    fileIn.close();

    // 3. 发送传输完成标记，让服务端切换回文本模式
    printWriter.println("FILE_TRANSFER_COMPLETE");
    printWriter.flush();
}

第三步：服务端实现文件接收 #

服务端需要监听指令，一旦识别到文件传输指令，就切换到字节流接收文件，完成后回复确认：

// 服务端处理文件传输的方法
private void handleFileTransfer(String[] commandParts) throws IOException {
    String fileName = commandParts[1];
    long totalFileSize = Long.parseLong(commandParts[2]);
    File receivedFile = new File("received_" + fileName); // 给接收的文件加前缀避免覆盖

    FileOutputStream fileOut = new FileOutputStream(receivedFile);
    InputStream socketIn = socket.getInputStream();
    byte[] buffer = new byte[4096];
    long receivedBytes = 0;
    int bytesRead;

    // 循环接收直到达到文件总大小
    while (receivedBytes < totalFileSize) {
        bytesRead = socketIn.read(buffer);
        if (bytesRead == -1) {
            // 连接意外中断，删除未完成的文件
            receivedFile.delete();
            printWriter.println("ERROR|文件传输中断");
            printWriter.flush();
            return;
        }
        fileOut.write(buffer, 0, bytesRead);
        receivedBytes += bytesRead;
    }
    fileOut.close();

    // 回复客户端接收成功
    printWriter.println("FILE_RECEIVED_SUCCESS|" + fileName);
    printWriter.flush();
}

// 服务端主监听循环（修改原有逻辑）
public void startListening() throws IOException {
    BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
    PrintWriter writer = new PrintWriter(socket.getOutputStream(), true);
    String message;

    while ((message = reader.readLine()) != null) {
        if (message.startsWith("SEND_FILE|")) {
            // 解析指令：拆分参数，注意转义如果文件名有|（这里简化处理，假设文件名不含|）
            String[] parts = message.split("\\|");
            if (parts.length == 3) {
                handleFileTransfer(parts);
            } else {
                writer.println("ERROR|无效的文件传输指令格式");
            }
        } else if (message.equals("FILE_TRANSFER_COMPLETE")) {
            // 可以忽略，或者记录日志
            System.out.println("文件传输已完成");
        } else {
            // 原有文本消息处理逻辑
            System.out.println("收到文本消息：" + message);
            writer.println("已收到你的消息：" + message);
        }
    }
}

关键注意事项 #

• 字符流与字节流不能混用缓冲区：一定要在切换流之前调用flush()清空字符流的缓冲区，否则指令可能会被留在缓冲区里，导致对方接收不到正确的指令。
• 处理异常与中断：要捕获IO异常，比如文件不存在、连接中断，及时发送错误反馈给对方，避免双方一直阻塞等待。
• 文件名与大小校验：服务端要校验文件名的合法性（比如避免路径遍历攻击，比如文件名不能包含../），文件大小也要做合理性判断，防止恶意发送超大文件。
• 大文件优化：如果要传大文件，可以加入进度反馈（比如每传输10%发送一个进度指令），或者用NIO的非阻塞模式，但简易应用用阻塞IO就足够了。

要是你在具体实现中遇到比如缓冲区冲突、传输超时这类细节问题，随时再细化提问～

内容的提问来源于stack exchange，提问作者Josh3248