Add measure on check_host_in_room

This commit is contained in:
Erik Johnston 2016-08-26 10:47:00 +01:00
parent 3f11953fcb
commit 25414b44a2

View File

@ -278,20 +278,21 @@ class Auth(object):
@defer.inlineCallbacks @defer.inlineCallbacks
def check_host_in_room(self, room_id, host): def check_host_in_room(self, room_id, host):
curr_state_id = yield self.state.get_current_state_ids(room_id) with Measure(self.clock, "check_host_in_room"):
curr_state_id = yield self.state.get_current_state_ids(room_id)
for (etype, state_key), event_id in curr_state_id.items(): for (etype, state_key), event_id in curr_state_id.items():
if etype == EventTypes.Member: if etype == EventTypes.Member:
try: try:
if get_domain_from_id(state_key) != host: if get_domain_from_id(state_key) != host:
continue
except:
logger.warn("state_key not user_id: %s", state_key)
continue continue
except:
logger.warn("state_key not user_id: %s", state_key)
continue
event = yield self.store.get_event(event_id, allow_none=True) event = yield self.store.get_event(event_id, allow_none=True)
if event and event.content["membership"] == Membership.JOIN: if event and event.content["membership"] == Membership.JOIN:
defer.returnValue(True) defer.returnValue(True)
defer.returnValue(False) defer.returnValue(False)