forked from goodboy/tractor
Allow specifying start_method and hard_kill
parent
c1089dbd95
commit
b7a4641674
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue