从 TCPSocket 读取整个消息而不挂起

Question

我在 Ruby 3.0.2 中组装了一个 TCPServer，我发现我似乎无法在不阻塞的情况下读取整个数据包（直到套接字关闭）。

编辑：我试图做的事情有些混乱 - 我的错 - 所以只是为了帮助澄清：我想阅读到目前为止通过 TCP 连接发送的所有内容。 （结束编辑）

我的第一次尝试是：

#!/snap/bin/ruby
require 'socket'

server = TCPServer.new('localhost', 4200)

loop {
  Thread.start(server.accept) do |connection|
    puts connection.gets  # The important line
  end
}

但这会一直挂起，直到客户端关闭连接。好的，所以我看一下 connection.methods 和 ruby​​ 文档，并尝试了一堆看起来很有希望的选项。基本上，有两种读取方法：阻塞和非阻塞。

我尝试的阻塞方法是.read、.gets、.readlines、.readline、.recv和.recvmsg。现在.read、.readlines 和.gets 都挂起（直到套接字关闭）——所以这没有帮助。其他的（例如.readline、recv 方法）不会阅读整个消息。现在，我可以读取每一行，直到看到一个空行并从那里解析 HTTP 标头。但是必须有更好的方法；我不想担心收到损坏的消息并挂起，因为我没有读取标题末尾的空行。

所以我去看了非阻塞选项。特别是.recv_nonblock 和.recvmsg_nonblock。这两个抛出错误（资源暂时不可用 - recvfrom(2) 会阻塞 和 资源暂时不可用 - recvmsg(2) 分别）。

对可能发生的事情有任何想法吗？我认为它与我使用 Ruby 3 有关系，因为在 Ruby 2.5 上尝试代码，client.gets 返回一行（不会挂起），尽管 .readlines 确实挂起 - 所以不会确定发生了什么。

理想情况下，我可以像 client.get_message 那样调用一些东西，我会得到已发送的整个消息，但我也可以在 TCP 级别工作并获取数据包大小，阅读大小，并从那里重建消息。

Answer 1

感谢所有评论/回答的人，但我找到了我认为是 Socket 类的创建者想要的解决方案！

recv_nonblock 方法接受一些可选参数 - 其中一个是 Socket 将存储它已读取内容的缓冲区。因此，像client.recv_nonblock(1000, 0, buffer) 这样的调用将多达 1000 个字符从 Socket 存储到 buffer，然后退出而不是阻塞。

为了方便起见，我为 TCPSocket 类做了一个猴子补丁：

class TCPSocket

  def eat_buffer
    contents = ''
    buffer = ''
    begin
    loop {
      recv_nonblock(256, 0, buffer)
      contents += buffer
    }
    rescue IO::EAGAINWaitReadable
      contents
    end
  end

end

Steffen 在 cmets 中提出的观点得到了很好的理解 - TCP 并非设计用于这种方式。这是一种 hacky（在不好的意义上）的方法，应该避免。

Answer 2

TCP 只是将您写入的字节传输到套接字，并保证按照发送顺序接收。如果您有“消息”的概念，那么您需要将其添加到您的服务器和客户端中。

.gets 特别会阻塞，直到它读取新的“行”，或者您定义为字符串分隔符的任何内容 - 请参阅文档IO#gets。这意味着在您的服务器从客户端接收到该字节之前，它将阻塞。

在您的客户端中查看您是如何写入数据的 - 如果您使用的是 ruby​​，那么 puts 会起作用，因为它将用新行终止字符串。如果您使用的是write，那么它只会写入没有新行的字符串

即。

# client.rb
c = TCPSocket.new 'localhost', 5000
c.puts "foo"
c.write "bar"
c.write "baz\n"

# server.rb
s = TCPServer.new 5000
loop do
  client = s.accept
  puts client.gets
  puts client.gets
end

会输出

foo
barbaz