gitlab-org--gitlab-foss/lib/gitlab/request_context.rb

# frozen_string_literal: true

module Gitlab
  class RequestContext
    class << self
      def client_ip
        Gitlab::SafeRequestStore[:client_ip]
      end

      def start_thread_cpu_time
        Gitlab::SafeRequestStore[:start_thread_cpu_time]
      end
    end

    def initialize(app)
      @app = app
    end

    def call(env)
      # We should be using ActionDispatch::Request instead of
      # Rack::Request to be consistent with Rails, but due to a Rails
      # bug described in
      # https://gitlab.com/gitlab-org/gitlab-foss/issues/58573#note_149799010
      # hosts behind a load balancer will only see 127.0.0.1 for the
      # load balancer's IP.
      req = Rack::Request.new(env)

      Gitlab::SafeRequestStore[:client_ip] = req.ip

      Gitlab::SafeRequestStore[:start_thread_cpu_time] = Gitlab::Metrics::System.thread_cpu_time

      @app.call(env)
    end
  end
end