MirrorHub/synapse
0
0
Fork 1
mirror of https://mau.dev/maunium/synapse.git synced 2024-12-15 20:13:51 +01:00
Code Issues Projects Releases Wiki Activity

Add cache to remote room lists

Browse source 
Poll for updates from remote servers, waiting for the poll if there's no cache entry.
This commit is contained in:
David Baker 2016-05-31 18:27:23 +01:00
parent 6ca4d3ae9a
commit 2a449fec4d
1 changed files with 30 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
synapse/handlers/room.py
View file

@ -36,6 +36,8 @@ import string
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
REMOTE_ROOM_LIST_POLL_INTERVAL = 60 * 1000
id_server_scheme = "https://" id_server_scheme = "https://"
@ -344,6 +346,12 @@ class RoomListHandler(BaseHandler):
def __init__(self, hs): def __init__(self, hs):
super(RoomListHandler, self).__init__(hs) super(RoomListHandler, self).__init__(hs)
self.response_cache = ResponseCache() self.response_cache = ResponseCache()
self.remote_list_request_cache = ResponseCache()
self.remote_list_cache = {}
self.fetch_looping_call = hs.get_clock().looping_call(
self.fetch_all_remote_lists, REMOTE_ROOM_LIST_POLL_INTERVAL
)
self.fetch_all_remote_lists()
def get_local_public_room_list(self): def get_local_public_room_list(self):
result = self.response_cache.get(()) result = self.response_cache.get(())
@ -427,6 +435,14 @@ class RoomListHandler(BaseHandler):
# FIXME (erikj): START is no longer a valid value # FIXME (erikj): START is no longer a valid value
defer.returnValue({"start": "START", "end": "END", "chunk": results}) defer.returnValue({"start": "START", "end": "END", "chunk": results})
@defer.inlineCallbacks
def fetch_all_remote_lists(self):
deferred = self.hs.get_replication_layer().get_public_rooms(
self.hs.config.secondary_directory_servers
)
self.remote_list_request_cache.set((), deferred)
yield deferred
@defer.inlineCallbacks @defer.inlineCallbacks
def get_aggregated_public_room_list(self): def get_aggregated_public_room_list(self):
""" """
@ -434,9 +450,19 @@ class RoomListHandler(BaseHandler):
specified in the secondary_directory_servers config option. specified in the secondary_directory_servers config option.
XXX: Pagination... XXX: Pagination...
""" """
federated_by_server = yield self.hs.get_replication_layer().get_public_rooms( # We return the results from out cache which is updated by a looping call,
self.hs.config.secondary_directory_servers # unless we're missing a cache entry, in which case wait for the result
) # of the fetch if there's one in progress. If not, omit that server.
wait = False
for s in self.hs.config.secondary_directory_servers:
if s not in self.remote_list_cache:
logger.warn("No cached room list from %s: waiting for fetch", s)
wait = True
break
if wait and self.remote_list_request_cache.get(()):
yield self.remote_list_request_cache.get(())
public_rooms = yield self.get_local_public_room_list() public_rooms = yield self.get_local_public_room_list()
# keep track of which room IDs we've seen so we can de-dup # keep track of which room IDs we've seen so we can de-dup
@ -449,7 +475,7 @@ class RoomListHandler(BaseHandler):
room_ids.add(room["room_id"]) room_ids.add(room["room_id"])
# Now add the results from federation # Now add the results from federation
for server_name, server_result in federated_by_server.items(): for server_name, server_result in self.remote_list_cache.items():
for room in server_result["chunk"]: for room in server_result["chunk"]:
if room["room_id"] not in room_ids: if room["room_id"] not in room_ids:
room["server_name"] = server_name room["server_name"] = server_name