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

Return `Dict[Tuple, Tuple]` from `.get_registry()`

ensure_deregister
Tyler Goodlet 2020-08-03 18:42:23 -04:00
parent fbd68d2d91
commit 56b81f07e5
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/_actor.py
View File

@ -440,7 +440,7 @@ class Actor:
f"Cancelling all tasks for {chan} from {chan.uid}") f"Cancelling all tasks for {chan} from {chan.uid}")
for (channel, cid) in self._rpc_tasks: for (channel, cid) in self._rpc_tasks:
if channel is chan: if channel is chan:
self._cancel_task(cid, channel) await self._cancel_task(cid, channel)
log.debug( log.debug(
f"Msg loop signalled to terminate for" f"Msg loop signalled to terminate for"
f" {chan} from {chan.uid}") f" {chan} from {chan.uid}")
@ -871,7 +871,11 @@ class Arbiter(Actor):
) -> Dict[str, Tuple[str, str]]: ) -> Dict[str, Tuple[str, str]]:
"""Return current name registry. """Return current name registry.
""" """
return list(self._registry) # NOTE: requires ``strict_map_key=False`` to the msgpack
# unpacker since we have tuples as keys (not this makes the
# arbiter suscetible to hashdos):
# https://github.com/msgpack/msgpack-python#major-breaking-changes-in-msgpack-10
return self._registry
async def wait_for_actor( async def wait_for_actor(
self, name: str self, name: str