mirror of
https://mau.dev/maunium/synapse.git
synced 2024-12-14 16:13:50 +01:00
Make snapshot_all_rooms return results in the correct form, including start and end tokens.
This commit is contained in:
parent
709a92cee8
commit
4eb8f84aa8
2 changed files with 28 additions and 6 deletions
|
@ -269,11 +269,16 @@ class MessageHandler(BaseHandler):
|
|||
if event.membership != Membership.JOIN:
|
||||
continue
|
||||
try:
|
||||
messages = yield self.store.get_recent_events_for_room(
|
||||
messages, token = yield self.store.get_recent_events_for_room(
|
||||
event.room_id,
|
||||
limit=50,
|
||||
)
|
||||
d["messages"] = [m.get_dict() for m in messages]
|
||||
|
||||
d["messages"] = {
|
||||
"chunk": [m.get_dict() for m in messages],
|
||||
"start": token[0],
|
||||
"end": token[1],
|
||||
}
|
||||
except:
|
||||
pass
|
||||
|
||||
|
|
|
@ -104,19 +104,36 @@ class StreamStore(SQLBaseStore):
|
|||
def get_recent_events_for_room(self, room_id, limit, with_feedback=False):
|
||||
# TODO (erikj): Handle compressed feedback
|
||||
|
||||
end_token = yield self.get_room_events_max_id()
|
||||
|
||||
sql = (
|
||||
"SELECT * FROM events WHERE room_id = ? "
|
||||
"ORDER BY token_ordering, rowid DESC LIMIT ? "
|
||||
"SELECT * FROM events WHERE "
|
||||
"WHERE room_id = ? AND token_ordering <= ? "
|
||||
"ORDER BY topological_ordering, rowid DESC LIMIT ? "
|
||||
)
|
||||
|
||||
rows = yield self._execute_and_decode(
|
||||
sql,
|
||||
room_id, limit
|
||||
room_id, end_token, limit
|
||||
)
|
||||
|
||||
rows.reverse() # As we selected with reverse ordering
|
||||
|
||||
defer.returnValue([self._parse_event_from_row(r) for r in rows])
|
||||
if rows:
|
||||
topo = rows[0]["topological_ordering"]
|
||||
row_id = rows[0]["rowid"]
|
||||
start_token = "p%s-%s" % (topo, row_id)
|
||||
|
||||
token = (start_token, end_token)
|
||||
else:
|
||||
token = ("START", end_token)
|
||||
|
||||
defer.returnValue(
|
||||
(
|
||||
[self._parse_event_from_row(r) for r in rows],
|
||||
token
|
||||
)
|
||||
)
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def get_room_events_max_id(self):
|
||||
|
|
Loading…
Reference in a new issue