class ActiveRecord::QueryCache

def call(env)
  enabled       = ActiveRecord::Base.connection.query_cache_enabled
  connection_id = ActiveRecord::Base.connection_id
  ActiveRecord::Base.connection.enable_query_cache!
  response = @app.call(env)
  response[2] = Rack::BodyProxy.new(response[2]) do
    restore_query_cache_settings(connection_id, enabled)
  end
  response
rescue Exception => e
  restore_query_cache_settings(connection_id, enabled)
  raise e
end