2018-11-17 00:37:17 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2017-07-12 22:46:17 +00:00
|
|
|
module Gitlab
|
|
|
|
module Metrics
|
|
|
|
class RequestsRackMiddleware
|
2019-10-02 21:06:22 +00:00
|
|
|
HTTP_METHODS = {
|
2020-09-21 12:09:34 +00:00
|
|
|
"delete" => %w(200 202 204 303 400 401 403 404 500 503),
|
|
|
|
"get" => %w(200 204 301 302 303 304 307 400 401 403 404 410 422 429 500 503),
|
|
|
|
"head" => %w(200 204 301 302 303 401 403 404 410 500),
|
2019-10-02 21:06:22 +00:00
|
|
|
"options" => %w(200 404),
|
2020-09-21 12:09:34 +00:00
|
|
|
"patch" => %w(200 202 204 400 403 404 409 416 500),
|
|
|
|
"post" => %w(200 201 202 204 301 302 303 304 400 401 403 404 406 409 410 412 422 429 500 503),
|
|
|
|
"put" => %w(200 202 204 400 401 403 404 405 406 409 410 422 500)
|
2019-10-02 21:06:22 +00:00
|
|
|
}.freeze
|
|
|
|
|
2020-04-06 15:10:04 +00:00
|
|
|
HEALTH_ENDPOINT = /^\/-\/(liveness|readiness|health|metrics)\/?$/.freeze
|
|
|
|
|
2020-10-15 09:08:41 +00:00
|
|
|
FEATURE_CATEGORY_HEADER = 'X-Gitlab-Feature-Category'
|
|
|
|
FEATURE_CATEGORY_DEFAULT = 'unknown'
|
|
|
|
|
2017-07-12 22:46:17 +00:00
|
|
|
def initialize(app)
|
|
|
|
@app = app
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.http_request_total
|
2019-02-14 18:05:35 +00:00
|
|
|
@http_request_total ||= ::Gitlab::Metrics.counter(:http_requests_total, 'Request count')
|
2017-07-12 22:46:17 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def self.rack_uncaught_errors_count
|
2019-02-14 18:05:35 +00:00
|
|
|
@rack_uncaught_errors_count ||= ::Gitlab::Metrics.counter(:rack_uncaught_errors_total, 'Request handling uncaught errors count')
|
2017-07-12 22:46:17 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def self.http_request_duration_seconds
|
2019-02-14 18:05:35 +00:00
|
|
|
@http_request_duration_seconds ||= ::Gitlab::Metrics.histogram(:http_request_duration_seconds, 'Request handling execution time',
|
2017-07-12 22:46:17 +00:00
|
|
|
{}, [0.05, 0.1, 0.25, 0.5, 0.7, 1, 2.5, 5, 10, 25])
|
|
|
|
end
|
|
|
|
|
2020-04-06 15:10:04 +00:00
|
|
|
def self.http_health_requests_total
|
|
|
|
@http_health_requests_total ||= ::Gitlab::Metrics.counter(:http_health_requests_total, 'Health endpoint request count')
|
|
|
|
end
|
|
|
|
|
2019-10-02 21:06:22 +00:00
|
|
|
def self.initialize_http_request_duration_seconds
|
|
|
|
HTTP_METHODS.each do |method, statuses|
|
|
|
|
statuses.each do |status|
|
2019-11-13 21:06:45 +00:00
|
|
|
http_request_duration_seconds.get({ method: method, status: status.to_s })
|
2019-10-02 21:06:22 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-07-12 22:46:17 +00:00
|
|
|
def call(env)
|
|
|
|
method = env['REQUEST_METHOD'].downcase
|
2020-09-21 12:09:34 +00:00
|
|
|
method = 'INVALID' unless HTTP_METHODS.key?(method)
|
2017-07-12 22:46:17 +00:00
|
|
|
started = Time.now.to_f
|
2020-09-28 15:09:44 +00:00
|
|
|
health_endpoint = health_endpoint?(env['PATH_INFO'])
|
2020-10-07 18:08:34 +00:00
|
|
|
status = 'undefined'
|
2020-10-15 09:08:41 +00:00
|
|
|
feature_category = nil
|
2020-04-06 15:10:04 +00:00
|
|
|
|
2017-07-12 22:46:17 +00:00
|
|
|
begin
|
|
|
|
status, headers, body = @app.call(env)
|
|
|
|
|
|
|
|
elapsed = Time.now.to_f - started
|
2020-10-15 09:08:41 +00:00
|
|
|
feature_category = headers&.fetch(FEATURE_CATEGORY_HEADER, nil)
|
2020-09-28 15:09:44 +00:00
|
|
|
|
|
|
|
unless health_endpoint
|
|
|
|
RequestsRackMiddleware.http_request_duration_seconds.observe({ method: method, status: status.to_s }, elapsed)
|
|
|
|
end
|
2017-07-12 22:46:17 +00:00
|
|
|
|
|
|
|
[status, headers, body]
|
|
|
|
rescue
|
|
|
|
RequestsRackMiddleware.rack_uncaught_errors_count.increment
|
|
|
|
raise
|
2020-10-07 18:08:34 +00:00
|
|
|
ensure
|
|
|
|
if health_endpoint
|
2020-10-15 09:08:41 +00:00
|
|
|
RequestsRackMiddleware.http_health_requests_total.increment(status: status, method: method)
|
2020-10-07 18:08:34 +00:00
|
|
|
else
|
2020-10-15 09:08:41 +00:00
|
|
|
RequestsRackMiddleware.http_request_total.increment(status: status, method: method, feature_category: feature_category || FEATURE_CATEGORY_DEFAULT)
|
2020-10-07 18:08:34 +00:00
|
|
|
end
|
2017-07-12 22:46:17 +00:00
|
|
|
end
|
|
|
|
end
|
2020-04-06 15:10:04 +00:00
|
|
|
|
|
|
|
def health_endpoint?(path)
|
|
|
|
return false if path.blank?
|
|
|
|
|
|
|
|
HEALTH_ENDPOINT.match?(CGI.unescape(path))
|
|
|
|
end
|
2017-07-12 22:46:17 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|