Class: Sinatra::Response

Inherits:
Object
  • Object
show all
Includes:
Sinatra
Defined in:
lib/tcell_agent/sinatra.rb

Overview

rubocop:disable Style/ClassAndModuleChildren

Instance Method Summary collapse

Instance Method Details

#finish ⇒ Object 



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
 
# File 'lib/tcell_agent/sinatra.rb', line 13

def finish
  status, headers, response = original_finish

  TCellAgent::Instrumentation.safe_block('Setting Headers') do
    headers_policy = TCellAgent.policy(TCellAgent::PolicyTypes::HEADERS)
    policy_headers = headers_policy.get_headers(
      request.env[TCellAgent::Instrumentation::TCELL_ID]
    )
    policy_headers.each do |header_info|
      header_name = header_info['name']
      header_value = header_info['value']
      existing_header_value = headers[header_name]
      headers[header_name] = if existing_header_value
                               "#{existing_header_value}, #{header_value}"
                             else
                               header_value
                             end
    end
    response = [status, headers, active_response]
  end

  [status, headers, response]
end

#original_finish ⇒ Object 



12
 
# File 'lib/tcell_agent/sinatra.rb', line 12

alias_method :original_finish, :finish