goodboy
/
tractor
1
0
Fork 1
Code Issues 1 Pull Requests 4 Packages Projects Releases Wiki Activity

Don't alert nursery on expected cancel result

zombie_lord_infinite
Tyler Goodlet 2021-10-10 16:47:17 -04:00
parent c01d2f8aea
commit 4d30e25591
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tractor/_spawn.py
View File

@ -31,8 +31,12 @@ from .log import get_logger
from ._portal import Portal
from ._actor import Actor
from ._entry import _mp_main
from ._exceptions import ActorFailure, RemoteActorError
from ._debug import maybe_wait_for_debugger
from ._exceptions import (
ActorFailure,
RemoteActorError,
ContextCancelled,
)
from ._debug import maybe_wait_for_debugger, breakpoint
log = get_logger('tractor')
@ -120,12 +124,12 @@ async def result_from_portal(
# a MultiError and we still send out a cancel request
# result = await exhaust_portal(portal, actor)
try:
log.debug(f"Waiting on final result from {actor.uid}")
log.info(f"Waiting on final result from {actor.uid}")
# XXX: streams should never be reaped here since they should
# always be established and shutdown using a context manager api
result = await portal.result()
log.debug(f"Returning final result: {result}")
log.info(f"Returning final result: {result}")
except (Exception, trio.MultiError) as err:
# we reraise in the parent task via a ``trio.MultiError``
@ -204,6 +208,7 @@ async def do_hard_kill(
async def reap_proc(
proc: trio.Process,
uid: tuple[str, str],
terminate_after: float = float('inf'),
hard_kill_after: int = 0.1,
@ -214,6 +219,7 @@ async def reap_proc(
# killing. This is a "light" (cancellable) join,
# the hard join is below after timeout
await proc.wait()
log.info(f'{uid} terminated gracefully')
if cs.cancelled_caught and terminate_after is not float('inf'):
# Always "hard" join lingering sub procs since no
@ -398,6 +404,8 @@ async def new_proc(
if portal.channel.connected() and ria:
# we wait for result and cancel on completion
# if uid[0] == 'odds':
# await breakpoint()
await result_from_portal(
portal,
subactor,
@ -414,11 +422,19 @@ async def new_proc(
# )
# soft & cancellable
await reap_proc(proc)
await reap_proc(proc, uid)
# # if proc terminates before portal result
# if cancel_scope:
# cancel_scope.cancel()
except (
ContextCancelled,
) as err:
if portal.cancel_called:
log.cancel('{uid} received expected cancel')
# soft & cancellable
await reap_proc(proc, uid, terminate_after=0.1)
except (
RemoteActorError,
@ -471,6 +487,7 @@ async def new_proc(
await reap_proc(
proc,
uid,
# this is the same as previous timeout
# setting before rewriting this spawn
# section
@ -488,6 +505,7 @@ async def new_proc(
with trio.CancelScope(shield=True):
await reap_proc(
proc,
uid,
terminate_after=0.1,
)