rename room_key_version table correctly, and fix opt args

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

View file

@ -143,7 +143,7 @@ class E2eRoomKeysHandler(object):
# lock everyone out until we've switched version
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
)
defer.returnValue(new_version)
@ -151,7 +151,7 @@ class E2eRoomKeysHandler(object):
@defer.inlineCallbacks
def get_version_info(self, user_id, version):
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
)
defer.returnValue(results)
@ -159,4 +159,4 @@ class E2eRoomKeysHandler(object):
@defer.inlineCallbacks
def delete_version(self, user_id, version):
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)

View file

@ -99,7 +99,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
@defer.inlineCallbacks
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 = {
@ -140,7 +140,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
@defer.inlineCallbacks
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 = {
@ -159,10 +159,10 @@ class EndToEndRoomKeyStore(SQLBaseStore):
)
@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(
table="e2e_room_key_versions",
table="e2e_room_keys_versions",
keyvalues={
"user_id": user_id,
"version": version,
@ -173,20 +173,20 @@ class EndToEndRoomKeyStore(SQLBaseStore):
"algorithm",
"auth_data",
),
desc="get_e2e_room_key_version_info",
desc="get_e2e_room_keys_version_info",
)
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
with the given version info.
"""
def _create_e2e_room_key_version_txn(txn):
def _create_e2e_room_keys_version_txn(txn):
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,)
)
current_version = txn.fetchone()[0]
@ -197,7 +197,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
self._simple_insert_txn(
txn,
table="e2e_room_key_versions",
table="e2e_room_keys_versions",
values={
"user_id": user_id,
"version": new_version,
@ -209,11 +209,11 @@ class EndToEndRoomKeyStore(SQLBaseStore):
return new_version
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
def delete_e2e_room_key_version(self, user_id, version):
def delete_e2e_room_keys_version(self, user_id, version):
keyvalues = {
"user_id": user_id,
@ -221,7 +221,7 @@ class EndToEndRoomKeyStore(SQLBaseStore):
}
yield self._simple_delete(
table="e2e_room_key_versions",
table="e2e_room_keys_versions",
keyvalues=keyvalues,
desc="delete_e2e_room_key_version",
desc="delete_e2e_room_keys_version",
)