我正在尝试使用Ruby和套接字模拟一些基本的HTTP持久连接,为了我的大学课程。
目标是构建一个服务器,能够处理多个客户端,接收文件路径并返回文件内容,就像HTTP GET一样。
当前的服务器实现循环监听客户端,当有新连接时启动一个新线程,并从该套接字读取文件路径。这非常简单,但在处理非持久连接(每个连接一个请求)时效果很好。
但应该是持久连接。
这意味着客户端不必担心关闭连接。在非持久版本中,服务器会回显响应并关闭连接 - 再见客户端,告别。但持久意味着服务器线程应该循环并等待更多的传入请求,直到... 直到没有更多的请求。服务器如何知道这一点?它不知道!需要某种超时。我试图使用Ruby的Timeout来实现这一点,但它没有起作用。
通过搜索解决方案 - 除了彻底建议避免使用Timeout模块之外 - 我看到了很多关于IO.select方法的帖子,它应该比使用线程等东西更好地处理这个套接字等待问题(考虑到Ruby线程的工作方式,这真的很酷)。我正在努力理解IO.select如何工作,但仍然无法在当前情况下使其工作。
所以我基本上问了两件事:
如何有效地处理服务器端的超时问题,可以使用一些基于线程的解决方案、低级套接字选项或一些IO.select魔法?
客户端如何知道服务器已关闭连接的一侧?
以下是服务器的当前代码:
require 'date'
module Sockettp
class Server
def initialize(dir, port = Sockettp::DEFAULT_PORT)
@dir = dir
@port = port
end
def start
puts "Starting Sockettp server..."
puts "Serving #{@dir.yellow} on port #{@port.to_s.green}"
Socket.tcp_server_loop(@port) do |socket, client_addrinfo|
handle socket, client_addrinfo
end
end
private
def handle(socket, addrinfo)
Thread.new(socket) do |client|
log "New client connected"
begin
loop do
if client.eof?
puts "#{'-' * 100} end connection"
break
end
input = client.gets.chomp
body = content_for(input)
response = {}
if body
response.merge!({
status: 200,
body: body
})
else
response.merge!({
status: 404,
body: Sockettp::STATUSES[404]
})
end
log "#{addrinfo.ip_address} #{input} -- #{response[:status]} #{Sockettp::STATUSES[response[:status]]}".send(response[:status] == 200 ? :green : :red)
client.puts(response.to_json)
end
ensure
socket.close
end
end
end
def content_for(path)
path = File.join(@dir, path)
return File.read(path) if File.file?(path)
return Dir["#{path}/*"] if File.directory?(path)
end
def log(msg)
puts "#{Thread.current} -- #{DateTime.now.to_s} -- #{msg}"
end
end
end
更新
我可以使用IO.select方法模拟超时行为,但是在与几个用于接受新连接和其他几个用于处理请求的线程结合使用时,实现感觉不好。并发使情况变得疯狂和不稳定,除非我能找到更好的使用该解决方案的方法,否则我可能不会坚持它。
更新2
看起来Timeout仍然是处理这个问题的最佳方式。我将坚持它直到找到一个更好的选择。 我仍然不知道如何处理僵尸客户端连接。
解决方案
最终我使用了IO.select(在查看webrick代码时受到启发)。您可以在此处检查最终版本(lib/http/server/client_handler.rb)。