Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Merge pull request #6499 from matrix-org/erikj/fix_sqlite_7
Browse files Browse the repository at this point in the history
Fix support for SQLite 3.7.
  • Loading branch information
erikjohnston committed Dec 10, 2019
2 parents e3f528c + 5234699 commit e726e18
Show file tree
Hide file tree
Showing 6 changed files with 42 additions and 11 deletions.
1 change: 1 addition & 0 deletions changelog.d/6499.bugfix
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Fix support for SQLite 3.7.
23 changes: 14 additions & 9 deletions synapse/storage/data_stores/main/events.py
Original file line number Diff line number Diff line change
Expand Up @@ -1039,20 +1039,25 @@ def _store_redaction(self, txn, event):
},
)

@defer.inlineCallbacks
def _censor_redactions(self):
async def _censor_redactions(self):
"""Censors all redactions older than the configured period that haven't
been censored yet.
By censor we mean update the event_json table with the redacted event.
Returns:
Deferred
"""

if self.hs.config.redaction_retention_period is None:
return

if not (
await self.db.updates.has_completed_background_update(
"redactions_have_censored_ts_idx"
)
):
# We don't want to run this until the appropriate index has been
# created.
return

before_ts = self._clock.time_msec() - self.hs.config.redaction_retention_period

# We fetch all redactions that:
Expand All @@ -1074,15 +1079,15 @@ def _censor_redactions(self):
LIMIT ?
"""

rows = yield self.db.execute(
rows = await self.db.execute(
"_censor_redactions_fetch", None, sql, before_ts, 100
)

updates = []

for redaction_id, event_id in rows:
redaction_event = yield self.get_event(redaction_id, allow_none=True)
original_event = yield self.get_event(
redaction_event = await self.get_event(redaction_id, allow_none=True)
original_event = await self.get_event(
event_id, allow_rejected=True, allow_none=True
)

Expand Down Expand Up @@ -1115,7 +1120,7 @@ def _update_censor_txn(txn):
updatevalues={"have_censored": True},
)

yield self.db.runInteraction("_update_censor_txn", _update_censor_txn)
await self.db.runInteraction("_update_censor_txn", _update_censor_txn)

def _censor_event_txn(self, txn, event_id, pruned_json):
"""Censor an event by replacing its JSON in the event_json table with the
Expand Down
8 changes: 8 additions & 0 deletions synapse/storage/data_stores/main/events_bg_updates.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,6 +90,14 @@ def __init__(self, database: Database, db_conn, hs):
"event_store_labels", self._event_store_labels
)

self.db.updates.register_background_index_update(
"redactions_have_censored_ts_idx",
index_name="redactions_have_censored_ts",
table="redactions",
columns=["received_ts"],
where_clause="NOT have_censored",
)

@defer.inlineCallbacks
def _background_reindex_fields_sender(self, progress, batch_size):
target_min_stream_id = progress["target_min_stream_id_inclusive"]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,4 +14,3 @@
*/

ALTER TABLE redactions ADD COLUMN have_censored BOOL NOT NULL DEFAULT false;
CREATE INDEX redactions_have_censored ON redactions(event_id) WHERE not have_censored;
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,9 @@
*/

ALTER TABLE redactions ADD COLUMN received_ts BIGINT;
CREATE INDEX redactions_have_censored_ts ON redactions(received_ts) WHERE not have_censored;

INSERT INTO background_updates (update_name, progress_json) VALUES
('redactions_received_ts', '{}');

INSERT INTO background_updates (update_name, progress_json) VALUES
('redactions_have_censored_ts_idx', '{}');
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
/* Copyright 2019 The Matrix.org Foundation C.I.C.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

DROP INDEX IF EXISTS redactions_have_censored;

0 comments on commit e726e18

Please sign in to comment.