Contents

require "faraday/middleware"

module ZendeskAPI
  module Middleware
    module Request
      # Faraday middleware to handle HTTP Status 429 (rate limiting) / 503 (maintenance)
      class Retry < Faraday::Middleware
        DEFAULT_RETRY_AFTER = 10
        ERROR_CODES = [429, 503]

        def initialize(app, options={})
          super(app)
          @logger = options[:logger]
        end

        def call(env)
          response = @app.call(env)

          if ERROR_CODES.include?(response.env[:status])
            seconds_left = (response.env[:response_headers][:retry_after] || DEFAULT_RETRY_AFTER).to_i
            @logger.warn "You have been rate limited. Retrying in #{seconds_left} seconds..." if @logger

            seconds_left.times do |i|
              sleep 1
              time_left = seconds_left - i
              @logger.warn "#{time_left}..." if time_left > 0 && time_left % 5 == 0 && @logger
            end

            @logger.warn "" if @logger

            @app.call(env)
          else
            response
          end
        end
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
zendesk_api-0.1.7	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.1.6	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.1.5	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.1.4	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.1.3	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.1.2	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.1.1	lib/zendesk_api/middleware/request/retry.rb
zendesk_api-0.0.9	lib/zendesk_api/middleware/request/retry.rb