Contents

# frozen_string_literal: true

module ErpIntegration
  module Middleware
    class ApiKeysRotation < Faraday::Middleware
      HTTP_ROTATE_CODES = [404, 429].freeze

      attr_reader :options

      def initialize(app, options = {})
        super(app)

        @options = options
      end

      def call(env)
        on_request(env)

        @app.call(env).on_complete { |response| on_complete(response) }
      end

      def on_request(env)
        env.request_headers['X-API-KEY'] = options[:api_keys_pool].current_key
      end

      def on_complete(env)
        return unless rotate?(env)

        options[:api_keys_pool].rotate_key!
      end

      def rotate?(env)
        return true if HTTP_ROTATE_CODES.include?(env.status)
        return false unless env.response_headers['x-ratelimit-remaining'].present?

        env.response_headers['x-ratelimit-remaining'].to_i < options[:rotation_theshold].to_i
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
erp_integration-0.61.0	lib/erp_integration/middleware/api_keys_rotation.rb
erp_integration-0.58.0	lib/erp_integration/middleware/api_keys_rotation.rb
erp_integration-0.60.0	lib/erp_integration/middleware/api_keys_rotation.rb
erp_integration-0.59.0	lib/erp_integration/middleware/api_keys_rotation.rb
erp_integration-0.57.0	lib/erp_integration/middleware/api_keys_rotation.rb
erp_integration-0.56.0	lib/erp_integration/middleware/api_keys_rotation.rb