Fix the deduplication of incoming direct-to-device messages

This commit is contained in:
Mark Haines 2016-08-25 18:14:02 +01:00
parent e993925279
commit 641efb6a39
2 changed files with 16 additions and 1 deletions

View File

@ -19,6 +19,7 @@ from twisted.internet import defer
from synapse.http.servlet import parse_json_object_from_request
from synapse.http import servlet
from synapse.rest.client.v1.transactions import HttpTransactionStore
from ._base import client_v2_patterns
logger = logging.getLogger(__name__)
@ -40,9 +41,17 @@ class SendToDeviceRestServlet(servlet.RestServlet):
self.auth = hs.get_auth()
self.store = hs.get_datastore()
self.is_mine_id = hs.is_mine_id
self.txns = HttpTransactionStore()
@defer.inlineCallbacks
def on_PUT(self, request, message_type, txn_id):
try:
defer.returnValue(
self.txns.get_client_transaction(request, txn_id)
)
except KeyError:
pass
requester = yield self.auth.get_user_by_req(request)
content = parse_json_object_from_request(request)
@ -62,7 +71,9 @@ class SendToDeviceRestServlet(servlet.RestServlet):
yield self.store.add_messages_to_device_inbox(local_messages)
defer.returnValue((200, {}))
response = (200, {})
self.txns.store_client_transaction(request, txn_id, response)
defer.returnValue(response)
def register_servlets(hs, http_server):

View File

@ -132,5 +132,9 @@ class DeviceInboxStore(SQLBaseStore):
)
txn.execute(sql, (user_id, device_id, up_to_stream_id))
return self.runInteraction(
"delete_messages_for_device", delete_messages_for_device_txn
)
def get_to_device_stream_token(self):
return self._device_inbox_id_gen.get_current_token()