voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use time.monotonic.

Browse source
This commit is contained in:
voussoir 2021-11-20 18:51:46 -08:00
parent edeff079f7
commit 6d8a406ceb
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
voussoirkit/ratelimiter.py
View file

@ -50,7 +50,7 @@ class Ratelimiter:
self.mode = mode self.mode = mode
self.lock = threading.Lock() self.lock = threading.Lock()
self.last_operation = time.time() self.last_operation = time.monotonic()
self.balance = 0 self.balance = 0
@property @property
@ -58,7 +58,8 @@ class Ratelimiter:
return self.allowance / self.period return self.allowance / self.period
def _limit(self, cost): def _limit(self, cost):
time_diff = time.time() - self.last_operation now = time.monotonic()
time_diff = now - self.last_operation
self.balance += time_diff * self.gain_rate self.balance += time_diff * self.gain_rate
self.balance = min(self.balance, self.allowance) self.balance = min(self.balance, self.allowance)
@ -76,7 +77,7 @@ class Ratelimiter:
self.balance = 0 self.balance = 0
successful = True successful = True
self.last_operation = time.time() self.last_operation = now
return successful return successful
def limit(self, cost=None): def limit(self, cost=None):