Fix generality of query

This commit is contained in:
Olivier Wilkinson (reivilibre) 2019-08-13 15:07:00 +01:00
parent 1e0bd9a9c0
commit fd184f6cd0

View file

@ -42,9 +42,9 @@ def _run_create_generic(stats_type, cursor, database_engine):
cursor.execute( cursor.execute(
""" """
CREATE INDEX IF NOT EXISTS %s_stats_not_complete CREATE INDEX IF NOT EXISTS %s_stats_not_complete
ON %s_stats_current (completed_delta_stream_id, room_id); ON %s_stats_current (completed_delta_stream_id, %s_id);
""" """
% (stats_type, stats_type) % (stats_type, stats_type, stats_type)
) )
elif isinstance(database_engine, PostgresEngine): elif isinstance(database_engine, PostgresEngine):
# This partial index helps us with finding dirty stats rows # This partial index helps us with finding dirty stats rows
@ -60,10 +60,10 @@ def _run_create_generic(stats_type, cursor, database_engine):
cursor.execute( cursor.execute(
""" """
CREATE INDEX IF NOT EXISTS %s_stats_not_complete CREATE INDEX IF NOT EXISTS %s_stats_not_complete
ON %s_stats_current (room_id) ON %s_stats_current (%s_id)
WHERE completed_delta_stream_id IS NULL; WHERE completed_delta_stream_id IS NULL;
""" """
% (stats_type, stats_type) % (stats_type, stats_type, stats_type)
) )
else: else:
raise NotImplementedError("Unknown database engine.") raise NotImplementedError("Unknown database engine.")