|
|
|
@@ -40,14 +40,10 @@ class RateLimit: |
|
|
|
self.last_recalculate_time = time.time() |
|
|
|
# flush max active requests |
|
|
|
if use_local_value or not redis_client.exists(self.max_active_requests_key): |
|
|
|
with redis_client.pipeline() as pipe: |
|
|
|
pipe.set(self.max_active_requests_key, self.max_active_requests) |
|
|
|
pipe.expire(self.max_active_requests_key, timedelta(days=1)) |
|
|
|
pipe.execute() |
|
|
|
redis_client.setex(self.max_active_requests_key, timedelta(days=1), self.max_active_requests) |
|
|
|
else: |
|
|
|
with redis_client.pipeline() as pipe: |
|
|
|
self.max_active_requests = int(redis_client.get(self.max_active_requests_key).decode("utf-8")) |
|
|
|
redis_client.expire(self.max_active_requests_key, timedelta(days=1)) |
|
|
|
self.max_active_requests = int(redis_client.get(self.max_active_requests_key).decode("utf-8")) |
|
|
|
redis_client.expire(self.max_active_requests_key, timedelta(days=1)) |
|
|
|
|
|
|
|
# flush max active requests (in-transit request list) |
|
|
|
if not redis_client.exists(self.active_requests_key): |