zuul
/
nodepool
Code Issues Proposed changes

Merge "Measure rate limits from start times"

This commit is contained in:
Zuul 2022-05-11 05:58:08 +00:00 committed by Gerrit Code Review
parent 46c9e01ab8 a62060a14e
commit 778d912e4e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodepool/driver/utils.py
View File

@ -445,16 +445,18 @@ class RateLimiter:
if self.last_ts is None:
return total_delay
while True:
delta = time.monotonic() - self.last_ts
now = time.monotonic()
delta = now - self.last_ts
if delta >= self.delta:
break
delay = self.delta - delta
time.sleep(delay)
total_delay += delay
self.last_ts = time.monotonic()
return total_delay
def __exit__(self, etype, value, tb):
self._exit(etype, value, tb)
def _exit(self, etype, value, tb):
self.last_ts = time.monotonic()
pass