ruby-on-rails - 限制每个用户在 rails 中的函数调用

Question

有人知道我该怎么做吗？很难在网上找到信息。我发现最好的是它的宝石，但我只能想到如何实现该应用程序。

Answer 1

它可以通过以下方式处理：1）网络服务器 2）机架应用程序。一切都取决于你需要什么。我们使用内置的 nginx 功能来限制 API 请求：

     limit_req_zone $binary_remote_addr zone=one:10m rate=5r/s;
     limit_req zone=one burst=2;

另一种解决方案是rack-throttle。

这是 Rack 中间件，它提供了对 Rack 应用程序的传入 HTTP 请求进行速率限制的逻辑。您可以将 Rack::Throttle 与任何基于 Rack 的 Ruby Web 框架一起使用，包括与 Ruby on Rails 3.0 和 Sinatra 一起使用。

Answer 2

这是一个如何使用 Redis 和时间戳实现的示例。你会在 user.rb 中包含这个模块，然后你可以调用user.allowed_to?(:reveal_email)

# Lets you limit the number of actions a user can take per time period
# Keeps an integer timestamp with some buffer in the past and each action increments the timestamp
# If the counter exceeds Time.now the action is disallowed and the user must wait for some time to pass.

module UserRateLimiting

  class RateLimit < Struct.new(:max, :per)
    def minimum
      Time.now.to_i - (step_size * max)
    end

    def step_size
      seconds = case per
      when :month  then 18144000 # 60 * 60 * 24 * 7 * 30
      when :week   then 604800   # 60 * 60 * 24 * 7
      when :day    then 86400    # 60 * 60 * 24
      when :hour   then 3600     # 60 * 60
      when :minute then 60
      else raise 'invalid per param (day, hour, etc)'
      end
      seconds / max
    end
  end

  LIMITS = {
    :reveal_email => RateLimit.new(200, :day)
    # add new rate limits here...
  }

  def allowed_to? action
    inc_counter(action) < Time.now.to_i
  end

  private

  def inc_counter action
    rl = LIMITS[action]
    raise "couldn't find that action" if rl.nil?
    val = REDIS_COUNTERS.incrby redis_key(action), rl.step_size
    if val < rl.minimum
      val = REDIS_COUNTERS.set redis_key(action), rl.minimum
    end
    val.to_i
  end

  def redis_key action
    "rate_limit_#{action}_for_user_#{self.id}"
  end

end