Review comments

pull/4942/head
Erik Johnston 2019-03-28 13:48:41 +00:00
parent d5a5d1c632
commit 40e56997bc
2 changed files with 73 additions and 51 deletions

View File

@ -934,6 +934,9 @@ class PresenceHandler(object):
joining rooms and require being sent presence. joining rooms and require being sent presence.
""" """
if self._event_processing:
return
@defer.inlineCallbacks @defer.inlineCallbacks
def _process_presence(): def _process_presence():
if self._event_processing: if self._event_processing:
@ -993,58 +996,73 @@ class PresenceHandler(object):
# Ignore changes to join events. # Ignore changes to join events.
continue continue
if self.is_mine_id(state_key): yield self._on_user_joined_room(room_id, state_key)
# If this is a local user then we need to send their presence
# out to hosts in the room (who don't already have it)
# TODO: We should be able to filter the hosts down to those that @defer.inlineCallbacks
# haven't previously seen the user def _on_user_joined_room(self, room_id, user_id):
"""Called when we detect a user joining the room via the current state
delta stream.
state = yield self.current_state_for_user(state_key) Args:
hosts = yield self.state.get_current_hosts_in_room(room_id) room_id (str)
user_id (str)
# Filter out ourselves. Returns:
hosts = set(host for host in hosts if host != self.server_name) Deferred
"""
if self.is_mine_id(user_id):
# If this is a local user then we need to send their presence
# out to hosts in the room (who don't already have it)
# TODO: We should be able to filter the hosts down to those that
# haven't previously seen the user
state = yield self.current_state_for_user(user_id)
hosts = yield self.state.get_current_hosts_in_room(room_id)
# Filter out ourselves.
hosts = set(host for host in hosts if host != self.server_name)
self.federation.send_presence_to_destinations(
states=[state],
destinations=hosts,
)
else:
# A remote user has joined the room, so we need to:
# 1. Check if this is a new server in the room
# 2. If so send any presence they don't already have for
# local users in the room.
# TODO: We should be able to filter the users down to those that
# the server hasn't previously seen
# TODO: Check that this is actually a new server joining the
# room.
user_ids = yield self.state.get_current_user_in_room(room_id)
user_ids = list(filter(self.is_mine_id, user_ids))
states = yield self.current_state_for_users(user_ids)
# Filter out old presence, i.e. offline presence states where
# the user hasn't been active for a week. We can change this
# depending on what we want the UX to be, but at the least we
# should filter out offline presence where the state is just the
# default state.
now = self.clock.time_msec()
states = [
state for state in states.values()
if state.state != PresenceState.OFFLINE
or now - state.last_active_ts < 7 * 24 * 60 * 60 * 1000
or state.status_msg is not None
]
if states:
self.federation.send_presence_to_destinations( self.federation.send_presence_to_destinations(
states=[state], states=states,
destinations=hosts, destinations=[get_domain_from_id(user_id)],
) )
else:
# A remote user has joined the room, so we need to:
# 1. Check if this is a new server in the room
# 2. If so send any presence they don't already have for
# local users in the room.
# TODO: We should be able to filter the users down to those that
# the server hasn't previously seen
# TODO: Check that this is actually a new server joining the
# room.
user_ids = yield self.state.get_current_user_in_room(room_id)
user_ids = list(filter(self.is_mine_id, user_ids))
states = yield self.current_state_for_users(user_ids)
# Filter out old presence, i.e. offline presence states where
# the user hasn't been active for a week. We can change this
# depending on what we want the UX to be, but at the least we
# should filter out offline presence where the state is just the
# default state.
now = self.clock.time_msec()
states = [
state for state in states.values()
if state.state != PresenceState.OFFLINE
or now - state.last_active_ts < 7 * 24 * 60 * 60 * 1000
or state.status_msg is not None
]
if states:
self.federation.send_presence_to_destinations(
states=states,
destinations=[get_domain_from_id(state_key)],
)
def should_notify(old_state, new_state): def should_notify(old_state, new_state):

View File

@ -461,7 +461,9 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase):
) )
self.reactor.pump([0]) # Wait for presence updates to be handled self.reactor.pump([0]) # Wait for presence updates to be handled
# Test that a new server gets told about existing presence # #
# Test that a new server gets told about existing presence
#
self.federation_sender.reset_mock() self.federation_sender.reset_mock()
@ -482,7 +484,9 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase):
destinations=["server2"], states=[expected_state] destinations=["server2"], states=[expected_state]
) )
# Test that only the new server gets sent presence and not existing servers # #
# Test that only the new server gets sent presence and not existing servers
#
self.federation_sender.reset_mock() self.federation_sender.reset_mock()
self._add_new_user(room_id, "@bob:server3") self._add_new_user(room_id, "@bob:server3")
@ -517,7 +521,9 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase):
self.reactor.pump([0]) # Wait for presence updates to be handled self.reactor.pump([0]) # Wait for presence updates to be handled
# Test that when a local join happens remote servers get told about it # #
# Test that when a local join happens remote servers get told about it
#
self.federation_sender.reset_mock() self.federation_sender.reset_mock()
@ -547,7 +553,6 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase):
room_version = self.get_success(self.store.get_room_version(room_id)) room_version = self.get_success(self.store.get_room_version(room_id))
# No we want to have other servers "join"
builder = EventBuilder( builder = EventBuilder(
state=self.state, state=self.state,
auth=self.auth, auth=self.auth,
@ -555,7 +560,6 @@ class PresenceJoinTestCase(unittest.HomeserverTestCase):
clock=self.clock, clock=self.clock,
hostname=hostname, hostname=hostname,
signing_key=self.random_signing_key, signing_key=self.random_signing_key,
format_version=room_version_to_event_format(room_version), format_version=room_version_to_event_format(room_version),
room_id=room_id, room_id=room_id,
type=EventTypes.Member, type=EventTypes.Member,