1880 lines
67 KiB
Python
1880 lines
67 KiB
Python
# tractor: structured concurrent "actors".
|
|
# Copyright 2018-eternity Tyler Goodlet.
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
'''
|
|
The fundamental cross process SC abstraction: an inter-actor,
|
|
cancel-scope linked task "context".
|
|
|
|
A ``Context`` is very similar to the ``trio.Nursery.cancel_scope`` built
|
|
into each ``trio.Nursery`` except it links the lifetimes of memory space
|
|
disjoint, parallel executing tasks in separate actors.
|
|
|
|
'''
|
|
from __future__ import annotations
|
|
from collections import deque
|
|
from contextlib import asynccontextmanager as acm
|
|
from dataclasses import (
|
|
dataclass,
|
|
field,
|
|
)
|
|
from functools import partial
|
|
import inspect
|
|
from pprint import pformat
|
|
from typing import (
|
|
Any,
|
|
Callable,
|
|
AsyncGenerator,
|
|
TYPE_CHECKING,
|
|
)
|
|
import warnings
|
|
|
|
import trio
|
|
|
|
from .msg import NamespacePath
|
|
from ._exceptions import (
|
|
ContextCancelled,
|
|
InternalError,
|
|
RemoteActorError,
|
|
StreamOverrun,
|
|
pack_error,
|
|
unpack_error,
|
|
)
|
|
from .log import get_logger
|
|
from ._ipc import Channel
|
|
from ._streaming import MsgStream
|
|
from ._state import current_actor
|
|
|
|
if TYPE_CHECKING:
|
|
from ._portal import Portal
|
|
from ._runtime import Actor
|
|
from ._ipc import MsgTransport
|
|
|
|
|
|
log = get_logger(__name__)
|
|
|
|
|
|
async def _drain_to_final_msg(
|
|
ctx: Context,
|
|
|
|
hide_tb: bool = True,
|
|
msg_limit: int = 6,
|
|
|
|
) -> list[dict]:
|
|
'''
|
|
Drain IPC msgs delivered to the underlying rx-mem-chan
|
|
`Context._recv_chan` from the runtime in search for a final
|
|
result or error msg.
|
|
|
|
The motivation here is to ideally capture errors during ctxc
|
|
conditions where a canc-request/or local error is sent but the
|
|
local task also excepts and enters the
|
|
`Portal.open_context().__aexit__()` block wherein we prefer to
|
|
capture and raise any remote error or ctxc-ack as part of the
|
|
`ctx.result()` cleanup and teardown sequence.
|
|
|
|
'''
|
|
__tracebackhide__: bool = hide_tb
|
|
raise_overrun: bool = not ctx._allow_overruns
|
|
|
|
# wait for a final context result by collecting (but
|
|
# basically ignoring) any bi-dir-stream msgs still in transit
|
|
# from the far end.
|
|
pre_result_drained: list[dict] = []
|
|
while not (
|
|
ctx.maybe_error
|
|
and not ctx._final_result_is_set()
|
|
):
|
|
try:
|
|
# TODO: can remove?
|
|
# await trio.lowlevel.checkpoint()
|
|
|
|
# NOTE: this REPL usage actually works here dawg! Bo
|
|
# from .devx._debug import pause
|
|
# await pause()
|
|
|
|
# TODO: bad idea?
|
|
# -[ ] wrap final outcome channel wait in a scope so
|
|
# it can be cancelled out of band if needed?
|
|
#
|
|
# with trio.CancelScope() as res_cs:
|
|
# ctx._res_scope = res_cs
|
|
# msg: dict = await ctx._recv_chan.receive()
|
|
# if res_cs.cancelled_caught:
|
|
|
|
# TODO: ensure there's no more hangs, debugging the
|
|
# runtime pretty preaase!
|
|
# from .devx._debug import pause
|
|
# await pause()
|
|
|
|
# TODO: can remove this finally?
|
|
# we have no more need for the sync draining right
|
|
# since we're can kinda guarantee the async
|
|
# `.receive()` below will never block yah?
|
|
#
|
|
# if (
|
|
# ctx._cancel_called and (
|
|
# ctx.cancel_acked
|
|
# # or ctx.chan._cancel_called
|
|
# )
|
|
# # or not ctx._final_result_is_set()
|
|
# # ctx.outcome is not
|
|
# # or ctx.chan._closed
|
|
# ):
|
|
# try:
|
|
# msg: dict = await ctx._recv_chan.receive_nowait()()
|
|
# except trio.WouldBlock:
|
|
# log.warning(
|
|
# 'When draining already `.cancel_called` ctx!\n'
|
|
# 'No final msg arrived..\n'
|
|
# )
|
|
# break
|
|
# else:
|
|
# msg: dict = await ctx._recv_chan.receive()
|
|
|
|
# TODO: don't need it right jefe?
|
|
# with trio.move_on_after(1) as cs:
|
|
# if cs.cancelled_caught:
|
|
# from .devx._debug import pause
|
|
# await pause()
|
|
|
|
# pray to the `trio` gawds that we're corrent with this
|
|
msg: dict = await ctx._recv_chan.receive()
|
|
|
|
# NOTE: we get here if the far end was
|
|
# `ContextCancelled` in 2 cases:
|
|
# 1. we requested the cancellation and thus
|
|
# SHOULD NOT raise that far end error,
|
|
# 2. WE DID NOT REQUEST that cancel and thus
|
|
# SHOULD RAISE HERE!
|
|
except trio.Cancelled:
|
|
|
|
# CASE 2: mask the local cancelled-error(s)
|
|
# only when we are sure the remote error is
|
|
# the source cause of this local task's
|
|
# cancellation.
|
|
if re := ctx._remote_error:
|
|
ctx._maybe_raise_remote_err(re)
|
|
|
|
# CASE 1: we DID request the cancel we simply
|
|
# continue to bubble up as normal.
|
|
raise
|
|
|
|
try:
|
|
ctx._result: Any = msg['return']
|
|
log.runtime(
|
|
'Context delivered final draining msg:\n'
|
|
f'{pformat(msg)}'
|
|
)
|
|
# XXX: only close the rx mem chan AFTER
|
|
# a final result is retreived.
|
|
# if ctx._recv_chan:
|
|
# await ctx._recv_chan.aclose()
|
|
# TODO: ^ we don't need it right?
|
|
break
|
|
|
|
except KeyError:
|
|
# always capture unexpected/non-result msgs
|
|
pre_result_drained.append(msg)
|
|
|
|
if 'yield' in msg:
|
|
# far end task is still streaming to us so discard
|
|
# and report per local context state.
|
|
if (
|
|
(ctx._stream.closed
|
|
and (reason := 'stream was already closed')
|
|
)
|
|
or (ctx._cancelled_caught
|
|
and (reason := 'ctx caught a cancel')
|
|
)
|
|
or (ctx._cancel_called
|
|
and (reason := 'ctx called `.cancel()`')
|
|
)
|
|
or (len(pre_result_drained) > msg_limit
|
|
and (reason := f'"yield" limit={msg_limit}')
|
|
)
|
|
):
|
|
log.cancel(
|
|
'Cancelling `MsgStream` drain since '
|
|
f'{reason}\n\n'
|
|
f'<= {ctx.chan.uid}\n'
|
|
f' |_{ctx._nsf}()\n\n'
|
|
f'=> {ctx._task}\n'
|
|
f' |_{ctx._stream}\n\n'
|
|
|
|
f'{pformat(msg)}\n'
|
|
)
|
|
return pre_result_drained
|
|
|
|
# drain up to the `msg_limit` hoping to get
|
|
# a final result or error/ctxc.
|
|
else:
|
|
log.warning(
|
|
'Ignoring "yield" msg during `ctx.result()` drain..\n'
|
|
f'<= {ctx.chan.uid}\n'
|
|
f' |_{ctx._nsf}()\n\n'
|
|
f'=> {ctx._task}\n'
|
|
f' |_{ctx._stream}\n\n'
|
|
|
|
f'{pformat(msg)}\n'
|
|
)
|
|
continue
|
|
|
|
# TODO: work out edge cases here where
|
|
# a stream is open but the task also calls
|
|
# this?
|
|
# -[ ] should be a runtime error if a stream is open
|
|
# right?
|
|
elif 'stop' in msg:
|
|
log.cancel(
|
|
'Remote stream terminated due to "stop" msg:\n\n'
|
|
f'{pformat(msg)}\n'
|
|
)
|
|
continue
|
|
|
|
# It's an internal error if any other msg type without
|
|
# a`'cid'` field arrives here!
|
|
if not msg.get('cid'):
|
|
raise InternalError(
|
|
'Unexpected cid-missing msg?\n\n'
|
|
f'{msg}\n'
|
|
)
|
|
|
|
# XXX fallthrough to handle expected error XXX
|
|
re: Exception|None = ctx._remote_error
|
|
if re:
|
|
log.critical(
|
|
'Remote ctx terminated due to "error" msg:\n'
|
|
f'{re}'
|
|
)
|
|
assert msg is ctx._cancel_msg
|
|
# NOTE: this solved a super dupe edge case XD
|
|
# this was THE super duper edge case of:
|
|
# - local task opens a remote task,
|
|
# - requests remote cancellation of far end
|
|
# ctx/tasks,
|
|
# - needs to wait for the cancel ack msg
|
|
# (ctxc) or some result in the race case
|
|
# where the other side's task returns
|
|
# before the cancel request msg is ever
|
|
# rxed and processed,
|
|
# - here this surrounding drain loop (which
|
|
# iterates all ipc msgs until the ack or
|
|
# an early result arrives) was NOT exiting
|
|
# since we are the edge case: local task
|
|
# does not re-raise any ctxc it receives
|
|
# IFF **it** was the cancellation
|
|
# requester..
|
|
# will raise if necessary, ow break from
|
|
# loop presuming any error terminates the
|
|
# context!
|
|
ctx._maybe_raise_remote_err(
|
|
re,
|
|
# NOTE: obvi we don't care if we
|
|
# overran the far end if we're already
|
|
# waiting on a final result (msg).
|
|
# raise_overrun_from_self=False,
|
|
raise_overrun_from_self=raise_overrun,
|
|
)
|
|
|
|
break # OOOOOF, yeah obvi we need this..
|
|
|
|
# XXX we should never really get here
|
|
# right! since `._deliver_msg()` should
|
|
# always have detected an {'error': ..}
|
|
# msg and already called this right!?!
|
|
elif error := unpack_error(
|
|
msg=msg,
|
|
chan=ctx._portal.channel,
|
|
hide_tb=False,
|
|
):
|
|
log.critical('SHOULD NEVER GET HERE!?')
|
|
assert msg is ctx._cancel_msg
|
|
assert error.msgdata == ctx._remote_error.msgdata
|
|
from .devx._debug import pause
|
|
await pause()
|
|
ctx._maybe_cancel_and_set_remote_error(error)
|
|
ctx._maybe_raise_remote_err(error)
|
|
|
|
else:
|
|
# bubble the original src key error
|
|
raise
|
|
else:
|
|
log.cancel(
|
|
'Skipping `MsgStream` drain since final outcome is set\n\n'
|
|
f'{ctx.outcome}\n'
|
|
)
|
|
|
|
return pre_result_drained
|
|
|
|
|
|
class Unresolved:
|
|
'''
|
|
Placeholder value for `Context._result` until
|
|
a final return value or raised error is resolved.
|
|
|
|
'''
|
|
...
|
|
|
|
|
|
# TODO: make this a .msg.types.Struct!
|
|
# -[ ] ideally we can freeze it
|
|
# -[ ] let's us do field diffing nicely in tests Bo
|
|
@dataclass
|
|
class Context:
|
|
'''
|
|
An inter-actor, SC transitive, `trio.Task` communication context.
|
|
|
|
NB: This class should **never be instatiated directly**, it is allocated
|
|
by the runtime in 2 ways:
|
|
- by entering ``Portal.open_context()`` which is the primary
|
|
public API for any "caller" task or,
|
|
- by the RPC machinery's `._runtime._invoke()` as a `ctx` arg
|
|
to a remotely scheduled "callee" function.
|
|
|
|
AND is always constructed using the below ``mk_context()``.
|
|
|
|
Allows maintaining task or protocol specific state between
|
|
2 cancel-scope-linked, communicating and parallel executing
|
|
`trio.Task`s. Contexts are allocated on each side of any task
|
|
RPC-linked msg dialog, i.e. for every request to a remote
|
|
actor from a `Portal`. On the "callee" side a context is
|
|
always allocated inside ``._runtime._invoke()``.
|
|
|
|
# TODO: more detailed writeup on cancellation, error and
|
|
# streaming semantics..
|
|
|
|
A context can be cancelled and (possibly eventually restarted) from
|
|
either side of the underlying IPC channel, it can also open task
|
|
oriented message streams, and acts more or less as an IPC aware
|
|
inter-actor-task ``trio.CancelScope``.
|
|
|
|
'''
|
|
chan: Channel
|
|
cid: str # "context id", more or less a unique linked-task-pair id
|
|
|
|
_actor: Actor
|
|
|
|
# the "feeder" channels for delivering message values to the
|
|
# local task from the runtime's msg processing loop.
|
|
_recv_chan: trio.MemoryReceiveChannel
|
|
_send_chan: trio.MemorySendChannel
|
|
|
|
# full "namespace-path" to target RPC function
|
|
_nsf: NamespacePath
|
|
|
|
# the "invocation type" of the far end task-entry-point
|
|
# function, normally matching a logic block inside
|
|
# `._runtime.invoke()`.
|
|
_remote_func_type: str | None = None
|
|
|
|
# NOTE: (for now) only set (a portal) on the caller side since
|
|
# the callee doesn't generally need a ref to one and should
|
|
# normally need to explicitly ask for handle to its peer if
|
|
# more the the `Context` is needed?
|
|
_portal: Portal | None = None
|
|
|
|
# NOTE: each side of the context has its own cancel scope
|
|
# which is exactly the primitive that allows for
|
|
# cross-actor-task-supervision and thus SC.
|
|
_scope: trio.CancelScope|None = None
|
|
_task: trio.lowlevel.Task|None = None
|
|
|
|
# TODO: cs around result waiting so we can cancel any
|
|
# permanently blocking `._recv_chan.receive()` call in
|
|
# a drain loop?
|
|
# _res_scope: trio.CancelScope|None = None
|
|
|
|
# on a clean exit there should be a final value
|
|
# delivered from the far end "callee" task, so
|
|
# this value is only set on one side.
|
|
# _result: Any | int = None
|
|
_result: Any|Unresolved = Unresolved
|
|
|
|
# if the local "caller" task errors this value is always set
|
|
# to the error that was captured in the
|
|
# `Portal.open_context().__aexit__()` teardown block OR, in
|
|
# 2 special cases when an (maybe) expected remote error
|
|
# arrives that we purposely swallow silently:
|
|
# - `ContextCancelled` with `.canceller` set to our uid:
|
|
# a self-cancel,
|
|
# - `RemoteActorError[StreamOverrun]` which was caught during
|
|
# a self-cancellation teardown msg drain.
|
|
_local_error: BaseException|None = None
|
|
|
|
# if the either side gets an error from the other
|
|
# this value is set to that error unpacked from an
|
|
# IPC msg.
|
|
_remote_error: BaseException|None = None
|
|
|
|
# only set if an actor-local task called `.cancel()`
|
|
_cancel_called: bool = False # did WE request cancel of the far end?
|
|
|
|
# TODO: do we even need this? we can assume that if we're
|
|
# cancelled that the other side is as well, so maybe we should
|
|
# instead just have a `.canceller` pulled from the
|
|
# `ContextCancelled`?
|
|
_canceller: tuple[str, str] | None = None
|
|
|
|
# NOTE: we try to ensure assignment of a "cancel msg" since
|
|
# there's always going to be an "underlying reason" that any
|
|
# context was closed due to either a remote side error or
|
|
# a call to `.cancel()` which triggers `ContextCancelled`.
|
|
_cancel_msg: str|dict|None = None
|
|
|
|
# NOTE: this state var used by the runtime to determine if the
|
|
# `pdbp` REPL is allowed to engage on contexts terminated via
|
|
# a `ContextCancelled` due to a call to `.cancel()` triggering
|
|
# "graceful closure" on either side:
|
|
# - `._runtime._invoke()` will check this flag before engaging
|
|
# the crash handler REPL in such cases where the "callee"
|
|
# raises the cancellation,
|
|
# - `.devx._debug.lock_tty_for_child()` will set it to `False` if
|
|
# the global tty-lock has been configured to filter out some
|
|
# actors from being able to acquire the debugger lock.
|
|
_enter_debugger_on_cancel: bool = True
|
|
|
|
# init and streaming state
|
|
_started_called: bool = False
|
|
_stream_opened: bool = False
|
|
_stream: MsgStream|None = None
|
|
|
|
# overrun handling machinery
|
|
# NOTE: none of this provides "backpressure" to the remote
|
|
# task, only an ability to not lose messages when the local
|
|
# task is configured to NOT transmit ``StreamOverrun``s back
|
|
# to the other side.
|
|
_overflow_q: deque[dict] = field(
|
|
default_factory=partial(
|
|
deque,
|
|
maxlen=616,
|
|
)
|
|
)
|
|
|
|
# NOTE: this was originally a legacy interface from when we
|
|
# were raising remote errors (set to `._remote_error`) by
|
|
# starting a task inside this nursery that simply raised the
|
|
# boxed exception. NOW, it's used for spawning overrun queuing
|
|
# tasks when `.allow_overruns == True` !!!
|
|
_scope_nursery: trio.Nursery|None = None
|
|
|
|
# streaming overrun state tracking
|
|
_in_overrun: bool = False
|
|
_allow_overruns: bool = False
|
|
|
|
|
|
def __str__(self) -> str:
|
|
ds: str = '='
|
|
# ds: str = ': '
|
|
|
|
# only show if opened
|
|
maybe_stream_repr: str = ''
|
|
if stream := self._stream:
|
|
# TODO: a `MsgStream.reprol()` !!
|
|
# f' stream{ds}{self._stream}\n'
|
|
# f' {self._stream}\n'
|
|
maybe_stream_repr: str = (
|
|
f' {stream}\n'
|
|
)
|
|
|
|
outcome_str: str = self.repr_outcome(
|
|
show_error_fields=True
|
|
)
|
|
outcome_typ_str: str = self.repr_outcome(
|
|
type_only=True
|
|
)
|
|
|
|
return (
|
|
f'<Context(\n'
|
|
# f'\n'
|
|
# f' ---\n'
|
|
f' |_ipc: {self.dst_maddr}\n'
|
|
# f' dst_maddr{ds}{self.dst_maddr}\n'
|
|
f" uid{ds}'{self.chan.uid}'\n"
|
|
f" cid{ds}'{self.cid}'\n"
|
|
# f' ---\n'
|
|
f'\n'
|
|
# f' ---\n'
|
|
f' |_rpc: {self.repr_rpc}\n'
|
|
f" side{ds}'{self.side}'\n"
|
|
# f' rpc_sig{ds}{self.repr_rpc}\n'
|
|
f' {self._task}\n'
|
|
f'{maybe_stream_repr}'
|
|
# f' ---\n'
|
|
f'\n'
|
|
# f' -----\n'
|
|
#
|
|
# TODO: better state `str`ids?
|
|
# -[ ] maybe map err-types to strs like 'cancelled',
|
|
# 'errored', 'streaming', 'started', .. etc.
|
|
# -[ ] as well as a final result wrapper like
|
|
# `outcome.Value`?
|
|
#
|
|
f' |_state: {outcome_typ_str}\n'
|
|
|
|
f' outcome{ds}{outcome_str}\n'
|
|
f' result{ds}{self._result}\n'
|
|
f' cancel_called{ds}{self.cancel_called}\n'
|
|
f' cancel_acked{ds}{self.cancel_acked}\n'
|
|
f' canceller{ds}{self._canceller}\n'
|
|
|
|
# TODO: any other fields?
|
|
# f' maybe_error{ds}{self.maybe_error}\n'
|
|
# -[ ] ^ already covered by outcome yah?
|
|
# f' cancelled_caught={self.cancelled_caught}\n'
|
|
# -[ ] remove this ^ right?
|
|
|
|
# f' _remote_error={self._remote_error}
|
|
')>\n'
|
|
)
|
|
# NOTE: making this return a value that can be passed to
|
|
# `eval()` is entirely **optional** dawggg B)
|
|
# https://docs.python.org/3/library/functions.html#repr
|
|
# https://docs.python.org/3/reference/datamodel.html#object.__repr__
|
|
#
|
|
# XXX: Currently we target **readability** from a (console)
|
|
# logging perspective over `eval()`-ability since we do NOT
|
|
# target serializing non-struct instances!
|
|
# def __repr__(self) -> str:
|
|
__repr__ = __str__
|
|
|
|
@property
|
|
def cancel_called(self) -> bool:
|
|
'''
|
|
Records whether cancellation has been requested for this context
|
|
by a call to `.cancel()` either due to,
|
|
- either an explicit call by some local task,
|
|
- or an implicit call due to an error caught inside
|
|
the ``Portal.open_context()`` block.
|
|
|
|
'''
|
|
return self._cancel_called
|
|
|
|
@property
|
|
def canceller(self) -> tuple[str, str]|None:
|
|
'''
|
|
``Actor.uid: tuple[str, str]`` of the (remote)
|
|
actor-process who's task was cancelled thus causing this
|
|
(side of the) context to also be cancelled.
|
|
|
|
'''
|
|
if canc := self._canceller:
|
|
return tuple(canc)
|
|
|
|
return None
|
|
|
|
def _is_self_cancelled(
|
|
self,
|
|
remote_error: Exception|None = None,
|
|
|
|
) -> bool:
|
|
|
|
if not self._cancel_called:
|
|
return False
|
|
|
|
re: BaseException|None = (
|
|
remote_error
|
|
or self._remote_error
|
|
)
|
|
if not re:
|
|
return False
|
|
|
|
if from_uid := re.src_actor_uid:
|
|
from_uid: tuple = tuple(from_uid)
|
|
|
|
our_uid: tuple = self._actor.uid
|
|
our_canceller = self.canceller
|
|
|
|
return bool(
|
|
isinstance(re, ContextCancelled)
|
|
and from_uid == self.chan.uid
|
|
and re.canceller == our_uid
|
|
and our_canceller == from_uid
|
|
)
|
|
|
|
@property
|
|
def cancel_acked(self) -> bool:
|
|
'''
|
|
Records whether the task on the remote side of this IPC
|
|
context acknowledged a cancel request via a relayed
|
|
`ContextCancelled` with the `.canceller` attr set to the
|
|
`Actor.uid` of the local actor who's task entered
|
|
`Portal.open_context()`.
|
|
|
|
This will only be `True` when `.cancel()` is called and
|
|
the ctxc response contains a `.canceller: tuple` field
|
|
equal to the uid of the calling task's actor.
|
|
|
|
'''
|
|
return self._is_self_cancelled()
|
|
|
|
@property
|
|
def cancelled_caught(self) -> bool:
|
|
'''
|
|
Exactly the value of `self._scope.cancelled_caught`
|
|
(delegation) and should only be (able to be read as)
|
|
`True` for a `.side == "caller"` ctx wherein the
|
|
`Portal.open_context()` block was exited due to a call to
|
|
`._scope.cancel()` - which should only ocurr in 2 cases:
|
|
|
|
- a caller side calls `.cancel()`, the far side cancels
|
|
and delivers back a `ContextCancelled` (making
|
|
`.cancel_acked == True`) and `._scope.cancel()` is
|
|
called by `._maybe_cancel_and_set_remote_error()` which
|
|
in turn cancels all `.open_context()` started tasks
|
|
(including any overrun queuing ones).
|
|
=> `._scope.cancelled_caught == True` by normal `trio`
|
|
cs semantics.
|
|
|
|
- a caller side is delivered a `._remote_error:
|
|
RemoteActorError` via `._deliver_msg()` and a transitive
|
|
call to `_maybe_cancel_and_set_remote_error()` calls
|
|
`._scope.cancel()` and that cancellation eventually
|
|
results in `trio.Cancelled`(s) caught in the
|
|
`.open_context()` handling around the @acm's `yield`.
|
|
|
|
Only as an FYI, in the "callee" side case it can also be
|
|
set but never is readable by any task outside the RPC
|
|
machinery in `._invoke()` since,:
|
|
- when a callee side calls `.cancel()`, `._scope.cancel()`
|
|
is called immediately and handled specially inside
|
|
`._invoke()` to raise a `ContextCancelled` which is then
|
|
sent to the caller side.
|
|
|
|
However, `._scope.cancelled_caught` can NEVER be
|
|
accessed/read as `True` by any RPC invoked task since it
|
|
will have terminated before the cs block exit.
|
|
|
|
'''
|
|
return bool(
|
|
# the local scope was cancelled either by
|
|
# remote error or self-request
|
|
(self._scope and self._scope.cancelled_caught)
|
|
|
|
# the local scope was never cancelled
|
|
# and instead likely we received a remote side
|
|
# # cancellation that was raised inside `.result()`
|
|
# or (
|
|
# (se := self._local_error)
|
|
# and se is re
|
|
# )
|
|
)
|
|
|
|
# @property
|
|
# def is_waiting_result(self) -> bool:
|
|
# return bool(self._res_scope)
|
|
|
|
@property
|
|
def side(self) -> str:
|
|
'''
|
|
Return string indicating which task this instance is wrapping.
|
|
|
|
'''
|
|
return 'caller' if self._portal else 'callee'
|
|
|
|
async def send_yield(
|
|
self,
|
|
data: Any,
|
|
|
|
) -> None:
|
|
|
|
warnings.warn(
|
|
"`Context.send_yield()` is now deprecated. "
|
|
"Use ``MessageStream.send()``. ",
|
|
DeprecationWarning,
|
|
stacklevel=2,
|
|
)
|
|
await self.chan.send({'yield': data, 'cid': self.cid})
|
|
|
|
async def send_stop(self) -> None:
|
|
# await pause()
|
|
await self.chan.send({
|
|
'stop': True,
|
|
'cid': self.cid
|
|
})
|
|
|
|
def _maybe_cancel_and_set_remote_error(
|
|
self,
|
|
error: BaseException,
|
|
|
|
) -> None:
|
|
'''
|
|
(Maybe) cancel this local scope due to a received remote
|
|
error (normally via an IPC msg) which the actor runtime
|
|
routes to this context.
|
|
|
|
Acts as a form of "relay" for a remote error raised in the
|
|
corresponding remote task's `Context` wherein the next time
|
|
the local task exectutes a checkpoint, a `trio.Cancelled`
|
|
will be raised and depending on the type and source of the
|
|
original remote error, and whether or not the local task
|
|
called `.cancel()` itself prior, an equivalent
|
|
`ContextCancelled` or `RemoteActorError` wrapping the
|
|
remote error may be raised here by any of,
|
|
|
|
- `Portal.open_context()`
|
|
- `Portal.result()`
|
|
- `Context.open_stream()`
|
|
- `Context.result()`
|
|
|
|
when called/closed by actor local task(s).
|
|
|
|
NOTEs:
|
|
- It is expected that the caller has previously unwrapped
|
|
the remote error using a call to `unpack_error()` and
|
|
provides that output exception value as the input
|
|
`error` argument *here*.
|
|
|
|
TODOs:
|
|
- If this is an error message from a context opened by
|
|
`Portal.open_context()` (ideally) we want to interrupt
|
|
any ongoing local tasks operating within that
|
|
`Context`'s cancel-scope so as to be notified ASAP of
|
|
the remote error and engage any caller handling (eg.
|
|
for cross-process task supervision).
|
|
|
|
- In some cases we may want to raise the remote error
|
|
immediately since there is no guarantee the locally
|
|
operating task(s) will attempt to execute a checkpoint
|
|
any time soon; in such cases there are 2 possible
|
|
approaches depending on the current task's work and
|
|
wrapping "thread" type:
|
|
|
|
- Currently we only support
|
|
a `trio`-native-and-graceful approach: we only ever
|
|
wait for local tasks to exec a next
|
|
`trio.lowlevel.checkpoint()` assuming that any such
|
|
task must do so to interact with the actor runtime
|
|
and IPC interfaces and will then be cancelled by
|
|
the internal `._scope` block.
|
|
|
|
- (NOT IMPLEMENTED) system-level-aggressive: maybe we
|
|
could eventually interrupt sync code (invoked using
|
|
`trio.to_thread` or some other adapter layer) with
|
|
a signal (a custom unix one for example?
|
|
https://stackoverflow.com/a/5744185) depending on the
|
|
task's wrapping thread-type such that long running
|
|
sync code should never cause the delay of actor
|
|
supervision tasks such as cancellation and respawn
|
|
logic.
|
|
|
|
'''
|
|
# XXX: currently this should only be used when
|
|
# `Portal.open_context()` has been opened since it's
|
|
# assumed that other portal APIs like,
|
|
# - `Portal.run()`,
|
|
# - `ActorNursery.run_in_actor()`
|
|
# do their own error checking at their own call points and
|
|
# result processing.
|
|
|
|
# TODO: never do this right?
|
|
# if self._remote_error:
|
|
# return
|
|
|
|
# XXX: denote and set the remote side's error so that
|
|
# after we cancel whatever task is the opener of this
|
|
# context, it can raise or swallow that error
|
|
# appropriately.
|
|
log.runtime(
|
|
'Setting remote error for ctx\n\n'
|
|
f'<= remote ctx uid: {self.chan.uid}\n'
|
|
f'=>{error}'
|
|
)
|
|
self._remote_error: BaseException = error
|
|
|
|
# self-cancel (ack) or,
|
|
# peer propagated remote cancellation.
|
|
if isinstance(error, ContextCancelled):
|
|
|
|
whom: str = (
|
|
'us' if error.canceller == self._actor.uid
|
|
else 'peer'
|
|
)
|
|
log.cancel(
|
|
f'IPC context cancelled by {whom}!\n\n'
|
|
f'{error}'
|
|
)
|
|
|
|
else:
|
|
log.error(
|
|
f'Remote context error:\n\n'
|
|
|
|
f'{error}\n'
|
|
f'{pformat(self)}\n'
|
|
)
|
|
|
|
# always record the cancelling actor's uid since its
|
|
# cancellation state is linked and we want to know
|
|
# which process was the cause / requester of the
|
|
# cancellation.
|
|
maybe_error_src: tuple = getattr(
|
|
error,
|
|
'src_actor_uid',
|
|
None,
|
|
)
|
|
self._canceller = (
|
|
maybe_error_src
|
|
or
|
|
# XXX: in the case we get a non-boxed error?
|
|
# -> wait but this should never happen right?
|
|
self.chan.uid
|
|
)
|
|
|
|
# Cancel the local `._scope`, catch that
|
|
# `._scope.cancelled_caught` and re-raise any remote error
|
|
# once exiting (or manually calling `.result()`) the
|
|
# `.open_context()` block.
|
|
cs: trio.CancelScope = self._scope
|
|
if (
|
|
cs
|
|
|
|
# XXX this is an expected cancel request response
|
|
# message and we **don't need to raise it** in the
|
|
# local cancel `._scope` since it will potentially
|
|
# override a real error. After this method returns
|
|
# if `._cancel_called` then `.cancel_acked and .cancel_called`
|
|
# always should be set.
|
|
and not self._is_self_cancelled()
|
|
|
|
and not cs.cancel_called
|
|
and not cs.cancelled_caught
|
|
):
|
|
# TODO: it'd sure be handy to inject our own
|
|
# `trio.Cancelled` subtype here ;)
|
|
# https://github.com/goodboy/tractor/issues/368
|
|
self._scope.cancel()
|
|
|
|
# TODO: maybe we should also call `._res_scope.cancel()` if it
|
|
# exists to support cancelling any drain loop hangs?
|
|
|
|
# TODO: add to `Channel`?
|
|
@property
|
|
def dst_maddr(self) -> str:
|
|
chan: Channel = self.chan
|
|
dst_addr, dst_port = chan.raddr
|
|
trans: MsgTransport = chan.transport
|
|
# cid: str = self.cid
|
|
# cid_head, cid_tail = cid[:6], cid[-6:]
|
|
return (
|
|
f'/ipv4/{dst_addr}'
|
|
f'/{trans.name_key}/{dst_port}'
|
|
# f'/{self.chan.uid[0]}'
|
|
# f'/{self.cid}'
|
|
|
|
# f'/cid={cid_head}..{cid_tail}'
|
|
# TODO: ? not use this ^ right ?
|
|
)
|
|
|
|
dmaddr = dst_maddr
|
|
|
|
@property
|
|
def repr_rpc(
|
|
self,
|
|
) -> str:
|
|
# TODO: how to show the transport interchange fmt?
|
|
# codec: str = self.chan.transport.codec_key
|
|
outcome_str: str = self.repr_outcome(
|
|
show_error_fields=True,
|
|
type_only=True,
|
|
)
|
|
return (
|
|
# f'{self._nsf}() -{{{codec}}}-> {repr(self.outcome)}:'
|
|
f'{self._nsf}() -> {outcome_str}:'
|
|
)
|
|
|
|
async def cancel(
|
|
self,
|
|
timeout: float = 0.616,
|
|
|
|
) -> None:
|
|
'''
|
|
Cancel this inter-actor IPC context by requestng the
|
|
remote side's cancel-scope-linked `trio.Task` by calling
|
|
`._scope.cancel()` and delivering an `ContextCancelled`
|
|
ack msg in reponse.
|
|
|
|
Behaviour:
|
|
---------
|
|
- after the far end cancels, the `.cancel()` calling side
|
|
should receive a `ContextCancelled` with the
|
|
`.canceller: tuple` uid set to the current `Actor.uid`.
|
|
|
|
- timeout (quickly) on failure to rx this ACK error-msg in
|
|
an attempt to sidestep 2-generals when the transport
|
|
layer fails.
|
|
|
|
Note, that calling this method DOES NOT also necessarily
|
|
result in `Context._scope.cancel()` being called
|
|
**locally**!
|
|
|
|
=> That is, an IPC `Context` (this) **does not**
|
|
have the same semantics as a `trio.CancelScope`.
|
|
|
|
If the caller (who entered the `Portal.open_context()`)
|
|
desires that the internal block's cancel-scope be
|
|
cancelled it should open its own `trio.CancelScope` and
|
|
manage it as needed.
|
|
|
|
'''
|
|
side: str = self.side
|
|
self._cancel_called: bool = True
|
|
|
|
header: str = (
|
|
f'Cancelling ctx with peer from {side.upper()} side\n\n'
|
|
)
|
|
reminfo: str = (
|
|
# ' =>\n'
|
|
f'Context.cancel() => {self.chan.uid}\n'
|
|
# f'{self.chan.uid}\n'
|
|
f' |_ @{self.dst_maddr}\n'
|
|
f' >> {self.repr_rpc}\n'
|
|
# f' >> {self._nsf}() -> {codec}[dict]:\n\n'
|
|
# TODO: pull msg-type from spec re #320
|
|
)
|
|
|
|
# CALLER side task
|
|
# ------ - ------
|
|
# Aka the one that entered `Portal.open_context()`
|
|
#
|
|
# NOTE: on the call side we never manually call
|
|
# `._scope.cancel()` since we expect the eventual
|
|
# `ContextCancelled` from the other side to trigger this
|
|
# when the runtime finally receives it during teardown
|
|
# (normally in `.result()` called from
|
|
# `Portal.open_context().__aexit__()`)
|
|
if side == 'caller':
|
|
if not self._portal:
|
|
raise InternalError(
|
|
'No portal found!?\n'
|
|
'Why is this supposed caller context missing it?'
|
|
)
|
|
|
|
cid: str = self.cid
|
|
with trio.move_on_after(timeout) as cs:
|
|
cs.shield = True
|
|
log.cancel(
|
|
header
|
|
+
|
|
reminfo
|
|
)
|
|
|
|
# NOTE: we're telling the far end actor to cancel a task
|
|
# corresponding to *this actor*. The far end local channel
|
|
# instance is passed to `Actor._cancel_task()` implicitly.
|
|
await self._portal.run_from_ns(
|
|
'self',
|
|
'_cancel_task',
|
|
cid=cid,
|
|
)
|
|
|
|
if cs.cancelled_caught:
|
|
# XXX: there's no way to know if the remote task was indeed
|
|
# cancelled in the case where the connection is broken or
|
|
# some other network error occurred.
|
|
# if not self._portal.channel.connected():
|
|
if not self.chan.connected():
|
|
log.cancel(
|
|
'May have failed to cancel remote task?\n'
|
|
f'{reminfo}'
|
|
)
|
|
else:
|
|
log.cancel(
|
|
'Timed out on cancel request of remote task?\n'
|
|
f'{reminfo}'
|
|
)
|
|
|
|
# CALLEE side task
|
|
# ------ - ------
|
|
# Aka the one that DID NOT EVER enter a `Portal.open_context()`
|
|
# and instead was constructed and scheduled as an
|
|
# `_invoke()` RPC task.
|
|
#
|
|
# NOTE: on this side we ALWAYS cancel the local scope
|
|
# since the caller expects a `ContextCancelled` to be sent
|
|
# from `._runtime._invoke()` back to the other side. The
|
|
# logic for catching the result of the below
|
|
# `._scope.cancel()` is inside the `._runtime._invoke()`
|
|
# context RPC handling block.
|
|
else:
|
|
log.cancel(
|
|
header
|
|
+
|
|
reminfo
|
|
)
|
|
# TODO: should we have an explicit cancel message
|
|
# or is relaying the local `trio.Cancelled` as an
|
|
# {'error': trio.Cancelled, cid: "blah"} enough?
|
|
# This probably gets into the discussion in
|
|
# https://github.com/goodboy/tractor/issues/36
|
|
assert self._scope
|
|
self._scope.cancel()
|
|
|
|
@acm
|
|
async def open_stream(
|
|
self,
|
|
allow_overruns: bool | None = False,
|
|
msg_buffer_size: int | None = None,
|
|
|
|
) -> AsyncGenerator[MsgStream, None]:
|
|
'''
|
|
Open a ``MsgStream``, a bi-directional stream connected to the
|
|
cross-actor (far end) task for this ``Context``.
|
|
|
|
This context manager must be entered on both the caller and
|
|
callee for the stream to logically be considered "connected".
|
|
|
|
A ``MsgStream`` is currently "one-shot" use, meaning if you
|
|
close it you can not "re-open" it for streaming and instead you
|
|
must re-establish a new surrounding ``Context`` using
|
|
``Portal.open_context()``. In the future this may change but
|
|
currently there seems to be no obvious reason to support
|
|
"re-opening":
|
|
- pausing a stream can be done with a message.
|
|
- task errors will normally require a restart of the entire
|
|
scope of the inter-actor task context due to the nature of
|
|
``trio``'s cancellation system.
|
|
|
|
'''
|
|
actor: Actor = self._actor
|
|
|
|
# If the surrounding context has been cancelled by some
|
|
# task with a handle to THIS, we error here immediately
|
|
# since it likely means the surrounding lexical-scope has
|
|
# errored, been `trio.Cancelled` or at the least
|
|
# `Context.cancel()` was called by some task.
|
|
if self._cancel_called:
|
|
|
|
# XXX NOTE: ALWAYS RAISE any remote error here even if
|
|
# it's an expected `ContextCancelled` due to a local
|
|
# task having called `.cancel()`!
|
|
#
|
|
# WHY: we expect the error to always bubble up to the
|
|
# surrounding `Portal.open_context()` call and be
|
|
# absorbed there (silently) and we DO NOT want to
|
|
# actually try to stream - a cancel msg was already
|
|
# sent to the other side!
|
|
if self._remote_error:
|
|
# NOTE: this is diff then calling
|
|
# `._maybe_raise_remote_err()` specifically
|
|
# because any task entering this `.open_stream()`
|
|
# AFTER cancellation has already been requested,
|
|
# we DO NOT want to absorb any ctxc ACK silently!
|
|
raise self._remote_error
|
|
|
|
# XXX NOTE: if no `ContextCancelled` has been responded
|
|
# back from the other side (yet), we raise a different
|
|
# runtime error indicating that this task's usage of
|
|
# `Context.cancel()` and then `.open_stream()` is WRONG!
|
|
task: str = trio.lowlevel.current_task().name
|
|
raise RuntimeError(
|
|
'Stream opened after `Context.cancel()` called..?\n'
|
|
f'task: {actor.uid[0]}:{task}\n'
|
|
f'{self}'
|
|
)
|
|
|
|
if (
|
|
not self._portal
|
|
and not self._started_called
|
|
):
|
|
raise RuntimeError(
|
|
'Context.started()` must be called before opening a stream'
|
|
)
|
|
|
|
# NOTE: in one way streaming this only happens on the
|
|
# caller side inside `Actor.start_remote_task()` so if you try
|
|
# to send a stop from the caller to the callee in the
|
|
# single-direction-stream case you'll get a lookup error
|
|
# currently.
|
|
ctx: Context = actor.get_context(
|
|
chan=self.chan,
|
|
cid=self.cid,
|
|
nsf=self._nsf,
|
|
msg_buffer_size=msg_buffer_size,
|
|
allow_overruns=allow_overruns,
|
|
)
|
|
ctx._allow_overruns: bool = allow_overruns
|
|
assert ctx is self
|
|
|
|
# XXX: If the underlying channel feeder receive mem chan has
|
|
# been closed then likely client code has already exited
|
|
# a ``.open_stream()`` block prior or there was some other
|
|
# unanticipated error or cancellation from ``trio``.
|
|
|
|
if ctx._recv_chan._closed:
|
|
raise trio.ClosedResourceError(
|
|
'The underlying channel for this stream was already closed!\n'
|
|
)
|
|
|
|
# NOTE: implicitly this will call `MsgStream.aclose()` on
|
|
# `.__aexit__()` due to stream's parent `Channel` type!
|
|
#
|
|
# XXX NOTE XXX: ensures the stream is "one-shot use",
|
|
# which specifically means that on exit,
|
|
# - signal ``trio.EndOfChannel``/``StopAsyncIteration`` to
|
|
# the far end indicating that the caller exited
|
|
# the streaming context purposefully by letting
|
|
# the exit block exec.
|
|
# - this is diff from the cancel/error case where
|
|
# a cancel request from this side or an error
|
|
# should be sent to the far end indicating the
|
|
# stream WAS NOT just closed normally/gracefully.
|
|
async with MsgStream(
|
|
ctx=self,
|
|
rx_chan=ctx._recv_chan,
|
|
) as stream:
|
|
|
|
# NOTE: we track all existing streams per portal for
|
|
# the purposes of attempting graceful closes on runtime
|
|
# cancel requests.
|
|
if self._portal:
|
|
self._portal._streams.add(stream)
|
|
|
|
try:
|
|
self._stream_opened: bool = True
|
|
self._stream = stream
|
|
|
|
# XXX: do we need this?
|
|
# ensure we aren't cancelled before yielding the stream
|
|
# await trio.lowlevel.checkpoint()
|
|
yield stream
|
|
|
|
|
|
# XXX: (MEGA IMPORTANT) if this is a root opened process we
|
|
# wait for any immediate child in debug before popping the
|
|
# context from the runtime msg loop otherwise inside
|
|
# ``Actor._push_result()`` the msg will be discarded and in
|
|
# the case where that msg is global debugger unlock (via
|
|
# a "stop" msg for a stream), this can result in a deadlock
|
|
# where the root is waiting on the lock to clear but the
|
|
# child has already cleared it and clobbered IPC.
|
|
#
|
|
# await maybe_wait_for_debugger()
|
|
|
|
# XXX TODO: pretty sure this isn't needed (see
|
|
# note above this block) AND will result in
|
|
# a double `.send_stop()` call. The only reason to
|
|
# put it here would be to due with "order" in
|
|
# terms of raising any remote error (as per
|
|
# directly below) or bc the stream's
|
|
# `.__aexit__()` block might not get run
|
|
# (doubtful)? Either way if we did put this back
|
|
# in we also need a state var to avoid the double
|
|
# stop-msg send..
|
|
#
|
|
# await stream.aclose()
|
|
|
|
# if re := ctx._remote_error:
|
|
# ctx._maybe_raise_remote_err(
|
|
# re,
|
|
# raise_ctxc_from_self_call=True,
|
|
# )
|
|
# await trio.lowlevel.checkpoint()
|
|
|
|
finally:
|
|
if self._portal:
|
|
try:
|
|
self._portal._streams.remove(stream)
|
|
except KeyError:
|
|
log.warning(
|
|
f'Stream was already destroyed?\n'
|
|
f'actor: {self.chan.uid}\n'
|
|
f'ctx id: {self.cid}'
|
|
)
|
|
|
|
# TODO: replace all the instances of this!! XD
|
|
def maybe_raise(
|
|
self,
|
|
**kwargs,
|
|
) -> Exception|None:
|
|
if re := self._remote_error:
|
|
return self._maybe_raise_remote_err(
|
|
re,
|
|
**kwargs,
|
|
)
|
|
|
|
def _maybe_raise_remote_err(
|
|
self,
|
|
remote_error: Exception,
|
|
raise_ctxc_from_self_call: bool = False,
|
|
raise_overrun_from_self: bool = True,
|
|
|
|
) -> (
|
|
ContextCancelled # `.cancel()` request to far side
|
|
|RemoteActorError # stream overrun caused and ignored by us
|
|
):
|
|
'''
|
|
Maybe raise a remote error depending on the type of error
|
|
and *who* (i.e. which task from which actor) requested
|
|
a cancellation (if any).
|
|
|
|
'''
|
|
our_uid: tuple = self.chan.uid
|
|
|
|
# XXX NOTE XXX: `ContextCancelled`/`StreamOverrun` absorption
|
|
# for "graceful cancellation" case:
|
|
#
|
|
# Whenever a "side" of a context (a `trio.Task` running in
|
|
# an actor) **is** the side which requested ctx
|
|
# cancellation (likekly via ``Context.cancel()``), we
|
|
# **don't** want to re-raise any eventually received
|
|
# `ContextCancelled` response locally (would be akin to
|
|
# a `trio.Nursery` nursery raising `trio.Cancelled`
|
|
# whenever `CancelScope.cancel()` was called).
|
|
#
|
|
# Instead, silently reap the remote delivered ctxc
|
|
# (`ContextCancelled`) as an expected
|
|
# error-msg-is-cancellation-ack IFF said
|
|
# `remote_error: ContextCancelled` has `.canceller`
|
|
# set to the `Actor.uid` of THIS task (i.e. the
|
|
# cancellation requesting task's actor is the actor
|
|
# checking whether it should absorb the ctxc).
|
|
if (
|
|
not raise_ctxc_from_self_call
|
|
and self._is_self_cancelled(remote_error)
|
|
|
|
# TODO: ?potentially it is useful to emit certain
|
|
# warning/cancel logs for the cases where the
|
|
# cancellation is due to a lower level cancel
|
|
# request, such as `Portal.cancel_actor()`, since in
|
|
# that case it's not actually this specific ctx that
|
|
# made a `.cancel()` call, but it is the same
|
|
# actor-process?
|
|
# or self.chan._cancel_called
|
|
# XXX: ^ should we have a special separate case
|
|
# for this ^, NO right?
|
|
|
|
) or (
|
|
# NOTE: whenever this context is the cause of an
|
|
# overrun on the remote side (aka we sent msgs too
|
|
# fast that the remote task was overrun according
|
|
# to `MsgStream` buffer settings) AND the caller
|
|
# has requested to not raise overruns this side
|
|
# caused, we also silently absorb any remotely
|
|
# boxed `StreamOverrun`. This is mostly useful for
|
|
# supressing such faults during
|
|
# cancellation/error/final-result handling inside
|
|
# `_drain_to_final_msg()` such that we do not
|
|
# raise such errors particularly in the case where
|
|
# `._cancel_called == True`.
|
|
not raise_overrun_from_self
|
|
and isinstance(remote_error, RemoteActorError)
|
|
and remote_error.msgdata['type_str'] == 'StreamOverrun'
|
|
and tuple(remote_error.msgdata['sender']) == our_uid
|
|
):
|
|
# NOTE: we set the local scope error to any "self
|
|
# cancellation" error-response thus "absorbing"
|
|
# the error silently B)
|
|
if self._local_error is None:
|
|
self._local_error = remote_error
|
|
|
|
else:
|
|
log.warning(
|
|
'Local error already set for ctx?\n'
|
|
f'{self._local_error}\n'
|
|
)
|
|
|
|
return remote_error
|
|
|
|
# NOTE: currently we are hiding underlying runtime errors
|
|
# which are often superfluous to user handler code. not
|
|
# sure if this is still needed / desired for all operation?
|
|
# TODO: maybe we can only NOT mask if:
|
|
# - [ ] debug mode is enabled or,
|
|
# - [ ] a certain log level is set?
|
|
# - [ ] consider using `.with_traceback()` to filter out
|
|
# runtime frames from the tb explicitly?
|
|
# https://docs.python.org/3/reference/simple_stmts.html#the-raise-statement
|
|
# https://stackoverflow.com/a/24752607
|
|
__tracebackhide__: bool = True
|
|
raise remote_error from None
|
|
|
|
# TODO: change to `.wait_for_result()`?
|
|
async def result(
|
|
self,
|
|
hide_tb: bool = False,
|
|
|
|
) -> Any|Exception:
|
|
'''
|
|
From some (caller) side task, wait for and return the final
|
|
result from the remote (callee) side's task.
|
|
|
|
This provides a mechanism for one task running in some actor to wait
|
|
on another task at the other side, in some other actor, to terminate.
|
|
|
|
If the remote task is still in a streaming state (it is delivering
|
|
values from inside a ``Context.open_stream():`` block, then those
|
|
msgs are drained but discarded since it is presumed this side of
|
|
the context has already finished with its own streaming logic.
|
|
|
|
If the remote context (or its containing actor runtime) was
|
|
canceled, either by a local task calling one of
|
|
``Context.cancel()`` or `Portal.cancel_actor()``, we ignore the
|
|
received ``ContextCancelled`` exception if the context or
|
|
underlying IPC channel is marked as having been "cancel called".
|
|
This is similar behavior to using ``trio.Nursery.cancel()``
|
|
wherein tasks which raise ``trio.Cancel`` are silently reaped;
|
|
the main different in this API is in the "cancel called" case,
|
|
instead of just not raising, we also return the exception *as
|
|
the result* since client code may be interested in the details
|
|
of the remote cancellation.
|
|
|
|
'''
|
|
__tracebackhide__ = hide_tb
|
|
assert self._portal, (
|
|
"Context.result() can not be called from callee side!"
|
|
)
|
|
if self._final_result_is_set():
|
|
return self._result
|
|
|
|
assert self._recv_chan
|
|
raise_overrun: bool = not self._allow_overruns
|
|
# res_placeholder: int = id(self)
|
|
if (
|
|
# self._result == res_placeholder
|
|
# and not self._remote_error
|
|
self.maybe_error is None
|
|
# not self._remote_error
|
|
# and not self._local_error
|
|
and not self._recv_chan._closed # type: ignore
|
|
):
|
|
|
|
# wait for a final context result/error by "draining"
|
|
# (by more or less ignoring) any bi-dir-stream "yield"
|
|
# msgs still in transit from the far end.
|
|
drained_msgs: list[dict] = await _drain_to_final_msg(
|
|
ctx=self,
|
|
hide_tb=hide_tb,
|
|
)
|
|
for msg in drained_msgs:
|
|
|
|
# TODO: mask this by default..
|
|
if 'return' in msg:
|
|
# from .devx import pause
|
|
# await pause()
|
|
raise InternalError(
|
|
'Final `return` msg should never be drained !?!?\n\n'
|
|
f'{msg}\n'
|
|
)
|
|
|
|
log.cancel(
|
|
'Ctx drained pre-result msgs:\n'
|
|
f'{drained_msgs}'
|
|
)
|
|
|
|
if (
|
|
(re := self._remote_error)
|
|
# and self._result == res_placeholder
|
|
):
|
|
self._maybe_raise_remote_err(
|
|
re,
|
|
# NOTE: obvi we don't care if we
|
|
# overran the far end if we're already
|
|
# waiting on a final result (msg).
|
|
# raise_overrun_from_self=False,
|
|
raise_overrun_from_self=(
|
|
raise_overrun
|
|
and
|
|
# only when we ARE NOT the canceller
|
|
# should we raise overruns, bc ow we're
|
|
# raising something we know might happen
|
|
# during cancellation ;)
|
|
(not self._cancel_called)
|
|
),
|
|
)
|
|
# if maybe_err:
|
|
# self._result = maybe_err
|
|
|
|
return self.outcome
|
|
# None if self._result == res_placeholder
|
|
# else self._result
|
|
# )
|
|
|
|
# TODO: switch this with above which should be named
|
|
# `.wait_for_outcome()` and instead do
|
|
# a `.outcome.Outcome.unwrap()` ?
|
|
# @property
|
|
# def result(self) -> Any|None:
|
|
# if self._final_result_is_set():
|
|
# return self._result
|
|
|
|
# raise RuntimeError('No result is available!')
|
|
|
|
@property
|
|
def maybe_error(self) -> BaseException|None:
|
|
le: Exception|None = self._local_error
|
|
re: RemoteActorError|ContextCancelled|None = self._remote_error
|
|
|
|
match (le, re):
|
|
# NOTE: remote errors always get precedence since even
|
|
# in the cases where a local error was the cause, the
|
|
# received boxed ctxc should include the src info
|
|
# caused by us right?
|
|
case (
|
|
_,
|
|
RemoteActorError(),
|
|
):
|
|
# give precedence to remote error if it's
|
|
# NOT a cancel ack (ctxc).
|
|
return (
|
|
re or le
|
|
)
|
|
|
|
# TODO: extra logic to handle ctxc ack case(s)?
|
|
# -[ ] eg. we error, call .cancel(), rx ack but should
|
|
# raise the _local_error instead?
|
|
# -[ ] are there special error conditions where local vs.
|
|
# remote should take precedence?
|
|
# case (
|
|
# _,
|
|
# ContextCancelled(canceller=),
|
|
# ):
|
|
|
|
error: Exception|None = le or re
|
|
if error:
|
|
return error
|
|
|
|
if cancmsg := self._cancel_msg:
|
|
# NOTE: means we're prolly in the process of
|
|
# processing the cancellation caused by
|
|
# this msg (eg. logging from `Actor._cancel_task()`
|
|
# method after receiving a `Context.cancel()` RPC)
|
|
# though there shouldn't ever be a `._cancel_msg`
|
|
# without it eventually resulting in this property
|
|
# delivering a value!
|
|
log.debug(
|
|
'`Context._cancel_msg` is set but has not yet resolved to `.maybe_error`?\n\n'
|
|
f'{cancmsg}\n'
|
|
)
|
|
|
|
# assert not self._cancel_msg
|
|
return None
|
|
|
|
def _final_result_is_set(self) -> bool:
|
|
# return not (self._result == id(self))
|
|
return self._result is not Unresolved
|
|
|
|
# def get_result_nowait(self) -> Any|None:
|
|
# TODO: use `outcome.Outcome` here instead?
|
|
@property
|
|
def outcome(self) -> (
|
|
Any|
|
|
RemoteActorError|
|
|
ContextCancelled
|
|
):
|
|
'''
|
|
The final "outcome" from an IPC context which can either be
|
|
some Value returned from the target `@context`-decorated
|
|
remote task-as-func, or an `Error` wrapping an exception
|
|
raised from an RPC task fault or cancellation.
|
|
|
|
Note that if the remote task has not terminated then this
|
|
field always resolves to the module defined `Unresolved` handle.
|
|
|
|
TODO: implement this using `outcome.Outcome` types?
|
|
|
|
'''
|
|
return self.maybe_error or self._result
|
|
|
|
# @property
|
|
def repr_outcome(
|
|
self,
|
|
show_error_fields: bool = False,
|
|
type_only: bool = False,
|
|
|
|
) -> str:
|
|
'''
|
|
Deliver a (simplified) `str` representation (as in
|
|
`.__repr__()`) of the final `.outcome`
|
|
|
|
'''
|
|
merr: Exception|None = self.maybe_error
|
|
if merr:
|
|
if type_only:
|
|
return type(merr).__name__
|
|
|
|
# if the error-type is one of ours and has the custom
|
|
# defined "repr-(in)-one-line" method call it, ow
|
|
# just deliver the type name.
|
|
if (
|
|
(reprol := getattr(merr, 'reprol', False))
|
|
and show_error_fields
|
|
):
|
|
return reprol()
|
|
|
|
elif isinstance(merr, BaseExceptionGroup):
|
|
# TODO: maybe for multis we should just show
|
|
# a one-line count per error type, like with
|
|
# `collections.Counter`?
|
|
#
|
|
# just the type name for now to avoid long lines
|
|
# when tons of cancels..
|
|
return type(merr).__name__
|
|
|
|
# just the type name
|
|
# else: # but wen?
|
|
# return type(merr).__name__
|
|
|
|
# for all other errors show their regular output
|
|
return str(merr)
|
|
|
|
return str(self._result)
|
|
|
|
async def started(
|
|
self,
|
|
value: Any | None = None
|
|
|
|
) -> None:
|
|
'''
|
|
Indicate to calling actor's task that this linked context
|
|
has started and send ``value`` to the other side via IPC.
|
|
|
|
On the calling side ``value`` is the second item delivered
|
|
in the tuple returned by ``Portal.open_context()``.
|
|
|
|
'''
|
|
if self._portal:
|
|
raise RuntimeError(
|
|
f'Caller side context {self} can not call started!'
|
|
)
|
|
|
|
elif self._started_called:
|
|
raise RuntimeError(
|
|
f'called `.started()` twice on context with {self.chan.uid}'
|
|
)
|
|
|
|
await self.chan.send({'started': value, 'cid': self.cid})
|
|
self._started_called = True
|
|
|
|
async def _drain_overflows(
|
|
self,
|
|
) -> None:
|
|
'''
|
|
Private task spawned to push newly received msgs to the local
|
|
task which getting overrun by the remote side.
|
|
|
|
In order to not block the rpc msg loop, but also not discard
|
|
msgs received in this context, we need to async push msgs in
|
|
a new task which only runs for as long as the local task is in
|
|
an overrun state.
|
|
|
|
'''
|
|
self._in_overrun = True
|
|
try:
|
|
while self._overflow_q:
|
|
# NOTE: these msgs should never be errors since we always do
|
|
# the check prior to checking if we're in an overrun state
|
|
# inside ``._deliver_msg()``.
|
|
msg = self._overflow_q.popleft()
|
|
try:
|
|
await self._send_chan.send(msg)
|
|
except trio.BrokenResourceError:
|
|
log.warning(
|
|
f"{self._send_chan} consumer is already closed"
|
|
)
|
|
return
|
|
except trio.Cancelled:
|
|
# we are obviously still in overrun
|
|
# but the context is being closed anyway
|
|
# so we just warn that there are un received
|
|
# msgs still..
|
|
self._overflow_q.appendleft(msg)
|
|
fmt_msgs = ''
|
|
for msg in self._overflow_q:
|
|
fmt_msgs += f'{pformat(msg)}\n'
|
|
|
|
log.warning(
|
|
f'Context for {self.cid} is being closed while '
|
|
'in an overrun state!\n'
|
|
'Discarding the following msgs:\n'
|
|
f'{fmt_msgs}\n'
|
|
)
|
|
raise
|
|
|
|
finally:
|
|
# task is now finished with the backlog so mark us as
|
|
# no longer in backlog.
|
|
self._in_overrun = False
|
|
|
|
async def _deliver_msg(
|
|
self,
|
|
msg: dict,
|
|
|
|
) -> bool:
|
|
'''
|
|
Deliver an IPC msg received from a transport-channel to
|
|
this context's underlying mem chan for handling by local
|
|
user application tasks; deliver `bool` indicating whether
|
|
the msg was able to be delivered.
|
|
|
|
If `._allow_overruns == True` (maybe) append the msg to an
|
|
"overflow queue" and start a "drainer task" (inside the
|
|
`._scope_nursery: trio.Nursery`) which ensures that such
|
|
messages are queued up and eventually sent if possible.
|
|
|
|
'''
|
|
cid: str = self.cid
|
|
chan: Channel = self.chan
|
|
from_uid: tuple[str, str] = chan.uid
|
|
send_chan: trio.MemorySendChannel = self._send_chan
|
|
nsf: NamespacePath = self._nsf
|
|
|
|
re: Exception|None
|
|
if re := unpack_error(
|
|
msg,
|
|
self.chan,
|
|
):
|
|
if not isinstance(re, ContextCancelled):
|
|
log_meth = log.error
|
|
else:
|
|
log_meth = log.runtime
|
|
|
|
log_meth(
|
|
f'Delivering error-msg to caller\n\n'
|
|
|
|
f'<= peer: {from_uid}\n'
|
|
f' |_ {nsf}()\n\n'
|
|
|
|
f'=> cid: {cid}\n'
|
|
f' |_{self._task}\n\n'
|
|
|
|
f'{pformat(re)}\n'
|
|
)
|
|
self._cancel_msg: dict = msg
|
|
|
|
# NOTE: this will not raise an error, merely set
|
|
# `._remote_error` and maybe cancel any task currently
|
|
# entered in `Portal.open_context()` presuming the
|
|
# error is "cancel causing" (i.e. a `ContextCancelled`
|
|
# or `RemoteActorError`).
|
|
self._maybe_cancel_and_set_remote_error(re)
|
|
|
|
# XXX NEVER do this XXX..!!
|
|
# bc if the error is a ctxc and there is a task
|
|
# waiting on `.result()` we need the msg to be sent
|
|
# over the `send_chan`/`._recv_chan` so that the error
|
|
# is relayed to that waiter task..
|
|
# return True
|
|
#
|
|
# XXX ALSO NO!! XXX
|
|
# => NEVER raise remote errors from the calling
|
|
# runtime task, they should always be raised by
|
|
# consumer side tasks operating on the
|
|
# `Portal`/`Context` APIs.
|
|
# if self._remote_error:
|
|
# self._maybe_raise_remote_err(error)
|
|
|
|
if self._in_overrun:
|
|
log.warning(
|
|
f'Queueing OVERRUN msg on caller task:\n'
|
|
f'<= peer: {from_uid}\n'
|
|
f' |_ {nsf}()\n\n'
|
|
|
|
f'=> cid: {cid}\n'
|
|
f' |_{self._task}\n\n'
|
|
|
|
f'{pformat(msg)}\n'
|
|
)
|
|
self._overflow_q.append(msg)
|
|
return False
|
|
|
|
try:
|
|
log.runtime(
|
|
f'Delivering msg from IPC ctx:\n'
|
|
f'<= {from_uid}\n'
|
|
f' |_ {nsf}()\n\n'
|
|
|
|
f'=> {self._task}\n'
|
|
f' |_cid={self.cid}\n\n'
|
|
|
|
f'{pformat(msg)}\n'
|
|
)
|
|
|
|
# NOTE: if an error is deteced we should always still
|
|
# send it through the feeder-mem-chan and expect
|
|
# it to be raised by any context (stream) consumer
|
|
# task via the consumer APIs on both the `Context` and
|
|
# `MsgStream`!
|
|
#
|
|
# XXX the reason is that this method is always called
|
|
# by the IPC msg handling runtime task and that is not
|
|
# normally the task that should get cancelled/error
|
|
# from some remote fault!
|
|
send_chan.send_nowait(msg)
|
|
return True
|
|
|
|
except trio.BrokenResourceError:
|
|
# TODO: what is the right way to handle the case where the
|
|
# local task has already sent a 'stop' / StopAsyncInteration
|
|
# to the other side but and possibly has closed the local
|
|
# feeder mem chan? Do we wait for some kind of ack or just
|
|
# let this fail silently and bubble up (currently)?
|
|
|
|
# XXX: local consumer has closed their side
|
|
# so cancel the far end streaming task
|
|
log.warning(
|
|
'Rx chan for `Context` alfready closed?\n'
|
|
f'cid: {self.cid}\n'
|
|
'Failed to deliver msg:\n'
|
|
f'send_chan: {send_chan}\n\n'
|
|
f'{pformat(msg)}\n'
|
|
)
|
|
return False
|
|
|
|
# NOTE XXX: by default we do **not** maintain context-stream
|
|
# backpressure and instead opt to relay stream overrun errors to
|
|
# the sender; the main motivation is that using bp can block the
|
|
# msg handling loop which calls into this method!
|
|
except trio.WouldBlock:
|
|
|
|
# XXX: always push an error even if the local receiver
|
|
# is in overrun state - i.e. if an 'error' msg is
|
|
# delivered then
|
|
# `._maybe_cancel_and_set_remote_error(msg)` should
|
|
# have already been called above!
|
|
#
|
|
# XXX QUESTION XXX: if we rx an error while in an
|
|
# overrun state and that msg isn't stuck in an
|
|
# overflow queue what happens?!?
|
|
|
|
local_uid = self._actor.uid
|
|
txt: str = (
|
|
'on IPC context:\n'
|
|
|
|
f'<= sender: {from_uid}\n'
|
|
f' |_ {self._nsf}()\n\n'
|
|
|
|
f'=> overrun: {local_uid}\n'
|
|
f' |_cid: {cid}\n'
|
|
f' |_task: {self._task}\n'
|
|
)
|
|
if not self._stream_opened:
|
|
txt += (
|
|
f'\n*** No stream open on `{local_uid[0]}` side! ***\n\n'
|
|
f'{msg}\n'
|
|
)
|
|
|
|
# XXX: lul, this really can't be backpressure since any
|
|
# blocking here will block the entire msg loop rpc sched for
|
|
# a whole channel.. maybe we should rename it?
|
|
if self._allow_overruns:
|
|
txt += (
|
|
'\n*** Starting overflow queuing task on msg ***\n\n'
|
|
f'{msg}\n'
|
|
)
|
|
log.warning(txt)
|
|
if (
|
|
not self._in_overrun
|
|
):
|
|
self._overflow_q.append(msg)
|
|
tn: trio.Nursery = self._scope_nursery
|
|
assert not tn.child_tasks
|
|
try:
|
|
tn.start_soon(
|
|
self._drain_overflows,
|
|
)
|
|
return True
|
|
|
|
except RuntimeError:
|
|
# if the nursery is already cancelled due to
|
|
# this context exiting or in error, we ignore
|
|
# the nursery error since we never expected
|
|
# anything different.
|
|
return False
|
|
else:
|
|
txt += f'\n{msg}\n'
|
|
# raise local overrun and immediately pack as IPC
|
|
# msg for far end.
|
|
try:
|
|
raise StreamOverrun(
|
|
txt,
|
|
sender=from_uid,
|
|
)
|
|
except StreamOverrun as err:
|
|
err_msg: dict[str, dict] = pack_error(
|
|
err,
|
|
cid=cid,
|
|
)
|
|
try:
|
|
# relay condition to sender side remote task
|
|
await chan.send(err_msg)
|
|
return True
|
|
|
|
except trio.BrokenResourceError:
|
|
# XXX: local consumer has closed their side
|
|
# so cancel the far end streaming task
|
|
log.warning(
|
|
'Channel for ctx is already closed?\n'
|
|
f'|_{chan}\n'
|
|
)
|
|
|
|
# ow, indicate unable to deliver by default
|
|
return False
|
|
|
|
|
|
def mk_context(
|
|
chan: Channel,
|
|
cid: str,
|
|
nsf: NamespacePath,
|
|
|
|
msg_buffer_size: int = 2**6,
|
|
|
|
**kwargs,
|
|
|
|
) -> Context:
|
|
'''
|
|
Internal factory to create an inter-actor task ``Context``.
|
|
|
|
This is called by internals and should generally never be called
|
|
by user code.
|
|
|
|
'''
|
|
send_chan: trio.MemorySendChannel
|
|
recv_chan: trio.MemoryReceiveChannel
|
|
send_chan, recv_chan = trio.open_memory_channel(msg_buffer_size)
|
|
|
|
ctx = Context(
|
|
chan=chan,
|
|
cid=cid,
|
|
_actor=current_actor(),
|
|
_send_chan=send_chan,
|
|
_recv_chan=recv_chan,
|
|
_nsf=nsf,
|
|
_task=trio.lowlevel.current_task(),
|
|
**kwargs,
|
|
)
|
|
# TODO: we can drop the old placeholder yah?
|
|
# ctx._result: int | Any = id(ctx)
|
|
ctx._result = Unresolved
|
|
return ctx
|
|
|
|
|
|
def context(func: Callable) -> Callable:
|
|
'''
|
|
Mark an async function as a streaming routine with ``@context``.
|
|
|
|
'''
|
|
# TODO: apply whatever solution ``mypy`` ends up picking for this:
|
|
# https://github.com/python/mypy/issues/2087#issuecomment-769266912
|
|
func._tractor_context_function = True # type: ignore
|
|
|
|
sig = inspect.signature(func)
|
|
params = sig.parameters
|
|
if 'ctx' not in params:
|
|
raise TypeError(
|
|
"The first argument to the context function "
|
|
f"{func.__name__} must be `ctx: tractor.Context`"
|
|
)
|
|
return func
|