Skip to content

An easy, distributed rate limiter using Redis

License

Notifications You must be signed in to change notification settings

tripscout-eng/redrate

 
 

Repository files navigation

Redrate Build Status

A distributed rate limiter for Ruby, based on ruby-limiter. By using redis as the ring store, Redrate can enforce the rate limit across multiple running instances of the same code, e.g., to enforce limits on shared use of a common rate-limited API or other resource.

Installation

Add this line to your application's Gemfile:

gem 'redrate'

And then execute:

$ bundle

Or install it yourself as:

$ gem install redrate

Usage

Basic Usage

To rate limit calling an instance method, a mixin is provided. Simply specify the method to me limited, and the maximum rate that the method can be called. This rate is (by default) a number of requests per minute.

class Widget
  extend Redrate::Mixin

  # limit the rate we can call tick to 300 times per minute
  #
  # when the rate has been exceeded, a call to tick will block 
  # until the rate limit would not be exceeded
  limit_method :tick, rate: 300

  ...
end

To specify the rate in terms of an interval shorter (or longer) than 1 minute, an optional interval parameter can be provided to specify the throttling period in seconds.

class Widget
  extend Redrate::Mixin

  # limit the rate we can call tick to 5 times per second
  #
  # when the rate has been exceeded, a call to tick will block 
  # until the rate limit would not be exceeded
  limit_method :tick, rate: 5, interval: 1

  ...
end

Advanced Usage

In cases where the mixin is not appropriate the RedRate::Queue class can be used directly. As in the mixin examples above, the interval parameter is optional (and defaults to 1 minute).

class Widget
  def initialize
    # create a rate-limited queue which allows 10000 operations per hour
    @queue = RedRate::Queue.new(10000, interval: 3600)
  end

  def tick
    # this operation will block until less than 10000 shift calls have been 
    # made within the last hour
    @queue.shift
    # do something
  end
end

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 tags, and push the .gem file to rubygems.org.

Contributing

Bug reports and pull requests are welcome on GitHub at https://github.com/nulib/redrate.

License

The gem is available as open source under the terms of the MIT License.

About

An easy, distributed rate limiter using Redis

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Ruby 98.5%
  • Shell 1.5%