Add optional max_retry and retry_statuses parameters to HttpClient

Author: struniu

README.md

@@ -49,6 +49,8 @@ The following options are available while instantiating a client:
  * __timeout__: Request timeout
  * __verbose__: Verbose/debug mode
  * __logger__: Logger used in verbose mode
+ * __max_retry__: Number of retries on failed requests. Passed to Faraday
+ * __retry_statuses__: By default only timeout error will be retries. This allows to retry on specific HTTP statuses. Passed to Faraday
 
 ### Architecture
 

lib/basecrm/configuration.rb

@@ -7,6 +7,8 @@ class Configuration
     attr_reader :user_agent
     attr_reader :timeout
     attr_reader :verify_ssl
+    attr_reader :max_retry
+    attr_reader :retry_statuses
 
     attr_reader :logger, :verbose
     alias_method :debug?, :verbose
@@ -19,6 +21,8 @@ def initialize(options={})
       @verbose = !!options[:verbose]
       @timeout = options[:timeout] || 30
       @verify_ssl = options.fetch(:verify_ssl, true)
+      @max_retry = options[:max_retry]
+      @retry_statuses = options[:retry_statuses]
     end
 
     def validate!

lib/basecrm/http_client.rb

@@ -22,6 +22,7 @@ def initialize(config)
       options[:ssl] = { verify: false } unless config.verify_ssl
 
       @client = Faraday.new(config.base_url, options) do |faraday|
+        faraday.request :retry, retry_options
         faraday.use BaseCRM::Middlewares::OAuthBearerToken, config.access_token
         faraday.use BaseCRM::Middlewares::RaiseError
         faraday.response :logger, config.logger if config.debug?
@@ -89,5 +90,12 @@ def extract_body(res)
       content_type = res.headers['Content-Type']
       content_type && content_type.include?('json') ? JSON.parse(res.body, symbolize_names: true) : res.body
     end
+
+    def retry_options
+      retry_options = {}
+      retry_options[:max] = @config.max_retry if @config.max_retry
+      retry_options[:retry_statuses] = @config.retry_statuses if @config.retry_statuses
+      retry_options
+    end
   end
 end

spec/http_client_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe BaseCRM::HttpClient do
+  describe 'Retries' do
+    let(:statuses_to_retry) { [503] }
+    let(:max_retry) { 3 }
+    subject(:client_with_retry) { client_with_basic_retry(max_retry: max_retry, on_statuses: statuses_to_retry) }
+
+    it "should pass retry statues to Faraday and do a valid request" do
+      expect(client_with_retry.http_client.client.app.options.retry_statuses).to match_array(statuses_to_retry)
+      expect(client_with_retry.http_client.client.app.options.max).to eq(max_retry)
+      expect(client_with_retry.accounts.self).to be_instance_of BaseCRM::Account
+    end
+  end
+end

spec/support/client_helpers.rb

@@ -3,6 +3,10 @@ def client
     @client ||= BaseCRM::Client.new(access_token: access_token, base_url: base_url)
   end
 
+  def client_with_basic_retry(max_retry: 1, on_statuses: [])
+    @client_with_basic_retry ||= BaseCRM::Client.new(access_token: access_token, base_url: base_url, max_retry: max_retry, retry_statuses: on_statuses)
+  end
+
   def access_token
     @access_token ||= ENV.fetch("BASECRM_ACCESS_TOKEN") { raise '"BASECRM_ACCESS_TOKEN" has not been found.'  }
   end