Intermediate commit
This commit is contained in:
parent
0e9e49df16
commit
3d4cfeff1a
65 changed files with 5507 additions and 333 deletions
|
|
@ -7,12 +7,28 @@ _REQUEST_LOCK = asyncio.Lock()
|
|||
_LAST_REQUEST_AT = 0.0
|
||||
|
||||
|
||||
def _normalize_interval_seconds(value: float) -> float:
|
||||
return max(float(value), 0.0)
|
||||
|
||||
|
||||
def remaining_scholar_slot_seconds(*, min_interval_seconds: float) -> float:
|
||||
interval_seconds = _normalize_interval_seconds(min_interval_seconds)
|
||||
if interval_seconds <= 0:
|
||||
return 0.0
|
||||
elapsed_seconds = time.monotonic() - _LAST_REQUEST_AT
|
||||
return max(interval_seconds - elapsed_seconds, 0.0)
|
||||
|
||||
|
||||
async def wait_for_scholar_slot(*, min_interval_seconds: float) -> None:
|
||||
global _LAST_REQUEST_AT
|
||||
interval = max(float(min_interval_seconds), 0.0)
|
||||
interval = _normalize_interval_seconds(min_interval_seconds)
|
||||
async with _REQUEST_LOCK:
|
||||
elapsed = time.monotonic() - _LAST_REQUEST_AT
|
||||
remaining = interval - elapsed
|
||||
remaining = remaining_scholar_slot_seconds(min_interval_seconds=interval)
|
||||
if remaining > 0:
|
||||
await asyncio.sleep(remaining)
|
||||
_LAST_REQUEST_AT = time.monotonic()
|
||||
|
||||
|
||||
def reset_scholar_rate_limit_state_for_tests() -> None:
|
||||
global _LAST_REQUEST_AT
|
||||
_LAST_REQUEST_AT = 0.0
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue