glebprop 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in glebprop.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/glebprop.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "glebprop/version"
+
+Gem::Specification.new do |s|
+  s.name        = "glebprop"
+  s.version     = Glebprop::VERSION
+  s.authors     = ["Gleb Pomykalov"]
+  s.email       = ["gleb.pomykalov@skype.net"]
+  s.homepage    = ""
+  s.summary     = %q{Implement rate limiting}
+  s.description = %q{See gem called Prop - this one is just a copy}
+
+  s.rubyforge_project = "glebprop"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+end

data/lib/glebprop/version.rb

@@ -0,0 +1,3 @@
+module Glebprop
+  VERSION = "0.0.1"
+end

data/lib/glebprop.rb

@@ -0,0 +1,6 @@
+require "glebprop/version"
+require 'prop'
+
+module Glebprop
+  # Your code goes here...
+end

data/lib/prop.rb

@@ -0,0 +1,116 @@
+require 'digest/md5'
+
+class Object
+  def define_prop_class_method(name, &blk)
+    (class << self; self; end).instance_eval { define_method(name, &blk) }
+  end
+end
+
+class Prop
+  class RateLimitExceededError < RuntimeError
+    attr_accessor :handle, :retry_after
+
+    def self.create(handle, key, threshold)
+      error = new("#{handle} threshold of #{threshold} exceeded for key '#{key}'")
+      error.handle      = handle
+      error.retry_after = threshold - Time.now.to_i % threshold if threshold > 0
+      raise error
+    end
+  end
+
+  class << self
+    attr_accessor :handles, :reader, :writer
+
+    def read(&blk)
+      self.reader = blk
+    end
+
+    def write(&blk)
+      self.writer = blk
+    end
+
+    def increment(&blk)
+      self.incrementer = blk
+    end
+
+    def incrementer=(value)
+      @incrementer = value
+    end
+
+    def incrementer
+      @incrementer ? Proc.new { |key, inc| @incrementer.call(key, inc) } :
+        Proc.new { |key, inc| self.writer.call(key, (self.reader.call(key) || 0).to_i + inc) }
+    end
+
+    def defaults(handle, defaults)
+      raise RuntimeError.new("Invalid threshold setting") unless defaults[:threshold].to_i > 0
+      raise RuntimeError.new("Invalid interval setting")  unless defaults[:interval].to_i > 0
+
+      self.handles ||= {}
+      self.handles[handle] = defaults
+    end
+
+    def throttle!(handle, key = nil, options = {})
+      options   = sanitized_prop_options(handle, key, options)
+      cache_key = sanitized_prop_key(key, options[:interval])
+      counter   = reader.call(cache_key).to_i
+
+      incrementer.call(cache_key, [ 1, options[:increment].to_i ].max)
+
+      if counter >= options[:threshold]
+        raise Prop::RateLimitExceededError.create(handle, normalize_cache_key(key), options[:threshold])
+      end
+    end
+
+    def throttled?(handle, key = nil, options = {})
+      options   = sanitized_prop_options(handle, key, options)
+      cache_key = sanitized_prop_key(key, options[:interval])
+
+      reader.call(cache_key).to_i >= options[:threshold]
+    end
+
+    def reset(handle, key = nil, options = {})
+      options   = sanitized_prop_options(handle, key, options)
+      cache_key = sanitized_prop_key(key, options[:interval])
+
+      writer.call(cache_key, 0)
+    end
+
+    def query(handle, key = nil, options = {})
+      options   = sanitized_prop_options(handle, key, options)
+      cache_key = sanitized_prop_key(key, options[:interval])
+
+      reader.call(cache_key).to_i
+    end
+
+    private
+
+    # Builds the expiring cache key
+    def sanitized_prop_key(key, interval)
+      window    = (Time.now.to_i / interval)
+      cache_key = "#{normalize_cache_key(key)}/#{ window }"
+      "prop/#{Digest::MD5.hexdigest(cache_key)}"
+    end
+
+    # Sanitizes the option set and sets defaults
+    def sanitized_prop_options(handle, key, options)
+      defaults = (handles || {})[handle] || {}
+      return {
+        :key       => normalize_cache_key(key),
+        :increment => defaults[:increment],
+        :threshold => defaults[:threshold].to_i,
+        :interval  => defaults[:interval].to_i
+      }.merge(options)
+    end
+
+    # Simple key expansion only supports arrays and primitives
+    def normalize_cache_key(key)
+      if key.is_a?(Array)
+        key.map { |part| normalize_cache_key(part) }.join('/')
+      else
+        key.to_s
+      end
+    end
+
+  end
+end

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification 
+name: glebprop
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Gleb Pomykalov
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-09-29 00:00:00 Z
+dependencies: []
+
+description: See gem called Prop - this one is just a copy
+email: 
+- gleb.pomykalov@skype.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Rakefile
+- glebprop.gemspec
+- lib/glebprop.rb
+- lib/glebprop/version.rb
+- lib/prop.rb
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: glebprop
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Implement rate limiting
+test_files: []
+