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

View File

@ -18,6 +18,8 @@ async def open_actor_cluster(
modules: list[str],
count: int = cpu_count(),
names: Optional[list[str]] = None,
start_method: Optional[str] = None,
hard_kill: bool = False,
) -> AsyncGenerator[
list[str],
@ -36,7 +38,7 @@ async def open_actor_cluster(
raise ValueError(
'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:
for index, key in zip(range(count), names):
@ -52,4 +54,4 @@ async def open_actor_cluster(
assert len(portals) == count
yield portals
await an.cancel()
await an.cancel(hard_kill=hard_kill)