forked from goodboy/tractor
1
0
Fork 0

Enforce async funcs on callee side, convert arbiter methods

drop_sync_funcs
Tyler Goodlet 2021-04-27 08:46:54 -04:00
parent c2a1612bf5
commit 5798ef6796
1 changed files with 9 additions and 8 deletions

View File

@ -64,12 +64,13 @@ async def _invoke(
kwargs['ctx'] = ctx
treat_as_gen = True
# errors raised inside this block are propgated back to caller
try:
is_async_partial = False
is_async_gen_partial = False
if isinstance(func, partial):
is_async_partial = inspect.iscoroutinefunction(func.func)
is_async_gen_partial = inspect.isasyncgenfunction(func.func)
if not (
inspect.isasyncgenfunction(func) or
inspect.iscoroutinefunction(func)
):
raise TypeError(f'{func} must be an async function!')
coro = func(**kwargs)
@ -1036,7 +1037,7 @@ class Arbiter(Actor):
self._waiters = {}
super().__init__(*args, **kwargs)
def find_actor(self, name: str) -> Optional[Tuple[str, int]]:
async def find_actor(self, name: str) -> Optional[Tuple[str, int]]:
for uid, sockaddr in self._registry.items():
if name in uid:
return sockaddr
@ -1077,7 +1078,7 @@ class Arbiter(Actor):
return sockaddrs
def register_actor(
async def register_actor(
self, uid: Tuple[str, str], sockaddr: Tuple[str, int]
) -> None:
name, uuid = uid
@ -1090,5 +1091,5 @@ class Arbiter(Actor):
if isinstance(event, trio.Event):
event.set()
def unregister_actor(self, uid: Tuple[str, str]) -> None:
async def unregister_actor(self, uid: Tuple[str, str]) -> None:
self._registry.pop(uid, None)