feat(rate-limits): use redis to support rate-limiting

relay-md · Feb 19, 2024 · d9bd0fb · d9bd0fb
1 parent c1f0966
commit d9bd0fb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/backend/api.py b/backend/api.py
@@ -50,11 +50,14 @@
 )
 
 # Rate limitations
-limiter = Limiter(
+limiter_args = dict(
  key_func=get_remote_address,
  default_limits=[get_config().RATE_LIMITS_DEFAULT],
  storage_uri=get_config().RATE_LIMITS_REDIS,
 )
+if get_config().SESSION_REDIS_URI:
+ limiter_args["storage_uri"] = get_config().SESSION_REDIS_URI
+limiter = Limiter(**limiter_args)
 app.state.limiter = limiter
 app.add_middleware(SlowAPIMiddleware)