Class: Rack::RequestID

Inherits:
Object
  • Object
show all
Defined in:
lib/rack/requestid.rb

Constant Summary collapse

REQUEST_ID_HEADER = 
'X-Request-Id'
REQUEST_ID_KEY = 
'HTTP_X_REQUEST_ID'

Instance Method Summary collapse

Constructor Details

#initialize(app, options = {}) ⇒ RequestID

Returns a new instance of RequestID.



9
10
11
12
13
14
15
 
# File 'lib/rack/requestid.rb', line 9

def initialize(app, options={})
  @app = app

  @include_response_header = options.fetch(:include_response_header, true)
  @overwrite = options.fetch(:overwrite, false)
  @generator = options[:generator] || lambda { SecureRandom.uuid }
end

Instance Method Details

#call(env) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/rack/requestid.rb', line 17

def call(env)
  if !env.key?(REQUEST_ID_KEY) || @overwrite
    request_id = @generator.call
    env[REQUEST_ID_KEY] = request_id
  end

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

  if @include_response_header
    headers[REQUEST_ID_HEADER] = env[REQUEST_ID_KEY]
  end

  [status, headers, body]
end