rt/wrolpi
1
0
Fork 0
mirror of https://github.com/lrnselfreliance/wrolpi.git synced 2026-10-26 19:01:39 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding throw argument to limit_concurrent.

Browse source
This commit is contained in:
Roland 2022-10-27 17:41:38 -06:00
parent 3c9d9ea402
commit 034b290712
No known key found for this signature in database
GPG key ID: 9D6DBD3ACF11AB93
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
wrolpi/common.py
View file

@ -946,10 +946,8 @@ def print_timer():
f' {calls} calls', file=sys.stderr)
def limit_concurrent(limit: int):
"""
Wrapper that limits the amount of concurrently running functions.
"""
def limit_concurrent(limit: int, throw: bool = False):
"""Wrapper that limits the amount of concurrently running functions."""
sema = multiprocessing.Semaphore(value=limit)
def wrapper(func: callable):
@ -958,6 +956,8 @@ def limit_concurrent(limit: int):
async def wrapped(*a, **kw):
acquired = sema.acquire(block=False)
if not acquired:
if throw:
raise ValueError(f'Reached concurrent limit! {func=}')
return
try:
return await func(*a, **kw)
@ -970,6 +970,8 @@ def limit_concurrent(limit: int):
def wrapped(*a, **kw):
acquired = sema.acquire(block=False)
if not acquired:
if throw:
raise ValueError(f'Reached concurrent limit! {func=}')
return
try:
return func(*a, **kw)