galago-rate_limiter 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f6e0d5c290a7a02a64fea368a8e9e751555646f1
-  data.tar.gz: eacb1622bc6fe97c08248ba25a0b33f9f8a90bad
+  metadata.gz: 0080799d8124b3365796777bfb21d0eb11724066
+  data.tar.gz: cf65551e8e8e21c38b26184c663822f3f5cb0890
 SHA512:
-  metadata.gz: 63bd3d93e8585df34f194ada5de47562bb803d54bdc9ea4ef455fdce6589b782f9164667f091930d6b6e7af1a4305f020e2adb8a3c31fe3655d7e9b80fc949f8
-  data.tar.gz: 77d1fff80116a15d6df892e61f0099dbeb55b2dc10eddf0cd0b2e6620779fec27b1aac7893114eb85d801b75d939e51de07d305cd0688a4e7450d7270cb7cd39
+  metadata.gz: 1728d0d5a24fcf6eb939b2c41fff0bc2b87c93ccc5a565b988c0c36f138e9394629ef3a3b76351d20298f68f92446d2dd2f02aaed76b8a49ecd444b4556f7783
+  data.tar.gz: 8f616cf90aab44f5182a21e0e898bd630a8cf2f46edf955e07f2794531668e29c2751db0423c86789b05e3f5753d029768480a309d804afeb00b5477fa3a4212

data/lib/galago/rate_limiter/configuration.rb

@@ -7,12 +7,14 @@ module Galago
 
       DEFAULT_LIMIT = 5_000
       DEFAULT_API_KEY_HEADER = 'HTTP_X_API_KEY'.freeze
+      DEFAULT_CALLBACK = Proc.new {}
 
       attr_reader :api_key_header, :limit, :counter
 
       def initialize
         @limit = DEFAULT_LIMIT
         @api_key_header = DEFAULT_API_KEY_HEADER
+        @callback = DEFAULT_CALLBACK
       end
 
       def limit=(limit)
@@ -36,6 +38,14 @@ module Galago
                    end
       end
 
+      def callback(&block)
+        if block_given?
+          @callback = block
+        else
+          @callback
+        end
+      end
+
       def reset!
         @limit = DEFAULT_LIMIT
         @api_key_header = DEFAULT_API_KEY_HEADER

data/lib/galago/rate_limiter.rb

@@ -26,6 +26,7 @@ module Galago
       throughput = @counter.increment(api_key, 1, expires_in: expires_in)
 
       if limit_exceeded?(throughput)
+        execute_callback(api_key)
         status = 403
         headers = {
           X_LIMIT_HEADER => @config.limit.to_s,
@@ -45,6 +46,10 @@ module Galago
 
     private
 
+    def execute_callback(api_key)
+      @config.callback.call(api_key)
+    end
+
     def limit_exceeded?(throughput)
       throughput > @config.limit
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: galago-rate_limiter
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Joe Karayusuf
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-22 00:00:00.000000000 Z
+date: 2015-03-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: dalli
   requirement: !ruby/object:Gem::Requirement