ruby-limiter 1.1.0 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5608ca6c1f3eafc6056387bb8e2addc54af6e8e0bc17394703d31fff14374f87
-  data.tar.gz: a30d6a8ca052fc4621575cbbcb1100e0de534a6fdd4d1ce293f9b23b67e9c1b4
+  metadata.gz: 144e27e5e203e66c7d753c3da5bce267a514de2f32be74849e96104d6450efe6
+  data.tar.gz: 6ec690da1476b519f6cbb6b22ba921b2c8006db09a706553304c0077a692e553
 SHA512:
-  metadata.gz: 0eb16ba74c5d5731e90cc094f707f294d47c67408955fde22b049ad078b6600d24a8e9215616b3a05853528066716df56052423b9d9e2631149636163c6411cb
-  data.tar.gz: dbe54f4cec3e94188f813c68ae9a91d6692d33a33a289394138d67a5d5a77ebb636cad98574d8fd655706243b9702f189541819ccd62f52724970f2016496d90
+  metadata.gz: e4568e49199cc0a518439c808bc28e3281bdc44994495a0bee5e379c40c0de29bf45deee1e55d16d4f80df86424f26411c2f8b0fe13af605d18f8eb4b08c61db
+  data.tar.gz: 385abafc6d5e638829c0cd278a311b1d997a5c2b41e1010ae4b46cac5772498906f5254ab44777d353f36ef7ca228ce134ee2d6b91ceae4310384c4d9d5809e7

data/.rubocop.yml

@@ -2,4 +2,4 @@ inherit_from:
   - https://shopify.github.io/ruby-style-guide/rubocop.yml
 
 AllCops:
-  TargetRubyVersion: 2.3
+  TargetRubyVersion: 2.7

data/.travis.yml

@@ -1,6 +1,6 @@
 sudo: false
 language: ruby
 rvm:
-  - 2.3
-  - 2.4
-  - 2.5
+  - 2.6
+  - 2.7
+  - 3.0

data/CHANGELOG.md

@@ -1,5 +1,18 @@
 # CHANGELOG
 
+## v2.1.0
+
+- add support to call a block when limiting takes place
+
+## v2.0.1
+
+- eliminate kwarg warning in ruby 2.7 (while still supporting 2.6)
+
+## v2.0.0
+
+- end support for ruby 2.3/2.4/2.5
+- test on ruby 2.6/2.7/3.0 (using ruby 2.7 for development)
+
 ## v1.1.0
 
 - using Process.clock_gettime(Process::CLOCK_MONOTONIC) instead of Time.now for improved accuracy

data/README.md

@@ -46,7 +46,10 @@ class Widget
 
   # limit the rate we can call tick to 5 times per second
   # when the rate has been exceeded, a call to tick will block until the rate limit would not be exceeded
-  limit_method :tick, rate: 5, interval: 1
+  # and the provided block will be executed
+  limit_method(:tick, rate: 5, interval: 1) do
+    puts 'Limit reached'
+  end
 
   ...
 end
@@ -55,13 +58,16 @@ end
 ### Advanced Usage
 
 In cases where the mixin is not appropriate the `RateQueue` class can be used directly. As in the mixin examples above,
-the `interval` parameter is optional (and defaults to 1 minute).
+the `interval` parameter is optional (and defaults to 1 minute). It is also possible
+to provide the block to `RateQueue`, which will be executed on each limit hit (useful for metrics).
 
 ``` ruby
 class Widget
   def initialize
     # create a rate-limited queue which allows 10000 operations per hour
-    @queue = Limiter::RateQueue.new(10000, interval: 3600)
+    @queue = Limiter::RateQueue.new(10000, interval: 3600) do
+      puts "Hit the limit, waiting"
+    end
   end
 
   def tick

data/Rakefile

@@ -7,6 +7,7 @@ Rake::TestTask.new(:test) do |t|
   t.libs << "test"
   t.libs << "lib"
   t.test_files = FileList["test/**/*_test.rb"]
+  t.warning = true
 end
 
-task default: :test
+task default: :test

data/dev.yml

@@ -3,7 +3,7 @@
 name: limiter
 
 up:
-  - ruby: 2.4.4
+  - ruby: 2.7.3
   - bundler
 
 commands:

data/lib/limiter/mixin.rb

@@ -2,13 +2,13 @@
 
 module Limiter
   module Mixin
-    def limit_method(method, rate:, interval: 60)
-      queue = RateQueue.new(rate, interval: interval)
+    def limit_method(method, rate:, interval: 60, &b)
+      queue = RateQueue.new(rate, interval: interval, &b)
 
       mixin = Module.new do
-        define_method(method) do |*args|
+        define_method(method) do |*args, **options, &blk|
           queue.shift
-          super(*args)
+          options.empty? ? super(*args, &blk) : super(*args, **options, &blk)
         end
       end
 

data/lib/limiter/rate_queue.rb

@@ -4,13 +4,14 @@ module Limiter
   class RateQueue
     EPOCH = 0.0
 
-    def initialize(size, interval: 60)
+    def initialize(size, interval: 60, &blk)
       @size = size
       @interval = interval
 
       @ring = Array.new(size, EPOCH)
       @head = 0
       @mutex = Mutex.new
+      @blk = blk
     end
 
     def shift
@@ -33,6 +34,7 @@ module Limiter
     def sleep_until(time)
       interval = time - clock.time
       return unless interval.positive?
+      @blk.call if @blk
       clock.sleep(interval)
     end
 

data/lib/limiter/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Limiter
-  VERSION = '1.1.0'
+  VERSION = '2.1.0'
 end

data/limiter.gemspec

@@ -13,6 +13,7 @@ Gem::Specification.new do |spec|
   spec.summary       = 'Simple Ruby rate limiting mechanism.'
   spec.homepage      = 'https://github.com/Shopify/limiter'
   spec.license       = 'MIT'
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.6.0")
 
   if spec.respond_to?(:metadata)
     spec.metadata['allowed_push_host'] = "https://rubygems.org"
@@ -29,6 +30,7 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'minitest', '~> 5.0'
+  spec.add_development_dependency 'minitest-focus', '~> 1.3'
   spec.add_development_dependency 'mocha', '~> 1.11'
   spec.add_development_dependency 'rake', '~> 10.0'
   spec.add_development_dependency 'rubocop', '~> 0.56'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-limiter
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 2.1.0
 platform: ruby
 authors:
 - S. Brent Faulkner
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-04-28 00:00:00.000000000 Z
+date: 2021-08-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: minitest-focus
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
 - !ruby/object:Gem::Dependency
   name: mocha
   requirement: !ruby/object:Gem::Requirement
@@ -119,14 +133,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.6.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.2.20
 signing_key: 
 specification_version: 4
 summary: Simple Ruby rate limiting mechanism.