Only return new device messages in /sync

This commit is contained in:
Mark Haines 2016-09-02 15:50:37 +01:00
parent 2aa39db681
commit 65fa37ac5e
2 changed files with 6 additions and 4 deletions

View file

@ -576,7 +576,7 @@ class SyncHandler(object):
logger.debug("Getting messages up to %d", now_token.to_device_key) logger.debug("Getting messages up to %d", now_token.to_device_key)
messages, stream_id = yield self.store.get_new_messages_for_device( messages, stream_id = yield self.store.get_new_messages_for_device(
user_id, device_id, now_token.to_device_key user_id, device_id, since_stream_id, now_token.to_device_key
) )
logger.debug("Got messages up to %d: %r", stream_id, messages) logger.debug("Got messages up to %d: %r", stream_id, messages)
sync_result_builder.now_token = now_token.copy_and_replace( sync_result_builder.now_token = now_token.copy_and_replace(

View file

@ -85,7 +85,7 @@ class DeviceInboxStore(SQLBaseStore):
defer.returnValue(self._device_inbox_id_gen.get_current_token()) defer.returnValue(self._device_inbox_id_gen.get_current_token())
def get_new_messages_for_device( def get_new_messages_for_device(
self, user_id, device_id, current_stream_id, limit=100 self, user_id, device_id, last_stream_id, current_stream_id, limit=100
): ):
""" """
Args: Args:
@ -101,11 +101,13 @@ class DeviceInboxStore(SQLBaseStore):
sql = ( sql = (
"SELECT stream_id, message_json FROM device_inbox" "SELECT stream_id, message_json FROM device_inbox"
" WHERE user_id = ? AND device_id = ?" " WHERE user_id = ? AND device_id = ?"
" AND stream_id <= ?" " AND ? < stream_id AND stream_id <= ?"
" ORDER BY stream_id ASC" " ORDER BY stream_id ASC"
" LIMIT ?" " LIMIT ?"
) )
txn.execute(sql, (user_id, device_id, current_stream_id, limit)) txn.execute(sql, (
user_id, device_id, last_stream_id, current_stream_id, limit
))
messages = [] messages = []
for row in txn.fetchall(): for row in txn.fetchall():
stream_pos = row[0] stream_pos = row[0]