From ae9016c06a886e87cc2382129ceac80c72e8f952 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Mon, 3 Aug 2020 18:46:18 -0400 Subject: [PATCH] Log on KBI cancelled termination --- tractor/_discovery.py | 4 +++- tractor/_entry.py | 6 ++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/tractor/_discovery.py b/tractor/_discovery.py index 3667c902..0407896b 100644 --- a/tractor/_discovery.py +++ b/tractor/_discovery.py @@ -23,6 +23,7 @@ async def get_arbiter( arbiter. """ actor = current_actor() + if not actor: raise RuntimeError("No actor instance has been defined yet?") @@ -38,7 +39,8 @@ async def get_arbiter( @asynccontextmanager async def find_actor( - name: str, arbiter_sockaddr: Tuple[str, int] = None + name: str, + arbiter_sockaddr: Tuple[str, int] = None ) -> typing.AsyncGenerator[Optional[Portal], None]: """Ask the arbiter to find actor(s) by name. diff --git a/tractor/_entry.py b/tractor/_entry.py index 1ccc0b56..883cc6b3 100644 --- a/tractor/_entry.py +++ b/tractor/_entry.py @@ -72,8 +72,10 @@ def _trio_main( actor._async_main, parent_addr=parent_addr ) + try: trio.run(trio_main) except KeyboardInterrupt: - pass # handle it the same way trio does? - log.info(f"Actor {actor.uid} terminated") \ No newline at end of file + log.warning(f"Actor {actor.uid} received KBI") + + log.info(f"Actor {actor.uid} terminated")