TrueCloudLab/frostfs-testlib
25
0
Fork 19
Code Issues Pull requests 2 Releases 1 Activity Actions 2

Context manager for parralel func #276

Merged
abereziny merged 1 commit from d.zayakin/frostfs-testlib:parallel-context-manager into master 2024-08-05 12:47:30 +00:00
Conversation 0 Commits 1 Files changed 1 +12
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/frostfs_testlib/testing/parallel.py
View file

@ -1,10 +1,22 @@
import itertools import itertools
from concurrent.futures import Future, ThreadPoolExecutor from concurrent.futures import Future, ThreadPoolExecutor
from contextlib import contextmanager
from typing import Callable, Collection, Optional, Union from typing import Callable, Collection, Optional, Union
MAX_WORKERS = 50 MAX_WORKERS = 50
@contextmanager
def parallel_workers_limit(workers_count: int):
global MAX_WORKERS
original_value = MAX_WORKERS
MAX_WORKERS = workers_count
try:
yield
finally:
MAX_WORKERS = original_value
def parallel( def parallel(
fn: Union[Callable, list[Callable]], fn: Union[Callable, list[Callable]],
parallel_items: Optional[Collection] = None, parallel_items: Optional[Collection] = None,