diff options
author | Nick Barrett <nick@fizzadar.com> | 2021-10-06 11:26:18 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-10-06 11:26:18 +0100 |
commit | c80878d22a013ed68d3929025bbd40074e66af01 (patch) | |
tree | 30fd07e4fa8d7821a5fa22e7ae41a18d49c0124a /scripts-dev/update_database | |
parent | More types for synapse.util, part 1 (#10888) (diff) | |
download | synapse-c80878d22a013ed68d3929025bbd40074e66af01.tar.xz |
Add `--run-background-updates` option to `update_database` script. (#10954)
Signed-off-by: Nick Barrett <nick@beeper.com>
Diffstat (limited to 'scripts-dev/update_database')
-rwxr-xr-x | scripts-dev/update_database | 100 |
1 files changed, 0 insertions, 100 deletions
diff --git a/scripts-dev/update_database b/scripts-dev/update_database deleted file mode 100755 index 87f709b6ed..0000000000 --- a/scripts-dev/update_database +++ /dev/null @@ -1,100 +0,0 @@ -#!/usr/bin/env python -# 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. - -import argparse -import logging -import sys - -import yaml - -from twisted.internet import defer, reactor - -import synapse -from synapse.config.homeserver import HomeServerConfig -from synapse.metrics.background_process_metrics import run_as_background_process -from synapse.server import HomeServer -from synapse.storage import DataStore -from synapse.util.versionstring import get_version_string - -logger = logging.getLogger("update_database") - - -class MockHomeserver(HomeServer): - DATASTORE_CLASS = DataStore - - def __init__(self, config, **kwargs): - super(MockHomeserver, self).__init__( - config.server_name, reactor=reactor, config=config, **kwargs - ) - - self.version_string = "Synapse/" + get_version_string(synapse) - - -if __name__ == "__main__": - parser = argparse.ArgumentParser( - description=( - "Updates a synapse database to the latest schema and runs background updates" - " on it." - ) - ) - parser.add_argument("-v", action="store_true") - parser.add_argument( - "--database-config", - type=argparse.FileType("r"), - required=True, - help="A database config file for either a SQLite3 database or a PostgreSQL one.", - ) - - args = parser.parse_args() - - logging_config = { - "level": logging.DEBUG if args.v else logging.INFO, - "format": "%(asctime)s - %(name)s - %(lineno)d - %(levelname)s - %(message)s", - } - - logging.basicConfig(**logging_config) - - # Load, process and sanity-check the config. - hs_config = yaml.safe_load(args.database_config) - - if "database" not in hs_config: - sys.stderr.write("The configuration file must have a 'database' section.\n") - sys.exit(4) - - config = HomeServerConfig() - config.parse_config_dict(hs_config, "", "") - - # Instantiate and initialise the homeserver object. - hs = MockHomeserver(config) - - # Setup instantiates the store within the homeserver object and updates the - # DB. - hs.setup() - store = hs.get_datastore() - - async def run_background_updates(): - await store.db_pool.updates.run_background_updates(sleep=False) - # Stop the reactor to exit the script once every background update is run. - reactor.stop() - - def run(): - # Apply all background updates on the database. - defer.ensureDeferred( - run_as_background_process("background_updates", run_background_updates) - ) - - reactor.callWhenRunning(run) - - reactor.run() |