Class: Aws::Xray::Faraday

Inherits:
Faraday::Middleware
  • Object
show all
Defined in:
lib/aws/xray/faraday.rb

Instance Method Summary collapse

Constructor Details

#initialize(app, name = nil) ⇒ Faraday

Returns a new instance of Faraday.

Parameters:

  • app (Object) —

    Faraday app.

  • name (String) (defaults to: nil) —

    Logical service name for downstream API.



8
9
10
11
 
# File 'lib/aws/xray/faraday.rb', line 8

def initialize(app, name = nil)
  super(app)
  @name = name
end

Instance Method Details

#call(req_env) ⇒ Object 



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
 
# File 'lib/aws/xray/faraday.rb', line 13

def call(req_env)
  name = @name || req_env.request_headers['Host'] || "unknown-request-from-#{Context.current.name}"

  Context.current.child_trace(remote: true, name: name) do |sub|
    propagate_trace_header = sub.generate_trace_header
    req_env.request_headers[TRACE_HEADER] = propagate_trace_header.to_header_value
    sub.set_http_request(Request.build_from_faraday_env(req_env))

    @app.call(req_env).on_complete do |res_env|
      sub.set_http_response(res_env.status, res_env.response_headers['Content-Length'])
      case res_env.status
      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
    end
  end
end