glutton_ratelimit 0.1.0 → 0.2.0

data/examples/simple_manual.rb

@@ -1,30 +1,32 @@
-require 'rubygems'
-require 'glutton_ratelimit'
-
-puts "Maximum of 12 executions every 5 seconds (Bursty):"
-rl = GluttonRatelimit::BurstyTokenBucket.new 12, 5
-
-start = Time.now
-25.times do |n|
-  # BurstyTokenBucket will allow for a full burst of executions followed by a pause.
-  rl.wait
-  puts "#{n+1} - #{Time.now - start}"
-  # Simulating a constant-time task:
-  sleep 0.1
-end
-
-# The 25th execution should occur after 10 seconds has elapsed.
-
-puts "Maximum of 3 executions every 3 seconds (Averaged):"
-rl = GluttonRatelimit::AveragedThrottle.new 3, 3
-
-start = Time.now
-7.times do |n|
-  # AverageThrottle will attempt to evenly space executions within the allowed period.
-  rl.wait
-  puts "#{n+1} - #{Time.now - start}"
-  # Simulating a 0 to 1 second random-time task:
-  sleep rand
-end
-
+# $LOAD_PATH << File.dirname(__FILE__) +'/../lib'
+require 'rubygems'
+require 'glutton_ratelimit'
+
+puts "Maximum of 12 executions every 5 seconds (Bursty):"
+rl = GluttonRatelimit::BurstyTokenBucket.new 12, 5
+
+start = Time.now
+n = 0
+
+rl.times(25) do
+  puts "#{n += 1} - #{Time.now - start}"
+  # Simulating a constant-time task:
+  sleep 0.1
+end
+
+# The 25th execution should occur after 10 seconds has elapsed.
+
+puts "Maximum of 3 executions every 3 seconds (Averaged):"
+rl = GluttonRatelimit::AveragedThrottle.new 3, 3
+# AverageThrottle will attempt to evenly space executions within the allowed period.
+
+start = Time.now
+n = 0
+
+rl.times(7) do 
+  puts "#{n += 1} - #{Time.now - start}"
+  # Simulating a 0 to 1 second random-time task:
+  sleep rand
+end
+
 # The 7th execution should occur after 6 seconds has elapsed.

data/glutton_ratelimit.gemspec

@@ -1,64 +1,64 @@
-# Generated by jeweler
-# DO NOT EDIT THIS FILE DIRECTLY
-# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
-# -*- encoding: utf-8 -*-
-
-Gem::Specification.new do |s|
-  s.name = %q{glutton_ratelimit}
-  s.version = "0.1.0"
-
-  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
-  s.authors = ["Wally Glutton"]
-  s.date = %q{2010-06-10}
-  s.description = %q{A Ruby library for limiting the number of times a method can be invoked within a specified time period.}
-  s.email = %q{stungeye@gmail.com}
-  s.extra_rdoc_files = [
-    "LICENSE",
-     "README.rdoc"
-  ]
-  s.files = [
-    ".document",
-     ".gitignore",
-     "LICENSE",
-     "README.rdoc",
-     "Rakefile",
-     "VERSION",
-     "examples/limit_instance_methods.rb",
-     "examples/simple_manual.rb",
-     "glutton_ratelimit.gemspec",
-     "lib/glutton_ratelimit.rb",
-     "lib/glutton_ratelimit/averaged_throttle.rb",
-     "lib/glutton_ratelimit/bursty_ring_buffer.rb",
-     "lib/glutton_ratelimit/bursty_token_bucket.rb",
-     "test/helper.rb",
-     "test/test_glutton_ratelimit_averaged_throttle.rb",
-     "test/test_glutton_ratelimit_bursty_ring_buffer.rb",
-     "test/test_glutton_ratelimit_bursty_token_bucket.rb",
-     "test/testing_module.rb"
-  ]
-  s.homepage = %q{http://github.com/stungeye/glutton_ratelimit}
-  s.rdoc_options = ["--charset=UTF-8"]
-  s.require_paths = ["lib"]
-  s.rubygems_version = %q{1.3.6}
-  s.summary = %q{Simple Ruby library for self-imposed rater-limiting.}
-  s.test_files = [
-    "test/test_glutton_ratelimit_averaged_throttle.rb",
-     "test/helper.rb",
-     "test/test_glutton_ratelimit_bursty_ring_buffer.rb",
-     "test/testing_module.rb",
-     "test/test_glutton_ratelimit_bursty_token_bucket.rb",
-     "examples/limit_instance_methods.rb",
-     "examples/simple_manual.rb"
-  ]
-
-  if s.respond_to? :specification_version then
-    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
-    s.specification_version = 3
-
-    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
-    else
-    end
-  else
-  end
-end
-
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{glutton_ratelimit}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Wally Glutton"]
+  s.date = %q{2010-06-10}
+  s.description = %q{A Ruby library for limiting the number of times a method can be invoked within a specified time period.}
+  s.email = %q{stungeye@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "examples/limit_instance_methods.rb",
+     "examples/simple_manual.rb",
+     "glutton_ratelimit.gemspec",
+     "lib/glutton_ratelimit.rb",
+     "lib/glutton_ratelimit/averaged_throttle.rb",
+     "lib/glutton_ratelimit/bursty_ring_buffer.rb",
+     "lib/glutton_ratelimit/bursty_token_bucket.rb",
+     "test/helper.rb",
+     "test/test_glutton_ratelimit_averaged_throttle.rb",
+     "test/test_glutton_ratelimit_bursty_ring_buffer.rb",
+     "test/test_glutton_ratelimit_bursty_token_bucket.rb",
+     "test/testing_module.rb"
+  ]
+  s.homepage = %q{http://github.com/stungeye/glutton_ratelimit}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{Simple Ruby library for self-imposed rater-limiting.}
+  s.test_files = [
+    "test/test_glutton_ratelimit_averaged_throttle.rb",
+     "test/helper.rb",
+     "test/test_glutton_ratelimit_bursty_ring_buffer.rb",
+     "test/testing_module.rb",
+     "test/test_glutton_ratelimit_bursty_token_bucket.rb",
+     "examples/limit_instance_methods.rb",
+     "examples/simple_manual.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end
+

data/lib/glutton_ratelimit.rb

@@ -1,29 +1,38 @@
-module GluttonRatelimit
-  
-  def rate_limit symbol, executions, time_period, rl_class = AveragedThrottle
-    rl = rl_class.new executions, time_period
-    old_symbol = "#{symbol}_old".to_sym
-    alias_method old_symbol, symbol
-    define_method symbol do |*args|
-      rl.wait
-      self.send old_symbol, *args
-    end
-  end
-  
-  private
-  # All the other classes extend this parent and are therefore
-  # constructed in the same manner.
-  class ParentLimiter
-    attr_reader :executions
-    
-    def initialize executions, time_period
-      @executions = executions
-      @time_period = time_period
-    end
-  end
-end
-
-dir = File.expand_path(File.dirname(__FILE__))
-require File.join(dir, "glutton_ratelimit", "bursty_ring_buffer")
-require File.join(dir, "glutton_ratelimit", "bursty_token_bucket")
-require File.join(dir, "glutton_ratelimit", "averaged_throttle")
+module GluttonRatelimit
+  
+  def rate_limit symbol, executions, time_period, rl_class = AveragedThrottle
+    rl = rl_class.new executions, time_period
+    old_symbol = "#{symbol}_old".to_sym
+    alias_method old_symbol, symbol
+    define_method symbol do |*args|
+      rl.wait
+      self.send old_symbol, *args
+    end
+  end
+  
+  private
+  # All the other classes extend this parent and are therefore
+  # constructed in the same manner.
+  class ParentLimiter
+    attr_reader :executions
+    
+    def initialize executions, time_period
+      @executions = executions
+      @time_period = time_period
+    end
+    
+    def times(num, &block)
+      raise ArgumentError, "Code block expected"  if not block
+      raise ArgumentError, "Parameter expected to be Fixnum but found a #{num.class}."  unless num.class.equal?(Fixnum)
+      num.times do
+        wait
+        yield
+      end
+    end
+  end
+end
+
+dir = File.expand_path(File.dirname(__FILE__))
+require File.join(dir, "glutton_ratelimit", "bursty_ring_buffer")
+require File.join(dir, "glutton_ratelimit", "bursty_token_bucket")
+require File.join(dir, "glutton_ratelimit", "averaged_throttle")

data/lib/glutton_ratelimit/averaged_throttle.rb

@@ -1,44 +1,44 @@
-module GluttonRatelimit
-  
-  class AveragedThrottle < ParentLimiter
-    
-    def reset_bucket
-      @before_previous_execution = Time.now if @before_previous_execution.nil?
-      @oldest_timestamp = Time.now
-      @tokens = @executions
-      @total_task_time = 0
-    end
-    
-    def executed_this_period
-      @executions - @tokens
-    end
-    
-    def average_task_time
-      @total_task_time.to_f / executed_this_period
-    end
-    
-    def wait
-      reset_bucket if @tokens.nil?
-      
-      now = Time.now
-      delta_since_previous = now - @before_previous_execution
-      @total_task_time += delta_since_previous
-      remaining_time = @time_period - (now - @oldest_timestamp)
-      
-      if @tokens.zero?
-        sleep(remaining_time) if remaining_time > 0
-        reset_bucket
-      elsif executed_this_period != 0
-        throttle = (remaining_time.to_f + delta_since_previous) / (@tokens+1) - average_task_time
-        sleep throttle if throttle > 0
-      end
-      
-      @tokens -= 1
-      @before_previous_execution = Time.now
-    end
-    
-  end
-
-end
-
-
+module GluttonRatelimit
+  
+  class AveragedThrottle < ParentLimiter
+    
+    def reset_bucket
+      @before_previous_execution = Time.now if @before_previous_execution.nil?
+      @oldest_timestamp = Time.now
+      @tokens = @executions
+      @total_task_time = 0
+    end
+    
+    def executed_this_period
+      @executions - @tokens
+    end
+    
+    def average_task_time
+      @total_task_time.to_f / executed_this_period
+    end
+    
+    def wait
+      reset_bucket if @tokens.nil?
+      
+      now = Time.now
+      delta_since_previous = now - @before_previous_execution
+      @total_task_time += delta_since_previous
+      remaining_time = @time_period - (now - @oldest_timestamp)
+      
+      if @tokens.zero?
+        sleep(remaining_time) if remaining_time > 0
+        reset_bucket
+      elsif executed_this_period != 0
+        throttle = (remaining_time.to_f + delta_since_previous) / (@tokens+1) - average_task_time
+        sleep throttle if throttle > 0
+      end
+      
+      @tokens -= 1
+      @before_previous_execution = Time.now
+    end
+    
+  end
+
+end
+
+

data/lib/glutton_ratelimit/bursty_ring_buffer.rb

@@ -1,24 +1,24 @@
-module GluttonRatelimit
-  
-  class BurstyRingBuffer < ParentLimiter
-    
-    def oldest_timestamp
-      @timestamps = Array.new executions, (Time.now - @time_period) if @timestamps.nil?
-      @timestamps[0]
-    end
-    
-    def current_timestamp= new_stamp
-      @timestamps.push(new_stamp).shift
-    end
-      
-    def wait
-      delta = Time.now - oldest_timestamp
-      sleep(@time_period - delta) if delta < @time_period
-      self.current_timestamp = Time.now
-    end
-    
-  end
-
-end
-
-
+module GluttonRatelimit
+  
+  class BurstyRingBuffer < ParentLimiter
+    
+    def oldest_timestamp
+      @timestamps = Array.new executions, (Time.now - @time_period) if @timestamps.nil?
+      @timestamps[0]
+    end
+    
+    def current_timestamp= new_stamp
+      @timestamps.push(new_stamp).shift
+    end
+      
+    def wait
+      delta = Time.now - oldest_timestamp
+      sleep(@time_period - delta) if delta < @time_period
+      self.current_timestamp = Time.now
+    end
+    
+  end
+
+end
+
+

data/lib/glutton_ratelimit/bursty_token_bucket.rb

@@ -1,26 +1,26 @@
-module GluttonRatelimit
-  
-  class BurstyTokenBucket < ParentLimiter
-    
-    def reset_bucket
-      @oldest_timestamp = Time.now
-      @tokens = @executions
-    end
-    
-    def wait
-      reset_bucket if @tokens.nil?
-      
-      if @tokens.zero?
-        delta = Time.now - @oldest_timestamp
-        sleep(@time_period - delta) if delta < @time_period
-        reset_bucket
-      end
-      
-      @tokens -= 1
-    end
-    
-  end
-
-end
-
-
+module GluttonRatelimit
+  
+  class BurstyTokenBucket < ParentLimiter
+    
+    def reset_bucket
+      @oldest_timestamp = Time.now
+      @tokens = @executions
+    end
+    
+    def wait
+      reset_bucket if @tokens.nil?
+      
+      if @tokens.zero?
+        delta = Time.now - @oldest_timestamp
+        sleep(@time_period - delta) if delta < @time_period
+        reset_bucket
+      end
+      
+      @tokens -= 1
+    end
+    
+  end
+
+end
+
+