Ruby 3.0.2 TCPServer如何不阻塞读取TCPSocket全部已发送消息

核心问题说明 #

TCP 本身是流式传输协议，没有内置数据包边界，不存在系统层的「读取所有已发送完整消息」的接口，你需要的能力本质是读取当前内核 socket 缓冲区中所有已到达的数据，同时避免永久阻塞。

你遇到的 Ruby 3 与 2.5 版本行为差异，是因为 Ruby 3.0 调整了 IO 类读取方法的默认缓冲逻辑：gets 方法默认会等待完整换行符才会返回，如果客户端发送的数据未携带换行符，就会一直阻塞到连接关闭；而 Ruby 2.5 版本存在缓冲自动刷新的兼容逻辑，无换行时也可能返回部分数据，属于版本设计差异。

可行实现方案 #

方案1：非阻塞读捕获异常实现（最贴合你的需求） #

recv_nonblock 抛出的 Resource temporarily unavailable 错误属于正常逻辑，说明当前内核缓冲区已经没有更多可读数据，捕获该异常即可终止读取，刚好拿到所有已到达的内容。
封装后的通用方法示例：

def get_all_available_data(socket, chunk_size = 4096)
  data = +'' # 可变字符串优化性能
  loop do
    begin
      data << socket.recv_nonblock(chunk_size)
    rescue IO::WaitReadable, EOFError
      # WaitReadable = 无更多已到达数据，EOFError = 连接已关闭
      break
    end
  end
  data.empty? ? nil : data
end

替换你原有代码中的 connection.gets 即可：

Thread.start(server.accept) do |connection|
  puts get_all_available_data(connection)
end

方案2：带超时的读取逻辑（适合处理有明确格式的协议，比如HTTP） #

如果需要等待可能延迟到达的部分数据、但又不想永久阻塞，可以结合 IO.select 增加超时控制，超过设定时间无新数据就终止读取：

def get_data_with_timeout(socket, timeout = 2, chunk_size = 4096)
  data = +''
  loop do
    # 等待socket可读，超时则返回nil，终止读取
    readable = IO.select([socket], nil, nil, timeout)
    break unless readable

    begin
      chunk = socket.recv_nonblock(chunk_size)
      data << chunk
    rescue IO::WaitReadable
      break
    rescue EOFError
      break
    end
  end
  data.empty? ? nil : data
end

如果是解析HTTP头的场景，还可以在读取过程中判断是否已经读到空行，提前终止读取，遇到超时直接判定为非法请求关闭连接即可，不会导致程序挂起。

方案3：应用层定义消息边界（生产环境推荐） #

如果是自定义TCP协议的场景，最稳妥的方式是在应用层约定消息格式，比如：

• 先发送4字节的消息长度，再发送对应长度的内容，读取时先读4字节拿到长度，再读指定长度的数据即可拿到完整消息
• 约定特殊分隔符（比如\r\n\r\n）作为消息结束标记，读取到标记就终止
  这种方式可以彻底避免TCP流无边界带来的粘包、半包问题。

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