Module: Aws::Xray::Hooks::NetHttp

Defined in:
lib/aws/xray/hooks/net_http.rb

Instance Method Summary collapse

Instance Method Details

#request(req, *args) ⇒ Object 



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
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/aws/xray/hooks/net_http.rb', line 7

def request(req, *args)
  return super unless Context.started?
  return super if Context.current.disabled?(:net_http)

  uri = URI('')
  uri.scheme = use_ssl? ? 'https' : 'http'
  uri.host = address
  uri.port = port
  uri.path = URI(req.path).path
  request_record = Request.build(
    method: req.method,
    url: uri.to_s,
    user_agent: req['User-Agent'],
  )
  Context.current.child_trace(remote: true, name: address) do |sub|
    propagate_trace = sub.generate_trace
    req[TRACE_HEADER] = propagate_trace.to_header_value
    sub.set_http_request(request_record)

    res = super

    sub.set_http_response(res.code.to_i, res['Content-Length'])
    case res.code.to_i
    when 499
      cause = Cause.new(stack: caller, message: 'Got 499', type: 'http_request_error')
      sub.set_error(error: true, throttle: true, cause: cause)
    when 400..498
      cause = Cause.new(stack: caller, message: 'Got 4xx', type: 'http_request_error')
      sub.set_error(error: true, cause: cause)
    when 500..599
      cause = Cause.new(stack: caller, message: 'Got 5xx', type: 'http_request_error')
      sub.set_error(fault: true, remote: true, cause: cause)
    else
      # pass
    end

    res
  end
end