add redis for rack:attack

estiens · Jun 16, 2018 · 84c1fa6 · 84c1fa6
1 parent a1a2861
commit 84c1fa6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/config/initializers/rack_attack.rb b/config/initializers/rack_attack.rb
@@ -1,7 +1,10 @@
-Rack::Attack.throttle("requests by ip", limit: 5, period: 2) do |request|
-  request.ip
+if ENV['REDISTOGO_URL']
+  redis_client = Redis.connect(url: ENV['REDISTOGO_URL'])
+  Rack::Attack.cache.store = Rack::Attack::StoreProxy::RedisStoreProxy.new(redis_client)
 end
 
+Rack::Attack.throttle('requests by ip', limit: 5, period: 2, &:ip)
+
 Rack::Attack.throttled_response = lambda do |env|
   now = Time.now
   match_data = env['rack.attack.match_data']
@@ -12,5 +15,5 @@
     'X-RateLimit-Reset' => (now + (match_data[:period] - now.to_i % match_data[:period])).to_s
   }
 
-  [ 429, headers, ["Throttled\n"]]
+  [429, headers, ["Throttled\n"]]
 end