Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/viafintech/gcra-ruby

Generic cell rate algorithm (leaky bucket) implementation for rate limiting
https://github.com/viafintech/gcra-ruby

rate-limiting redis ruby-gem

Last synced: 7 days ago
JSON representation

Generic cell rate algorithm (leaky bucket) implementation for rate limiting

Awesome Lists containing this project

README

        

# GCRA for Ruby

![Build Status](https://github.com/viafintech/gcra-ruby/actions/workflows/test.yml/badge.svg) [![Code Climate](https://codeclimate.com/github/Barzahlen/gcra-ruby/badges/gpa.svg)](https://codeclimate.com/github/Barzahlen/gcra-ruby) [![RubyDoc](https://img.shields.io/badge/ruby-doc-green.svg)](http://rubydoc.info/github/Barzahlen/gcra-ruby)

`gcra` is a Ruby implementation of a [generic cell rate algorithm](https://en.wikipedia.org/wiki/Generic_cell_rate_algorithm) (GCRA), ported from and data-format compatible with the Go implementation [throttled](https://github.com/throttled/throttled). It's useful for rate limiting (e.g. for HTTP requests) and allows weights specified per request.

## Getting Started

gcra currently uses Redis (>= 2.6.12 for EVALSHA, SCRIPT LOAD, SEX with NX and PX) as a data store, although it supports other store implementations.

Add to your `Gemfile`:

```ruby
gem 'gcra'
gem 'redis'
```

Create Redis, RedisStore and RateLimiter instances:

```ruby
require 'redis'
require 'gcra/rate_limiter'
require 'gcra/redis_store'

redis = Redis.new(host: 'localhost', port: 6379, timeout: 0.1)
key_prefix = 'rate-limit-app1:'
store = GCRA::RedisStore.new(
redis,
key_prefix,
{ reconnect_on_readonly: false },
)

rate_period = 0.5 # Two requests per second
max_burst = 10 # Allow 10 additional requests as a burst
limiter = GCRA::RateLimiter.new(store, rate_period, max_burst)
```

* `rate_period`: Period between two requests, allowed as a sustained rate. Example: 0.1 for 10 requests per second
* `max_burst`: Number of requests allowed as a burst in addition to the sustained rate. If the burst is used up, one additional request allowed as burst 'comes back' after each `rate_period` where no request was made.

Rate limit a request (call this before each request):

```ruby
key = '123' # e.g. an account identifier
quantity = 1 # the number of requests 'used up' by this request, useful e.g. for batch requests

exceeded, info = limiter.limit(key, quantity)
# => [false, #]
```

* `exceeded`: `false` means the request should be allowed, `true` means the request would exceed the limit and should be blocked.
* `info`: `GCRA::RateLimitInfo` contains information that might be useful for your API users. It's a `Struct` with the following fields:
- `limit`: Contains the number of requests that can be made if no previous requests have been made (or they were long enough ago). That's `max_burst` plus one. The latter is necessary so requests are allowed at all when `max_burst` is set to zero.
- `remaining`: The number of remaining requests that can be made immediately, i.e. the remaining burst.
- `reset_after`: The time in seconds until the full burst will be available again.
- `retry_after`: Set to `nil` if a request is allowed or it otherwise doesn't make sense to retry a request (if `quantity` is larger than `max_burst`). For a blocked request that can be retried later, set to the duration in seconds until the next request with the given quantity will be allowed.

`RateLimiter#limit` only tells you whether to limit a request or not. You'll have to react to its response yourself and e.g. return an error message and stop processing a request if the limit was exceeded.

## License

[MIT](LICENSE)