Merge pull request #107 from matrix-org/add_desc_to_storage_execute

Add desc to storage execute
This commit is contained in:
Paul Evans 2015-03-11 17:55:31 +00:00
commit d33a3b91c3
9 changed files with 18 additions and 17 deletions

View File

@ -450,7 +450,7 @@ class DataStore(RoomMemberStore, RoomStore,
else: else:
args = (room_id, ) args = (room_id, )
results = yield self._execute_and_decode(sql, *args) results = yield self._execute_and_decode("get_current_state", sql, *args)
events = yield self._parse_events(results) events = yield self._parse_events(results)
defer.returnValue(events) defer.returnValue(events)
@ -475,7 +475,7 @@ class DataStore(RoomMemberStore, RoomStore,
sql += " OR s.type = 'm.room.aliases')" sql += " OR s.type = 'm.room.aliases')"
args = (room_id,) args = (room_id,)
results = yield self._execute_and_decode(sql, *args) results = yield self._execute_and_decode("get_current_state", sql, *args)
events = yield self._parse_events(results) events = yield self._parse_events(results)
@ -495,8 +495,7 @@ class DataStore(RoomMemberStore, RoomStore,
@defer.inlineCallbacks @defer.inlineCallbacks
def _get_min_token(self): def _get_min_token(self):
row = yield self._execute( row = yield self._execute(
None, "_get_min_token", None, "SELECT MIN(stream_ordering) FROM events"
"SELECT MIN(stream_ordering) FROM events"
) )
self.min_token = row[0][0] if row and row[0] and row[0][0] else -1 self.min_token = row[0][0] if row and row[0] and row[0][0] else -1

View File

@ -259,7 +259,7 @@ class SQLBaseStore(object):
) )
return results return results
def _execute(self, decoder, query, *args): def _execute(self, desc, decoder, query, *args):
"""Runs a single query for a result set. """Runs a single query for a result set.
Args: Args:
@ -277,10 +277,10 @@ class SQLBaseStore(object):
else: else:
return cursor.fetchall() return cursor.fetchall()
return self.runInteraction("_execute", interaction) return self.runInteraction(desc, interaction)
def _execute_and_decode(self, query, *args): def _execute_and_decode(self, desc, query, *args):
return self._execute(self.cursor_to_dict, query, *args) return self._execute(desc, self.cursor_to_dict, query, *args)
# "Simple" SQL API methods that operate on a single table with no JOINs, # "Simple" SQL API methods that operate on a single table with no JOINs,
# no complex WHERE clauses, just a dict of values for columns. # no complex WHERE clauses, just a dict of values for columns.

View File

@ -296,7 +296,7 @@ class ApplicationServiceStore(SQLBaseStore):
# } # }
# ] # ]
services = {} services = {}
results = yield self._execute_and_decode(sql) results = yield self._execute_and_decode("_populate_cache", sql)
for res in results: for res in results:
as_token = res["token"] as_token = res["token"]
if as_token not in services: if as_token not in services:

View File

@ -37,7 +37,7 @@ class FeedbackStore(SQLBaseStore):
"WHERE feedback.target_event_id = ? " "WHERE feedback.target_event_id = ? "
) )
rows = yield self._execute_and_decode(sql, event_id) rows = yield self._execute_and_decode("get_feedback_for_event", sql, event_id)
defer.returnValue( defer.returnValue(
[ [

View File

@ -85,7 +85,9 @@ class KeyStore(SQLBaseStore):
" AND key_id in (" + ",".join("?" for key_id in key_ids) + ")" " AND key_id in (" + ",".join("?" for key_id in key_ids) + ")"
) )
rows = yield self._execute_and_decode(sql, server_name, *key_ids) rows = yield self._execute_and_decode(
"get_server_verify_keys", sql, server_name, *key_ids
)
keys = [] keys = []
for row in rows: for row in rows:

View File

@ -34,7 +34,7 @@ class PushRuleStore(SQLBaseStore):
"WHERE user_name = ? " "WHERE user_name = ? "
"ORDER BY priority_class DESC, priority DESC" "ORDER BY priority_class DESC, priority DESC"
) )
rows = yield self._execute(None, sql, user_name) rows = yield self._execute("get_push_rules_for_user", None, sql, user_name)
dicts = [] dicts = []
for r in rows: for r in rows:

View File

@ -37,7 +37,8 @@ class PusherStore(SQLBaseStore):
) )
rows = yield self._execute( rows = yield self._execute(
None, sql, app_id_and_pushkey[0], app_id_and_pushkey[1] "get_pushers_by_app_id_and_pushkey", None, sql,
app_id_and_pushkey[0], app_id_and_pushkey[1]
) )
ret = [ ret = [
@ -70,7 +71,7 @@ class PusherStore(SQLBaseStore):
"FROM pushers" "FROM pushers"
) )
rows = yield self._execute(None, sql) rows = yield self._execute("get_all_pushers", None, sql)
ret = [ ret = [
{ {

View File

@ -88,8 +88,7 @@ class RegistrationStore(SQLBaseStore):
query = ("SELECT users.name, users.password_hash FROM users" query = ("SELECT users.name, users.password_hash FROM users"
" WHERE users.name = ?") " WHERE users.name = ?")
return self._execute( return self._execute(
self.cursor_to_dict, "get_user_by_id", self.cursor_to_dict, query, user_id
query, user_id
) )
def get_user_by_token(self, token): def get_user_by_token(self, token):

View File

@ -68,7 +68,7 @@ class RoomStore(SQLBaseStore):
""" """
query = RoomsTable.select_statement("room_id=?") query = RoomsTable.select_statement("room_id=?")
return self._execute( return self._execute(
RoomsTable.decode_single_result, query, room_id, "get_room", RoomsTable.decode_single_result, query, room_id,
) )
@defer.inlineCallbacks @defer.inlineCallbacks