jc211
/
tractor
forked from goodboy/tractor
1
0
Fork 0
Code Pull Requests Activity

Allow specifying start_method and hard_kill

patch-async-enter-all
overclockworked64 2021-10-17 14:40:09 +02:00
parent b17bdbfa7b
commit f4af27953e
No known key found for this signature in database
GPG Key ID: 0E4A3AD9DB596812
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tractor/_clustering.py
View File

@ -18,6 +18,8 @@ async def open_actor_cluster(
modules: list[str], modules: list[str],
count: int = cpu_count(), count: int = cpu_count(),
names: Optional[list[str]] = None, names: Optional[list[str]] = None,
start_method: Optional[str] = None,
hard_kill: bool = False,
) -> AsyncGenerator[ ) -> AsyncGenerator[
list[str], list[str],
@ -36,7 +38,7 @@ async def open_actor_cluster(
raise ValueError( raise ValueError(
'Number of names is {len(names)} but count it {count}') 'Number of names is {len(names)} but count it {count}')
async with tractor.open_nursery() as an: async with tractor.open_nursery(start_method=start_method) as an:
async with trio.open_nursery() as n: async with trio.open_nursery() as n:
for index, key in zip(range(count), names): for index, key in zip(range(count), names):
@ -52,4 +54,4 @@ async def open_actor_cluster(
assert len(portals) == count assert len(portals) == count
yield portals yield portals
await an.cancel() await an.cancel(hard_kill=hard_kill)