make FederationHandler.do_remotely_reject_invite async

This commit is contained in:
Richard van der Hoff 2020-02-03 16:19:18 +00:00
parent c3f296af32
commit 4286e429a7
2 changed files with 11 additions and 8 deletions

View File

@ -1531,9 +1531,10 @@ class FederationHandler(BaseHandler):
return event
@defer.inlineCallbacks
def do_remotely_reject_invite(self, target_hosts, room_id, user_id, content):
origin, event, room_version = yield self._make_and_verify_event(
async def do_remotely_reject_invite(
self, target_hosts: Iterable[str], room_id: str, user_id: str, content: JsonDict
) -> EventBase:
origin, event, room_version = await self._make_and_verify_event(
target_hosts, room_id, user_id, "leave", content=content
)
# Mark as outlier as we don't have any state for this event; we're not
@ -1549,10 +1550,10 @@ class FederationHandler(BaseHandler):
except ValueError:
pass
yield self.federation_client.send_leave(target_hosts, event)
await self.federation_client.send_leave(target_hosts, event)
context = yield self.state_handler.compute_event_context(event)
yield self.persist_events_and_notify([(event, context)])
context = await self.state_handler.compute_event_context(event)
await self.persist_events_and_notify([(event, context)])
return event

View File

@ -984,8 +984,10 @@ class RoomMemberMasterHandler(RoomMemberHandler):
"""
fed_handler = self.federation_handler
try:
ret = yield fed_handler.do_remotely_reject_invite(
remote_room_hosts, room_id, target.to_string(), content=content,
ret = yield defer.ensureDeferred(
fed_handler.do_remotely_reject_invite(
remote_room_hosts, room_id, target.to_string(), content=content,
)
)
return ret
except Exception as e: