mirror of
https://mau.dev/maunium/synapse.git
synced 2024-08-25 11:45:30 +00:00
Add WHERE clause support to index creation
This commit is contained in:
parent
502c901e11
commit
c94de0ab60
@ -219,7 +219,7 @@ class BackgroundUpdateStore(SQLBaseStore):
|
|||||||
self._background_update_handlers[update_name] = update_handler
|
self._background_update_handlers[update_name] = update_handler
|
||||||
|
|
||||||
def register_background_index_update(self, update_name, index_name,
|
def register_background_index_update(self, update_name, index_name,
|
||||||
table, columns):
|
table, columns, where_clause=None):
|
||||||
"""Helper for store classes to do a background index addition
|
"""Helper for store classes to do a background index addition
|
||||||
|
|
||||||
To use:
|
To use:
|
||||||
@ -243,13 +243,19 @@ class BackgroundUpdateStore(SQLBaseStore):
|
|||||||
conc = True
|
conc = True
|
||||||
else:
|
else:
|
||||||
conc = False
|
conc = False
|
||||||
|
# We don't use partial indices on SQLite as it wasn't introduced
|
||||||
|
# until 3.8, and wheezy has 3.7
|
||||||
|
where_clause = None
|
||||||
|
|
||||||
sql = "CREATE INDEX %(conc)s %(name)s ON %(table)s (%(columns)s)" \
|
sql = (
|
||||||
% {
|
"CREATE INDEX %(conc)s %(name)s ON %(table)s (%(columns)s)"
|
||||||
|
" %(where_clause)s"
|
||||||
|
) % {
|
||||||
"conc": "CONCURRENTLY" if conc else "",
|
"conc": "CONCURRENTLY" if conc else "",
|
||||||
"name": index_name,
|
"name": index_name,
|
||||||
"table": table,
|
"table": table,
|
||||||
"columns": ", ".join(columns),
|
"columns": ", ".join(columns),
|
||||||
|
"where_clause": "WHERE " + where_clause if where_clause else ""
|
||||||
}
|
}
|
||||||
|
|
||||||
def create_index_concurrently(conn):
|
def create_index_concurrently(conn):
|
||||||
|
@ -36,7 +36,8 @@ class EventPushActionsStore(SQLBaseStore):
|
|||||||
self.EPA_HIGHLIGHT_INDEX,
|
self.EPA_HIGHLIGHT_INDEX,
|
||||||
index_name="event_push_actions_u_highlight",
|
index_name="event_push_actions_u_highlight",
|
||||||
table="event_push_actions",
|
table="event_push_actions",
|
||||||
columns=["user_id", "stream_ordering", "highlight"],
|
columns=["user_id", "stream_ordering"],
|
||||||
|
where_clause="highlight = 1",
|
||||||
)
|
)
|
||||||
|
|
||||||
def _set_push_actions_for_event_and_users_txn(self, txn, event, tuples):
|
def _set_push_actions_for_event_and_users_txn(self, txn, event, tuples):
|
||||||
|
Loading…
Reference in New Issue
Block a user