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

Wait on proc before killing stdio

drop_cloudpickle
Tyler Goodlet 2020-07-22 12:50:16 -04:00
parent efde3a5773
commit d3acb8d061
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tractor/_spawn.py
View File

@ -25,7 +25,7 @@ except ImportError:
from multiprocessing import forkserver # type: ignore from multiprocessing import forkserver # type: ignore
from typing import Tuple from typing import Tuple
from . import _forkserver_override, _child from . import _forkserver_override
from ._state import current_actor from ._state import current_actor
from .log import get_logger from .log import get_logger
from ._portal import Portal from ._portal import Portal
@ -122,6 +122,7 @@ async def exhaust_portal(
# we reraise in the parent task via a ``trio.MultiError`` # we reraise in the parent task via a ``trio.MultiError``
return err return err
else: else:
log.debug(f"Returning final result: {final}")
return final return final
@ -227,7 +228,10 @@ async def new_proc(
cancel_scope = await nursery.start( cancel_scope = await nursery.start(
cancel_on_completion, portal, subactor, errors) cancel_on_completion, portal, subactor, errors)
# run_in_process blocks here until process is complete # Wait for proc termination but **dont'** yet call
# ``trio.Process.__aexit__()`` (it tears down stdio
# which will kill any waiting remote pdb trace).
await proc.wait()
else: else:
# `multiprocessing` # `multiprocessing`
assert _ctx assert _ctx