当用户中止连接时执行某些操作（Sinatra + Thin）

Question

我正在编写一个有时需要非常长时间运行的数据库请求的应用程序。如果客户端重新加载或关闭页面以处理数据库请求，我想执行一些代码。

我本来希望 Rack 能够与这种事情挂钩，但显然从我所看到的来看，这比 Rack 更深层次。

到目前为止，我能找到的唯一挂钩是通过对 Thin Connection 类中的取消绑定函数进行猴子修补来进入 Thin 本身：

module Thin
  class Connection < EventMachine::Connection

    def unbind

      # DO something here

      @request.async_close.succeed if @request.async_close
      @response.body.fail if @response.body.respond_to?(:fail)
      @backend.connection_finished(self)
    end
  end
end

这会覆盖 Thin 的取消绑定函数，并让我挂钩由 EventMachine 调用的断开连接。

有更好的办法吗？

Answer 1

经过一番挖掘，我发现 Thin 提供了一种替换“后端”的机制，或者服务器如何连接到客户端的机制。我正在使用它，结合机架环境中的值来处理特定的请求实例，并知道我是否需要终止查询：

class Backend < Thin::Backends::TcpServer

  def initialize(host, port, options={})
    super(host, port)
  end

  def connection_finished(connection)
    super(connection)

    if connection.request.env["query_killer"]
      connection.request.env["query_killer"].kill
    end

  end

end

这可以通过命令行参数包含在 Thin 中：

thin start  -r 'my_module/backend' --backend MyModule::Backend