如何在Spring Boot中实现跨计算机文件传输

核心思路 #

你的需求本质是服务器作为中继的文件流式传输，避免落地存储，核心是把文件从发送端分块/流式传给服务器，服务器实时转发给接收端，而非先存到服务器再让接收端下载。

技术方向与具体实现建议 #

1. 选择合适的通信协议 #

• WebSocket（优先推荐）：既然已经实现文本聊天，WebSocket的长连接特性天然适配双向实时通信，可直接复用现有连接框架。

  • 实现要点：
    • 发送端将文件拆分成固定大小的二进制块（比如1024KB），携带文件元数据（文件名、大小、目标用户ID、块序号、总块数），通过WebSocket发送给服务器。
    • 服务器收到块后，根据目标用户ID找到对应的WebSocket会话，直接转发该二进制块和元数据。
    • 接收端接收所有块后，按序号拼接还原成完整文件。
  • 注意：需处理断重连、块丢失重传逻辑，比如接收端每接收N个块就向服务器发送确认，服务器再通知发送端哪些块需要重传。

• HTTP + 分块传输编码：若不想改动WebSocket，也可利用HTTP的分块特性实现。

  • 发送端用MultipartFile分块上传，每个请求携带块信息，服务器收到后立即通过HTTP响应流转发给接收端（接收端需保持长连接等待数据）。
  • 缺点：HTTP是短连接，需额外处理连接维持和状态跟踪，效率不如WebSocket。

2. Spring Boot端的关键实现 #

• WebSocket处理：
  • 采用Spring WebSocket的BinaryMessage传输文件块，元数据可放在消息头或作为独立的TextMessage提前发送。
  • 维护用户ID到WebSocket会话的映射（比如用ConcurrentHashMap），转发时根据目标ID找到对应会话。
  • 示例代码片段：

    // 服务器接收文件块并转发
    @OnMessage
    public void handleBinaryMessage(Session session, byte[] payload) {
        // 解析payload中的文件块元数据（可将元数据放在字节数组头部，或单独用TextMessage发送）
        FileChunk chunk = parseFileChunk(payload);
        Session targetSession = userSessionMap.get(chunk.getTargetUserId());
        if (targetSession != null && targetSession.isOpen()) {
            targetSession.getBasicRemote().sendBinary(new BinaryMessage(payload));
        }
    }
    

• 传输状态管理：服务器需临时记录每个文件传输的状态（已转发块、收发端连接状态），可使用内存缓存（如Guava Cache），超时自动清理未完成的传输任务。

3. 客户端侧处理 #

• 发送端：
  • 读取本地文件，按固定大小拆分块后依次发送，同时监听服务器的确认消息，处理重传逻辑。
  • 实时显示传输进度（已发送块数/总块数）。
• 接收端：
  • 监听服务器转发的块，按序号写入临时文件，所有块接收完成后重命名为原文件名。
  • 支持显示接收进度、取消传输操作。

4. 异常与边界处理 #

• 客户端断开处理：服务器检测到会话关闭后，立即通知另一端传输中断。
• 文件大小限制：设置单块大小和总文件大小阈值，避免内存溢出。
• 二进制编解码：直接使用字节数组传输，避免转Base64带来的额外开销，确保跨平台传输无编码问题。

5. 可选优化 #

• 断点续传：服务器记录已传输的块位置，客户端重连后可从断点继续发送。
• 传输加密：对文件块进行加密（如AES），保障数据传输安全。

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