1
2
3
4
5
6
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
46
47
48
49
50
51
52
53
54
55
56
57
|
# frozen_string_literal: true
require 'set'
module Excon
module Middleware
class Idempotent < Excon::Middleware::Base
def self.valid_parameter_keys
[
:idempotent,
:retries_remaining,
:retry_errors,
:retry_interval,
:retry_limit
]
end
def request_call(datum)
datum[:retries_remaining] ||= datum[:retry_limit]
@stack.request_call(datum)
end
def error_call(datum)
if datum[:idempotent]
if datum.has_key?(:request_block)
if datum[:request_block].respond_to?(:rewind)
datum[:request_block].rewind
else
Excon.display_warning('Excon requests with a :request_block must implement #rewind in order to be :idempotent.')
datum[:idempotent] = false
end
end
if datum.has_key?(:response_block) && datum[:response_block].respond_to?(:rewind)
datum[:response_block].rewind
end
if datum.has_key?(:pipeline)
Excon.display_warning("Excon requests can not be :idempotent when pipelining.")
datum[:idempotent] = false
end
end
if datum[:idempotent] && datum[:retry_errors].any? {|ex| datum[:error].kind_of?(ex) } && datum[:retries_remaining] > 1
sleep(datum[:retry_interval]) if datum[:retry_interval]
# reduces remaining retries, reset connection, and restart request_call
datum[:retries_remaining] -= 1
connection = datum.delete(:connection)
valid_keys = Set.new(connection.valid_request_keys(datum[:middlewares]))
datum.select! {|key, _| valid_keys.include?(key) }
connection.request(datum)
else
@stack.error_call(datum)
end
end
end
end
end
|
