如何正确处理持久性TCP套接字连接(模拟HTTP服务器)?

时间:2013-03-29 08:09:16

标签: ruby multithreading sockets http

所以,我正在尝试使用套接字和Ruby模拟一些基本的HTTP持久连接 - 用于大学课程。

重点是构建一个服务器 - 能够处理多个客户端 - 接收文件路径并返回文件内容 - 就像HTTP GET一样。

当前服务器实现循环侦听客户端,在有传入连接时触发新线程并从此套接字读取文件路径。它非常愚蠢,但在使用非预先连接时它可以正常工作 - 每个连接一个请求。

但他们应该坚持不懈。

这意味着客户端不应该担心关闭连接。在非持久版本中,服务器回应响应并关闭连接 - 再见客户端,告别。 但是持久意味着服务器线程应该循环并等待更多的传入请求,直到......直到没有更多的请求。服务器如何知道?它没有!需要某种超时。我尝试用Ruby的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代码时受到启发)。你检查最终版本here(lib / http / server / client_handler.rb)

1 个答案:

答案 0 :(得分:0)

您应该实现类似心跳包的内容。客户端应该在几秒/分钟后发送特殊数据包,以确保服务器不会在客户端上超时连接。您只是避免在此调用中执行任何操作。