|
@@ -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):
|