rollie 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1a4104ef316b1bf29321667fa9ea4e44dd642f4e
+  data.tar.gz: c695f97168e7ea2d7f47fe0a118bef23d8f5e24e
+SHA512:
+  metadata.gz: 7f9da971ea951921729261a1920e640dfa9b6d1b8fc4580bb87454e6d7657a90583a52b17005a38ee61fdbfaa0dd7d38501e91cea39e9feefd7bbaf73e5c3068
+  data.tar.gz: a3eb75347541a6092ec9a8bfc085dbb5e0368560abf6dd143892d625b6be6f51e3b7693f208b08a7dcc2a287086d4922ad4d2d7afc6ba90d70e5f1924d0a31fe

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2016 Zach Davis
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.md

@@ -0,0 +1,5 @@
+# Rollie
+
+Rollie is a generic rate limiter backed by Redis for efficient limiting using sliding windows.
+
+[![Build Status](https://travis-ci.org/zldavis/rollie.svg?branch=master)](https://travis-ci.org/zldavis/rollie)

data/lib/rollie.rb

@@ -0,0 +1,28 @@
+require "rollie/rate_limiter"
+require "rollie/redis_pool"
+require "rollie/status"
+require "rollie/version"
+
+module Rollie
+  class << self
+    def redis
+      raise ArgumentError, "requires a block" unless block_given?
+      redis_pool.with do |conn|
+        yield(conn)
+      end
+    end
+
+    def redis=(hash)
+      @redis_pool = if hash.is_a?(ConnectionPool)
+         hash
+       else
+         Rollie::RedisPool.create(hash)
+       end
+    end
+
+    def redis_pool
+      @redis_pool ||= Rollie::RedisPool.create
+    end
+  end
+
+end

data/lib/rollie/rate_limiter.rb

@@ -0,0 +1,42 @@
+module Rollie
+
+  class RateLimiter
+
+    def initialize(key, options = {})
+      @key = "#{options[:namespace]}#{key}"
+      @limit = options[:limit] || 25
+      @interval = (options[:interval] || 1000) * 1000
+    end
+
+    def within_limit
+      raise ArgumentError, "requires a block" unless block_given?
+
+      Rollie.redis do |conn|
+        status = inc(conn)
+        unless status.exceeded?
+          yield
+        end
+        status
+      end
+    end
+
+    private
+
+    def inc(conn)
+      time = (Time.now.to_r * 1000000).round
+      old = time - @interval
+      range = conn.multi do
+        conn.zremrangebyscore(@key, 0, old)
+        conn.zadd(@key, time, time)
+        conn.zrange(@key, 0, -1)
+        conn.expire(@key, (@interval / 1000000).ceil)
+      end[2]
+
+      exceeded = range.length > @limit
+      time_remaining = range.first.to_i - time + @interval
+      current_count = exceeded ? @limit : range.length
+      Rollie::Status.new((time_remaining / 1000).floor, current_count, exceeded)
+    end
+
+  end
+end

data/lib/rollie/redis_pool.rb

@@ -0,0 +1,37 @@
+require "connection_pool"
+require "redis"
+require "redis-namespace"
+
+module Rollie
+  class RedisPool
+    class << self
+
+      def create(options={})
+        puts "initialized with options: #{options}"
+        pool_size = options[:pool_size] || 5
+        pool_timeout = options[:pool_timeout] || 1
+
+        ConnectionPool.new(:timeout => pool_timeout, :size => pool_size) do
+          build_client(options)
+        end
+      end
+
+      private
+
+      def build_client(options)
+        namespace = options[:namespace] || "Rollie"
+        client = Redis.new redis_options(options)
+        Redis::Namespace.new(namespace, redis: client)
+      end
+
+      def redis_options(options)
+        redis = options.dup
+        redis[:url] ||= ENV["REDIS_URL"]
+        redis[:driver] ||= "ruby"
+        redis.delete(:namespace)
+        redis
+      end
+
+    end
+  end
+end

data/lib/rollie/status.rb

@@ -0,0 +1,15 @@
+module Rollie
+  class Status
+    attr_accessor :time_remaining, :count
+
+    def initialize(time_remaining, count, exceeded)
+      @time_remaining = time_remaining
+      @count = count
+      @exceeded = exceeded
+    end
+
+    def exceeded?
+      @exceeded
+    end
+  end
+end

data/lib/rollie/version.rb

@@ -0,0 +1,3 @@
+module Rollie
+  VERSION = "0.0.1".freeze
+end

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification
+name: rollie
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Zach Davis
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-12-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.2.1
+- !ruby/object:Gem::Dependency
+  name: redis-namespace
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+- !ruby/object:Gem::Dependency
+  name: connection_pool
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.2.0
+description: Generic rate limiter backed by Redis for efficient limiting using sliding
+  windows.
+email: zldavis@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- README.md
+- LICENSE
+files:
+- LICENSE
+- README.md
+- lib/rollie.rb
+- lib/rollie/rate_limiter.rb
+- lib/rollie/redis_pool.rb
+- lib/rollie/status.rb
+- lib/rollie/version.rb
+homepage: https://github.com/zldavis/rollie
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2
+signing_key: 
+specification_version: 4
+summary: Generic rate limiter backed by Redis for efficient limiting using sliding
+  windows.
+test_files: []