rate_limitable.rb

module RateLimitable
  extend ActiveSupport::Concern

  class_methods do
    def rate_limiter(name, key:, threshold:, interval:, retry_from: [])
      limiter_name = "#{name}_limiter"
      define_method(limiter_name) do |&block|
        limiter = instance_variable_get("@#{limiter_name}".to_sym)
        limiter ||= instance_variable_set("@#{limiter_name}".to_sym, Ratelimit.new(name))

        limiter.exec_within_threshold(__send__(key), threshold: threshold, interval: interval) do
          result = Retryable.retryable(tries: 3, on: retry_from) { block.call }
          limiter.add __send__(key)
          result
        end
      end
    end
  end
end