ruby-limiter 1.0.0 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 17d71e42e079a0ce7f9b5c9a093e742976998db2
-  data.tar.gz: 1fc5d17c9e2abf2034deebb98f8da7179bc49701
+SHA256:
+  metadata.gz: 144e27e5e203e66c7d753c3da5bce267a514de2f32be74849e96104d6450efe6
+  data.tar.gz: 6ec690da1476b519f6cbb6b22ba921b2c8006db09a706553304c0077a692e553
 SHA512:
-  metadata.gz: '0618773b6259592f784ec8e6d982e8772984757319bba95e83a4a8a5706d5f6732f90514774783f8fec51d2a1b593e3c8f74cccbdf5d3755484227b8f0a3f376'
-  data.tar.gz: e70732cd382c742a3f33674c2ee4392d38b322879fa8afdc21a3b9b1fb62107ced1b2fe4487a2de164c9f0d99d42540af8f133e8def0be2b0484da3597d06a68
+  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,31 @@
 # 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
+
+## v1.0.2
+
+- DOCFIX: fix name of gem in README
+- BUGFIX: add ruby-limiter.rb so that it works better with bundler
+
+## v1.0.1
+
+- BUGFIX: support arguments for throttled methods
+
 ## v1.0.0
 
 - initial release

data/README.md

@@ -7,7 +7,7 @@ This gem implements a simple mechanism to throttle or rate-limit operations in R
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'limiter'
+gem 'ruby-limiter'
 ```
 
 And then execute:
@@ -16,13 +16,13 @@ And then execute:
 
 Or install it yourself as:
 
-    $ gem install limiter
+    $ gem install ruby-limiter
 
 ## Usage
 
 ### Basic Usage
 
-To rate limit calling an instance method, a mixin is provided. Simply specify the method to me limited, and the maximum
+To rate limit calling an instance method, a mixin is provided. Simply specify the method to be limited, and the maximum
 rate that the method can be called. This rate is (by default) a number of requests per minute.
 
 ``` ruby
@@ -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 = 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/clock.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require 'singleton'
+require 'forwardable'
+
+module Limiter
+  class Clock
+    include Singleton
+
+    extend SingleForwardable
+    def_single_delegators :instance, :sleep, :time
+
+    def sleep(interval)
+      Kernel.sleep(interval)
+    end
+
+    def time
+      Process.clock_gettime(Process::CLOCK_MONOTONIC)
+    end
+  end
+end

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
+        define_method(method) do |*args, **options, &blk|
           queue.shift
-          super()
+          options.empty? ? super(*args, &blk) : super(*args, **options, &blk)
         end
       end
 

data/lib/limiter/rate_queue.rb

@@ -2,15 +2,16 @@
 
 module Limiter
   class RateQueue
-    EPOCH = Time.at(0)
+    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
@@ -21,7 +22,7 @@ module Limiter
 
         sleep_until(time + @interval)
 
-        @ring[@head] = Time.now
+        @ring[@head] = clock.time
         @head = (@head + 1) % @size
       end
 
@@ -31,9 +32,14 @@ module Limiter
     private
 
     def sleep_until(time)
-      interval = time - Time.now
+      interval = time - clock.time
       return unless interval.positive?
-      sleep(interval)
+      @blk.call if @blk
+      clock.sleep(interval)
+    end
+
+    def clock
+      Clock
     end
   end
 end

data/lib/limiter/version.rb

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

data/lib/limiter.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'limiter/clock'
 require 'limiter/mixin'
 require 'limiter/rate_queue'
 require 'limiter/version'

data/lib/ruby-limiter.rb

@@ -0,0 +1 @@
+require 'limiter'

data/limiter.gemspec

@@ -13,6 +13,13 @@ 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"
+  else
+    raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
+  end
 
   spec.files = `git ls-files -z`.split("\x0").reject do |f|
     f.match(%r{^(test|spec|features)/})
@@ -21,9 +28,10 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = %w(lib)
 
-  spec.add_development_dependency 'bundler', '~> 1.15'
+  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'
-  spec.add_development_dependency 'timecop', '~> 0.8.0'
 end

metadata

@@ -1,29 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: ruby-limiter
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 2.1.0
 platform: ruby
 authors:
 - S. Brent Faulkner
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-07-06 00:00:00.000000000 Z
+date: 2021-08-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.15'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.15'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement
@@ -39,47 +39,61 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '5.0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: minitest-focus
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '1.3'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '1.3'
 - !ruby/object:Gem::Dependency
-  name: rubocop
+  name: mocha
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.56'
+        version: '1.11'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.56'
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
 - !ruby/object:Gem::Dependency
-  name: timecop
+  name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.0
+        version: '0.56'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.0
+        version: '0.56'
 description: 
 email:
 - brent.faulkner@shopify.com
@@ -100,14 +114,17 @@ files:
 - bin/setup
 - dev.yml
 - lib/limiter.rb
+- lib/limiter/clock.rb
 - lib/limiter/mixin.rb
 - lib/limiter/rate_queue.rb
 - lib/limiter/version.rb
+- lib/ruby-limiter.rb
 - limiter.gemspec
 homepage: https://github.com/Shopify/limiter
 licenses:
 - MIT
-metadata: {}
+metadata:
+  allowed_push_host: https://rubygems.org
 post_install_message: 
 rdoc_options: []
 require_paths:
@@ -116,15 +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: []
-rubyforge_project: 
-rubygems_version: 2.6.14
+rubygems_version: 3.2.20
 signing_key: 
 specification_version: 4
 summary: Simple Ruby rate limiting mechanism.