Matrix-Mirrors/synapse-product
1
0
Fork 0
mirror of https://git.anonymousland.org/anonymousland/synapse-product.git synced 2024-12-31 19:46:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

rename room_key_version table correctly, and fix opt args

Browse Source
This commit is contained in:
Matthew Hodgson 2017-12-18 01:58:53 +00:00 committed by Hubert Chathi
parent 0abb205b47
commit cac0253799
2 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
synapse/handlers/e2e_room_keys.py
View File

@ -143,7 +143,7 @@ class E2eRoomKeysHandler(object):
# lock everyone out until we've switched version # lock everyone out until we've switched version
with (yield self._upload_linearizer.queue(user_id)): with (yield self._upload_linearizer.queue(user_id)):
new_version = yield self.store.create_e2e_room_key_version( new_version = yield self.store.create_e2e_room_keys_version(
user_id, version_info user_id, version_info
) )
defer.returnValue(new_version) defer.returnValue(new_version)
@ -151,7 +151,7 @@ class E2eRoomKeysHandler(object):
@defer.inlineCallbacks @defer.inlineCallbacks
def get_version_info(self, user_id, version): def get_version_info(self, user_id, version):
with (yield self._upload_linearizer.queue(user_id)): with (yield self._upload_linearizer.queue(user_id)):
results = yield self.store.get_e2e_room_key_version_info( results = yield self.store.get_e2e_room_keys_version_info(
user_id, version user_id, version
) )
defer.returnValue(results) defer.returnValue(results)
@ -159,4 +159,4 @@ class E2eRoomKeysHandler(object):
@defer.inlineCallbacks @defer.inlineCallbacks
def delete_version(self, user_id, version): def delete_version(self, user_id, version):
with (yield self._upload_linearizer.queue(user_id)): with (yield self._upload_linearizer.queue(user_id)):
yield self.store.delete_e2e_room_key_version(user_id, version) yield self.store.delete_e2e_room_keys_version(user_id, version)

26
synapse/storage/e2e_room_keys.py
View File

@ -99,7 +99,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
@defer.inlineCallbacks @defer.inlineCallbacks
def get_e2e_room_keys( def get_e2e_room_keys(
self, user_id, version, room_id=room_id, session_id=session_id self, user_id, version, room_id=None, session_id=None
): ):
keyvalues = { keyvalues = {
@ -140,7 +140,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
@defer.inlineCallbacks @defer.inlineCallbacks
def delete_e2e_room_keys( def delete_e2e_room_keys(
self, user_id, version, room_id=room_id, session_id=session_id self, user_id, version, room_id=None, session_id=None
): ):
keyvalues = { keyvalues = {
@ -159,10 +159,10 @@ class EndToEndRoomKeyStore(SQLBaseStore):
) )
@defer.inlineCallbacks @defer.inlineCallbacks
def get_e2e_room_key_version_info(self, user_id, version): def get_e2e_room_keys_version_info(self, user_id, version):
row = yield self._simple_select_one( row = yield self._simple_select_one(
table="e2e_room_key_versions", table="e2e_room_keys_versions",
keyvalues={ keyvalues={
"user_id": user_id, "user_id": user_id,
"version": version, "version": version,
@ -173,20 +173,20 @@ class EndToEndRoomKeyStore(SQLBaseStore):
"algorithm", "algorithm",
"auth_data", "auth_data",
), ),
desc="get_e2e_room_key_version_info", desc="get_e2e_room_keys_version_info",
) )
defer.returnValue(row) defer.returnValue(row)
def create_e2e_room_key_version(self, user_id, info): def create_e2e_room_keys_version(self, user_id, info):
"""Atomically creates a new version of this user's e2e_room_keys store """Atomically creates a new version of this user's e2e_room_keys store
with the given version info. with the given version info.
""" """
def _create_e2e_room_key_version_txn(txn): def _create_e2e_room_keys_version_txn(txn):
txn.execute( txn.execute(
"SELECT MAX(version) FROM e2e_room_key_versions WHERE user_id=?", "SELECT MAX(version) FROM e2e_room_keys_versions WHERE user_id=?",
(user_id,) (user_id,)
) )
current_version = txn.fetchone()[0] current_version = txn.fetchone()[0]
@ -197,7 +197,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
self._simple_insert_txn( self._simple_insert_txn(
txn, txn,
table="e2e_room_key_versions", table="e2e_room_keys_versions",
values={ values={
"user_id": user_id, "user_id": user_id,
"version": new_version, "version": new_version,
@ -209,11 +209,11 @@ class EndToEndRoomKeyStore(SQLBaseStore):
return new_version return new_version
return self.runInteraction( return self.runInteraction(
"create_e2e_room_key_version_txn", _create_e2e_room_key_version_txn "create_e2e_room_keys_version_txn", _create_e2e_room_keys_version_txn
) )
@defer.inlineCallbacks @defer.inlineCallbacks
def delete_e2e_room_key_version(self, user_id, version): def delete_e2e_room_keys_version(self, user_id, version):
keyvalues = { keyvalues = {
"user_id": user_id, "user_id": user_id,
@ -221,7 +221,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
} }
yield self._simple_delete( yield self._simple_delete(
table="e2e_room_key_versions", table="e2e_room_keys_versions",
keyvalues=keyvalues, keyvalues=keyvalues,
desc="delete_e2e_room_key_version", desc="delete_e2e_room_keys_version",
) )