You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.2 KiB
64 lines
1.2 KiB
# frozen_string_literal: true |
|
|
|
class RateLimiter |
|
include Redisable |
|
|
|
FAMILIES = { |
|
follows: { |
|
limit: 400, |
|
period: 24.hours.freeze, |
|
}.freeze, |
|
|
|
statuses: { |
|
limit: 300, |
|
period: 3.hours.freeze, |
|
}.freeze, |
|
|
|
reports: { |
|
limit: 400, |
|
period: 24.hours.freeze, |
|
}.freeze, |
|
}.freeze |
|
|
|
def initialize(by, options = {}) |
|
@by = by |
|
@family = options[:family] |
|
@limit = FAMILIES[@family][:limit] |
|
@period = FAMILIES[@family][:period].to_i |
|
end |
|
|
|
def record! |
|
count = redis.get(key) |
|
|
|
if count.nil? |
|
redis.set(key, 0) |
|
redis.expire(key, (@period - (last_epoch_time % @period) + 1).to_i) |
|
end |
|
|
|
raise Mastodon::RateLimitExceededError if count.present? && count.to_i >= @limit |
|
|
|
redis.incr(key) |
|
end |
|
|
|
def rollback! |
|
redis.decr(key) |
|
end |
|
|
|
def to_headers(now = Time.now.utc) |
|
{ |
|
'X-RateLimit-Limit' => @limit.to_s, |
|
'X-RateLimit-Remaining' => (@limit - (redis.get(key) || 0).to_i).to_s, |
|
'X-RateLimit-Reset' => (now + (@period - (now.to_i % @period))).iso8601(6), |
|
} |
|
end |
|
|
|
private |
|
|
|
def key |
|
@key ||= "rate_limit:#{@by.id}:#{@family}:#{(last_epoch_time / @period).to_i}" |
|
end |
|
|
|
def last_epoch_time |
|
@last_epoch_time ||= Time.now.to_i |
|
end |
|
end
|
|
|