jduff-api-throttling 0.1.0 → 0.1.1

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
 :major: 0
 :minor: 1
-:patch: 0
+:patch: 1

data/lib/handlers/handlers.rb

@@ -33,11 +33,11 @@ module Handlers
     end
   end
   
-  %w(redis_handler memcache_handler).each do |handler|
+  %w(redis_handler memcache_handler hash_handler).each do |handler|
     require File.expand_path(File.dirname(__FILE__) + "/#{handler}")
   end
   
-  HANDLERS = [RedisHandler, MemCacheHandler]
+  HANDLERS = [RedisHandler, MemCacheHandler, HashHandler]
   
   def self.cache_handler_for(info)
     HANDLERS.detect{|handler| handler.handles?(info)}

data/lib/handlers/hash_handler.rb

@@ -0,0 +1,13 @@
+module Handlers
+  class HashHandler < Handler
+    cache_class "Hash"
+    
+    def increment(key)
+      @cache[key] = (get(key)||0).to_i+1
+    end
+    
+    def get(key)
+      @cache[key]
+    end
+  end
+end

data/test/test_api_throttling_hash.rb

@@ -0,0 +1,23 @@
+require File.expand_path(File.dirname(__FILE__) + '/test_helper')
+
+class TestApiThrottlingHash < Test::Unit::TestCase
+  include Rack::Test::Methods
+  include BasicTests
+  HASH = Hash.new
+
+  def app
+    app = Rack::Builder.new {
+      use ApiThrottling, :requests_per_hour => 3, :cache => HASH, :read_method=>"get", :write_method=>"add"
+      run lambda {|env| [200, {'Content-Type' =>  'text/plain', 'Content-Length' => '12'}, ["Hello World!"] ] }
+    }
+  end
+  
+  def setup
+    HASH.replace({})
+  end
+  
+  def test_cache_handler_should_be_memcache
+    assert_equal "Handlers::HashHandler", app.to_app.instance_variable_get(:@handler).to_s
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jduff-api-throttling
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors: 
 - Luc Castera
@@ -31,9 +31,11 @@ files:
 - VERSION.yml
 - lib/api_throttling.rb
 - lib/handlers/handlers.rb
+- lib/handlers/hash_handler.rb
 - lib/handlers/memcache_handler.rb
 - lib/handlers/redis_handler.rb
 - test/test_api_throttling.rb
+- test/test_api_throttling_hash.rb
 - test/test_api_throttling_memcache.rb
 - test/test_handlers.rb
 - test/test_helper.rb
@@ -65,6 +67,7 @@ specification_version: 2
 summary: Rack Middleware to impose a rate limit on a web service (aka API Throttling)
 test_files: 
 - test/test_api_throttling.rb
+- test/test_api_throttling_hash.rb
 - test/test_api_throttling_memcache.rb
 - test/test_handlers.rb
 - test/test_helper.rb