Remove duplication

This commit is contained in:
Erik Johnston 2016-09-27 09:27:54 +01:00
parent 13122e5e24
commit 4974147aa3

View File

@ -386,15 +386,16 @@ class StateStore(SQLBaseStore):
# each (type, state_key). However, that was terribly slow # each (type, state_key). However, that was terribly slow
# without the right indicies (which we can't add until # without the right indicies (which we can't add until
# after we finish deduping state, which requires this func) # after we finish deduping state, which requires this func)
if types is not None: args = [next_group]
args = [next_group] + [i for typ in types for i in typ] if types:
args.extend(i for typ in types for i in typ)
txn.execute( txn.execute(
"SELECT type, state_key, event_id FROM state_groups_state" "SELECT type, state_key, event_id FROM state_groups_state"
" WHERE state_group = ? %s" % (where_clause,), " WHERE state_group = ? %s" % (where_clause,),
args args
) )
rows = txn.fetchall() rows = txn.fetchall()
results[group].update({ results[group].update({
(typ, state_key): event_id (typ, state_key): event_id
for typ, state_key, event_id in rows for typ, state_key, event_id in rows
@ -403,21 +404,8 @@ class StateStore(SQLBaseStore):
# If the lengths match then we must have all the types, # If the lengths match then we must have all the types,
# so no need to go walk further down the tree. # so no need to go walk further down the tree.
if len(results[group]) == len(types): if types is not None and len(results[group]) == len(types):
break break
else:
txn.execute(
"SELECT type, state_key, event_id FROM state_groups_state"
" WHERE state_group = ?",
(next_group,)
)
rows = txn.fetchall()
results[group].update({
(typ, state_key): event_id
for typ, state_key, event_id in rows
if (typ, state_key) not in results[group]
})
next_group = self._simple_select_one_onecol_txn( next_group = self._simple_select_one_onecol_txn(
txn, txn,