debian-mirror-gitlab/lib/gitlab/metrics/requests_rack_middleware.rb

# frozen_string_literal: true

module Gitlab
  module Metrics
    class RequestsRackMiddleware
      HTTP_METHODS = %w(delete get head options patch post put).to_set.freeze

      HEALTH_ENDPOINT = /^\/-\/(liveness|readiness|health|metrics)\/?$/.freeze

      FEATURE_CATEGORY_HEADER = 'X-Gitlab-Feature-Category'
      FEATURE_CATEGORY_DEFAULT = 'unknown'

      def initialize(app)
        @app = app
      end

      def self.http_request_total
        @http_request_total ||= ::Gitlab::Metrics.counter(:http_requests_total, 'Request count')
      end

      def self.rack_uncaught_errors_count
        @rack_uncaught_errors_count ||= ::Gitlab::Metrics.counter(:rack_uncaught_errors_total, 'Request handling uncaught errors count')
      end

      def self.http_request_duration_seconds
        @http_request_duration_seconds ||= ::Gitlab::Metrics.histogram(:http_request_duration_seconds, 'Request handling execution time',
                                                           {}, [0.05, 0.1, 0.25, 0.5, 0.7, 1, 2.5, 5, 10, 25])
      end

      def self.http_health_requests_total
        @http_health_requests_total ||= ::Gitlab::Metrics.counter(:http_health_requests_total, 'Health endpoint request count')
      end

      def self.initialize_http_request_duration_seconds
        HTTP_METHODS.each do |method|
          http_request_duration_seconds.get({ method: method })
        end
      end

      def call(env)
        method = env['REQUEST_METHOD'].downcase
        method = 'INVALID' unless HTTP_METHODS.include?(method)
        started = Time.now.to_f
        health_endpoint = health_endpoint?(env['PATH_INFO'])
        status = 'undefined'
        feature_category = nil

        begin
          status, headers, body = @app.call(env)

          elapsed = Time.now.to_f - started
          feature_category = headers&.fetch(FEATURE_CATEGORY_HEADER, nil)

          unless health_endpoint
            RequestsRackMiddleware.http_request_duration_seconds.observe({ method: method }, elapsed)
          end

          [status, headers, body]
        rescue
          RequestsRackMiddleware.rack_uncaught_errors_count.increment
          raise
        ensure
          if health_endpoint
            RequestsRackMiddleware.http_health_requests_total.increment(status: status, method: method)
          else
            RequestsRackMiddleware.http_request_total.increment(status: status, method: method, feature_category: feature_category || FEATURE_CATEGORY_DEFAULT)
          end
        end
      end

      def health_endpoint?(path)
        return false if path.blank?

        HEALTH_ENDPOINT.match?(CGI.unescape(path))
      end
    end
  end
end