summaryrefslogtreecommitdiff
path: root/lib/gitlab/middleware/correlation_id.rb
blob: 73542dd422e36df48b91c28a60c06a0e6072acf4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# frozen_string_literal: true

# A dumb middleware that steals correlation id
# and sets it as a global context for the request
module Gitlab
  module Middleware
    class CorrelationId
      include ActionView::Helpers::TagHelper

      def initialize(app)
        @app = app
      end

      def call(env)
        ::Gitlab::CorrelationId.use_id(correlation_id(env)) do
          @app.call(env)
        end
      end

      private

      def correlation_id(env)
        if Gitlab.rails5?
          request(env).request_id
        else
          request(env).uuid
        end
      end

      def request(env)
        ActionDispatch::Request.new(env)
      end
    end
  end
end