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.

Raises:

  • (ArgumentError) 


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

def initialize(app, name = nil)
  raise ArgumentError.new("name must be a String but given a #{name.class}") if name && !name.is_a?(String)
  super(app)
  @name = name
end

Instance Method Details

#call(req_env) ⇒ Object 



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

def call(req_env)
  return @app.call(req_env) unless Context.started?

  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 = sub.generate_trace
    req_env.request_headers[TRACE_HEADER] = propagate_trace.to_header_value
    sub.set_http_request(Request.build_from_faraday_env(req_env))

    res = Context.current.disable_trace(:net_http) { @app.call(req_env) }
    res.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