Make get_joined_hosts_for_room use get_users_in_room

pull/828/head
Erik Johnston 2016-06-02 12:47:06 +01:00
parent 065e739d6e
commit ccb56fc24b
1 changed files with 3 additions and 16 deletions

View File

@ -238,23 +238,10 @@ class RoomMemberStore(SQLBaseStore):
return results return results
@cached(max_entries=5000) @cachedInlineCallbacks(max_entries=5000)
def get_joined_hosts_for_room(self, room_id): def get_joined_hosts_for_room(self, room_id):
return self.runInteraction( user_ids = yield self.get_users_in_room(room_id)
"get_joined_hosts_for_room", defer.returnValue(set(get_domain_from_id(uid) for uid in user_ids))
self._get_joined_hosts_for_room_txn,
room_id,
)
def _get_joined_hosts_for_room_txn(self, txn, room_id):
rows = self._get_members_rows_txn(
txn,
room_id, membership=Membership.JOIN
)
joined_domains = set(get_domain_from_id(r["user_id"]) for r in rows)
return joined_domains
def _get_members_events_txn(self, txn, room_id, membership=None, user_id=None): def _get_members_events_txn(self, txn, room_id, membership=None, user_id=None):
rows = self._get_members_rows_txn( rows = self._get_members_rows_txn(