36 lines
680 B
Ruby
36 lines
680 B
Ruby
# 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
|