ruby_rate_limiter 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 440ed0f666d00afdb77283021ab3805b4fa18a08e5a5aea371b04d7b6d6cf299
+  data.tar.gz: 75309c235f17894fbe85bc034dcfb2126431d421e3c11ef089879fcedc280355
+SHA512:
+  metadata.gz: 9ae4e96315c08a5e84cadc97260e0a1c08f9f201e5ef1ecbb9a235718370007bbb69d9f8625103f97ab2868a000403350503c754f2ae984b0b088e93ec6626d1
+  data.tar.gz: 2377e14dd62b12f9e3640781b3be5561a0d6ef0e84301a4d2b52c49928ec2d9f89ac9b510ecb9e467527b34a2bd26c51cf18c132b739ead39e25aae2ba21c663

data/README.md

@@ -0,0 +1,39 @@
+# RateLimiter
+
+TODO: Delete this and the text below, and describe your gem
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/rate_limiter`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+## Installation
+
+TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/rate_limiter. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/rate_limiter/blob/main/CODE_OF_CONDUCT.md).
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the RateLimiter project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/rate_limiter/blob/main/CODE_OF_CONDUCT.md).

data/lib/ruby_rate_limiter/storage/abstract_storage.rb

@@ -0,0 +1,13 @@
+module RubyRateLimiter
+  module Storage
+    class AbstractStorage
+      def get(_key)
+        raise NotImplementedError, 'You must implement the get method'
+      end
+
+      def set(_key, _value)
+        raise NotImplementedError, 'You must implement the set method'
+      end
+    end
+  end
+end

data/lib/ruby_rate_limiter/storage/redis_storage.rb

@@ -0,0 +1,20 @@
+require 'redis'
+require_relative 'abstract_storage'
+
+module RubyRateLimiter
+  module Storage
+    class RedisStorage < AbstractStorage
+      def initialize
+        @redis = Redis.new
+      end
+
+      def get(key)
+        @redis.get(key)
+      end
+
+      def set(key, value)
+        @redis.set(key, value)
+      end
+    end
+  end
+end

data/lib/ruby_rate_limiter/token_bucket.rb

@@ -0,0 +1,61 @@
+require 'forwardable'
+require_relative 'storage/abstract_storage'
+require_relative 'storage/redis_storage'
+
+module RubyRateLimiter
+  class TokenBucket
+    extend Forwardable
+
+    DEFAULT_BUCKET_SIZE = 10
+    DEFAULT_REFILL_RATE = 1
+
+    def_delegators :@storage, :get, :set
+
+    def initialize(user_id, bucket_size: DEFAULT_BUCKET_SIZE, refill_rate: DEFAULT_REFILL_RATE, storage: RateLimiter::Storage::RedisStorage.new)
+      @user_id = user_id
+      @bucket_size = bucket_size
+      @refill_rate = refill_rate
+      @storage = storage
+    end
+
+    def allow_request?
+      refill_tokens      
+      tokens = get_bucket_size
+      return false if tokens < 1
+
+      update_bucket_size(tokens - 1)
+      true
+    end
+
+    private
+
+    def get_bucket_size
+      (@storage.get("#{@user_id}_tokens") || @bucket_size).to_i
+    end
+
+    def get_last_refill_time
+      (@storage.get("#{@user_id}_last_refill") || Time.now.to_f).to_f
+    end
+
+    def update_bucket_size(tokens)
+      @storage.set("#{@user_id}_tokens", tokens)
+    end
+
+    def update_last_refill_time(timestamp)
+      @storage.set("#{@user_id}_last_refill", timestamp)
+    end
+
+    def refill_tokens
+      current_time = Time.now.to_f
+      last_refill_time = get_last_refill_time
+      elapsed_time = current_time - last_refill_time
+
+      new_tokens = (elapsed_time * @refill_rate).to_i
+      return if new_tokens <= 0
+
+      tokens = [get_bucket_size + new_tokens, @bucket_size].min
+      update_bucket_size(tokens)
+      update_last_refill_time(current_time)
+    end
+  end
+end

data/lib/ruby_rate_limiter/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module RateLimiter
+  VERSION = "0.1.0"
+end

data/lib/ruby_rate_limiter.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require_relative "ruby_rate_limiter/version"
+
+module RubyRateLimiter
+  class Error < StandardError; end
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: ruby_rate_limiter
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Mutuba
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-07-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A Ruby gem for rate limiting using the token bucket algorithm.
+email:
+- danielmutubait@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- lib/ruby_rate_limiter.rb
+- lib/ruby_rate_limiter/storage/abstract_storage.rb
+- lib/ruby_rate_limiter/storage/redis_storage.rb
+- lib/ruby_rate_limiter/token_bucket.rb
+- lib/ruby_rate_limiter/version.rb
+homepage: https://github.com/Mutuba/rate-limiter
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/Mutuba/rate-limiter
+  changelog_uri: https://github.com/Mutuba/limiter/blob/main/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.10
+signing_key:
+specification_version: 4
+summary: Rate Limiter using Token Bucket Algorithm
+test_files: []