cca61980d5
This ensures that we have more visibility in the number of SQL queries that are executed in web requests. The current threshold is hardcoded to 100 as we will rarely (maybe once or twice) change it. In production and development we use Sentry if enabled, in the test environment we raise an error. This feature is also only enabled in production/staging when running on GitLab.com as it's not very useful to other users.
55 lines
1.2 KiB
Ruby
55 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module QueryLimiting
|
|
# Middleware for reporting (or raising) when a request performs more than a
|
|
# certain amount of database queries.
|
|
class Middleware
|
|
CONTROLLER_KEY = 'action_controller.instance'.freeze
|
|
ENDPOINT_KEY = 'api.endpoint'.freeze
|
|
|
|
def initialize(app)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
transaction, retval = Transaction.run do
|
|
@app.call(env)
|
|
end
|
|
|
|
transaction.action = action_name(env)
|
|
transaction.act_upon_results
|
|
|
|
retval
|
|
end
|
|
|
|
def action_name(env)
|
|
if env[CONTROLLER_KEY]
|
|
action_for_rails(env)
|
|
elsif env[ENDPOINT_KEY]
|
|
action_for_grape(env)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def action_for_rails(env)
|
|
controller = env[CONTROLLER_KEY]
|
|
action = "#{controller.class.name}##{controller.action_name}"
|
|
|
|
if controller.content_type == 'text/html'
|
|
action
|
|
else
|
|
"#{action} (#{controller.content_type})"
|
|
end
|
|
end
|
|
|
|
def action_for_grape(env)
|
|
endpoint = env[ENDPOINT_KEY]
|
|
route = endpoint.route rescue nil
|
|
|
|
"#{route.request_method} #{route.path}" if route
|
|
end
|
|
end
|
|
end
|
|
end
|