erp_integration 0.54.0 → 0.55.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 015dee8d29c0a8bcd763966f25c24b77efb9cd2e7ac4a1555aa104e9a708c2ec
-  data.tar.gz: 92f5a27785677904bd7b0cd60fc5a033195ed99d64d9c16941fbb4d942db8b31
+  metadata.gz: 865d21dfcb8c0caaed6bb6e76151e019424294a04c5a553c68459cbda871278b
+  data.tar.gz: 5da7b05a69bc74a8dcfa508ebe2b7b544e5902b26398b6da552fd139f20170f1
 SHA512:
-  metadata.gz: 526e86fb610c85d0b5c69f36e2cf1b1c396ff36e6f74c2db6d9764eae9eb1f076d9ed7771b5347ac3060bbdbd047777c1901c427351a280bcb26bf49b6edb269
-  data.tar.gz: f876d9874077923090353f7e2c9d6766cae725fe1558345ba98f4d5cfc07495bb272f451205f04c6a957402fdf10bc756d69ad8792a55405c91a07ce856d5e4e
+  metadata.gz: ea5e4f2a3f0e8bec393964d050ef9db305d1b7859032989d6899750306eba46efba5627e741fc469b95c215a23a42e6f6fcdee7f418f1ddf3da2f276cb201629
+  data.tar.gz: 71db26f1be6f150c53f34591d7ca46f7f89805fb65b5f44258c6dac86067bb8752feae8a08c11bf5cc4f6c096e7b3749b511583d1e3cf536921463e0d829e385

data/README.md

@@ -61,6 +61,47 @@ ErpIntegration::SalesOrder.api_keys_pool = '<your-api-key>'
 ErpIntegration::SalesOrder.api_keys_pool = ['<your-api-key1>', '<your-api-key2>']
 ```
 
+### Rate limiting
+
+To manage the number of requests made to avoid rate limiting by the API provider, you can configure rate limiters in the ERP Integration gem. Each rate limiter should implement the `api_key_fragment` and `within_limit` methods.
+
+### Setting Up Rate Limiters
+You can configure multiple rate limiters to be used for HTTP operations on the client.
+
+```ruby
+# config/initializers/erp_integration.rb
+ErpIntegration.configure do |config|
+  config.rate_limiters = [
+    MyRateLimiter.new(api_key_fragment: 'key1'),
+    MyRateLimiter.new(api_key_fragment: 'key2')
+  ]
+end
+```
+
+Each rate limiter must implement the following methods:
+- `api_key_fragment`: This method should return a string that is used to identify the rate limiter by the API key.
+- `within_limit`: This method should `yield` to the block if the rate limit is not exceeded.
+
+### Example Rate Limiter
+Here is an example implementation of a rate limiter:
+
+```ruby
+class MyRateLimiter
+  def initialize(api_key_fragment:)
+    @api_key_fragment = api_key_fragment
+  end
+
+  def api_key_fragment
+    @api_key_fragment
+  end
+
+  def within_limit
+    # Implement your rate limiting logic here
+    yield
+  end
+end
+```
+
 ### Supported Query Methods
 
 After configuring the gem, one can easily query all the available ERP resources from the connected third-parties. In all cases, the API will return a collection of resources.

data/lib/erp_integration/configuration.rb

@@ -299,6 +299,18 @@ module ErpIntegration
     def product_option_adapter
       @product_option_adapter || :fulfil
     end
+
+    # Rate limiters that will be used for HTTP operations on Client
+    # to manage the number of requests made to avoid rate limiting by the API provider.
+    def rate_limiters
+      @rate_limiters ||= []
+    end
+
+    # Sets the rate limiters that will be used
+    # @raise [MissingMethodError] if the rate limiter object doesn't respond to the required methods.
+    def rate_limiters=(rate_limiters)
+      @rate_limiters = rate_limiters.each { |limiter| RateLimiter.validate!(limiter) }
+    end
   end
 
   # Returns ERP Integration's configuration.

data/lib/erp_integration/fulfil/client.rb

@@ -48,7 +48,9 @@ module ErpIntegration
             raise ErpIntegration::Error, 'The Fulfil API key and/or base URL are missing.'
           end
 
-          connection.public_send(action_name, "api/#{version}/#{path}", options).body
+          rate_limiter.within_limit do
+            connection.public_send(action_name, "api/#{version}/#{path}", options).body
+          end
         end
       end
 
@@ -64,6 +66,10 @@ module ErpIntegration
         api_keys_pool.current_key
       end
 
+      def rate_limiter
+        RateLimiter.find_by_api_key(api_key)
+      end
+
       def default_headers
         {
           'Accept': 'application/json',

data/lib/erp_integration/middleware/logger.rb

@@ -43,6 +43,8 @@ module ErpIntegration
         @formatter.exception(exc) if @formatter.respond_to?(:exception)
       end
 
+      # @param [String] api_key
+      # @return [String] The last 4 characters of the API key
       def sanitize_api_key(api_key)
         api_key[-4..-1] if api_key
       end

data/lib/erp_integration/rate_limiter.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+module ErpIntegration
+  class RateLimiter
+    class MissingMethodError < StandardError; end
+    # The `api_key_fragment` method should return a string that is used to identify
+    # the rate limiter by the API key.
+    #
+    # The `within_limit` method should yield to the block if the rate limit is not exceeded.
+    REQUIRED_METHODS = %i[api_key_fragment within_limit].freeze
+
+    class << self
+      # Validates that the rate limiter object responds to the required methods.
+      # It requires the `api_key_fragment` and `within_limit` methods to be present.
+      #
+      # @raise [MissingMethodError]
+      # @param rate_limiter [Object]
+      def validate!(rate_limiter)
+        REQUIRED_METHODS.each do |method|
+          next if rate_limiter.respond_to?(method)
+
+          raise MissingMethodError, "'#{rate_limiter.class}##{method}' method is required."
+        end
+      end
+
+      # Finds the rate limiter by the API key.
+      # If the API key is not found, it returns an unlimited rate limiter.
+      #
+      # @param api_key [String]
+      # @return [Object] The rate limiter object.
+      def find_by_api_key(api_key)
+        rate_limiters[api_key] || unlimited
+      end
+
+      # Returns an unlimited rate limiter.
+      #
+      # @return [RateLimiter::Unlimited] The unlimited rate limiter object.
+      def unlimited
+        @unlimited ||= Unlimited.new
+      end
+
+      private
+
+      # The `rate_limiters` hash stores the rate limiter objects found by the API key.
+      #
+      # @return [Hash] The rate limiters hash.
+      def rate_limiters
+        @rate_limiters ||= Hash.new do |h, api_key|
+          h[api_key] = ErpIntegration.config.rate_limiters.find do |rate_limiter|
+            api_key.end_with?(rate_limiter.api_key_fragment)
+          end
+        end
+      end
+    end
+
+    # {Unlimited} is a rate limiter that allows all requests.
+    # When the rate limiter wasn't found by the API key, it defaults to this class.
+    # Or if the `rate_limiters` option is not set in the configuration.
+    class Unlimited
+      # Executes the block without any restrictions.
+      def within_limit
+        yield
+      end
+    end
+  end
+end

data/lib/erp_integration/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ErpIntegration
-  VERSION = '0.54.0'
+  VERSION = '0.55.0'
 end

data/lib/erp_integration.rb

@@ -11,6 +11,7 @@ require 'json'
 require_relative 'erp_integration/version'
 require_relative 'erp_integration/errors'
 require_relative 'erp_integration/api_keys_pool'
+require_relative 'erp_integration/rate_limiter'
 require_relative 'erp_integration/configuration'
 
 # Middleware

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: erp_integration
 version: !ruby/object:Gem::Version
-  version: 0.54.0
+  version: 0.55.0
 platform: ruby
 authors:
 - Stefan Vermaas
@@ -329,6 +329,7 @@ files:
 - lib/erp_integration/purchase_order.rb
 - lib/erp_integration/purchase_order_line.rb
 - lib/erp_integration/purchase_request.rb
+- lib/erp_integration/rate_limiter.rb
 - lib/erp_integration/resource.rb
 - lib/erp_integration/resources/errors.rb
 - lib/erp_integration/resources/persistence.rb
@@ -368,7 +369,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.11
+rubygems_version: 3.2.22
 signing_key:
 specification_version: 4
 summary: Connects Mejuri with third-party ERP vendors