33 lines
692 B
Ruby
33 lines
692 B
Ruby
|
class MailRateLimiter
|
||
|
|
attr_reader :delay, :current_window
|
||
|
|
|
||
|
|
def send_with_delay(mail)
|
||
|
|
if current_window_full?
|
||
|
|
@delay += @window
|
||
|
|
end
|
||
|
|
if current_window_full? || current_window_expired?
|
||
|
|
@current_window = { started_at: Time.current, sent: 0 }
|
||
|
|
end
|
||
|
|
@current_window[:sent] += 1
|
||
|
|
|
||
|
|
mail.deliver_later(wait: delay)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def initialize(limit:, window:)
|
||
|
|
@limit = limit
|
||
|
|
@window = window
|
||
|
|
@current_window = { started_at: Time.current, sent: 0 }
|
||
|
|
@delay = 0
|
||
|
|
end
|
||
|
|
|
||
|
|
def current_window_expired?
|
||
|
|
@current_window[:started_at] + @window <= Time.zone.now.utc
|
||
|
|
end
|
||
|
|
|
||
|
|
def current_window_full?
|
||
|
|
@current_window[:sent] == @limit
|
||
|
|
end
|
||
|
|
end
|