12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055 |
- #!/usr/bin/env python
- # -*- coding: utf-8 -*-
- # Copyright 2015, 2016 OpenMarket Ltd
- # Copyright 2018 New Vector Ltd
- # 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 curses
- import logging
- import sys
- import time
- import traceback
- from six import string_types
- import yaml
- from twisted.internet import defer, reactor
- import synapse
- from synapse.config.database import DatabaseConnectionConfig
- from synapse.config.homeserver import HomeServerConfig
- from synapse.logging.context import (
- LoggingContext,
- make_deferred_yieldable,
- run_in_background,
- )
- from synapse.storage.data_stores.main.client_ips import ClientIpBackgroundUpdateStore
- from synapse.storage.data_stores.main.deviceinbox import (
- DeviceInboxBackgroundUpdateStore,
- )
- from synapse.storage.data_stores.main.devices import DeviceBackgroundUpdateStore
- from synapse.storage.data_stores.main.events_bg_updates import (
- EventsBackgroundUpdatesStore,
- )
- from synapse.storage.data_stores.main.media_repository import (
- MediaRepositoryBackgroundUpdateStore,
- )
- from synapse.storage.data_stores.main.registration import (
- RegistrationBackgroundUpdateStore,
- )
- from synapse.storage.data_stores.main.room import RoomBackgroundUpdateStore
- from synapse.storage.data_stores.main.roommember import RoomMemberBackgroundUpdateStore
- from synapse.storage.data_stores.main.search import SearchBackgroundUpdateStore
- from synapse.storage.data_stores.main.state import MainStateBackgroundUpdateStore
- from synapse.storage.data_stores.main.stats import StatsStore
- from synapse.storage.data_stores.main.user_directory import (
- UserDirectoryBackgroundUpdateStore,
- )
- from synapse.storage.data_stores.state.bg_updates import StateBackgroundUpdateStore
- from synapse.storage.database import Database, make_conn
- from synapse.storage.engines import create_engine
- from synapse.storage.prepare_database import prepare_database
- from synapse.util import Clock
- from synapse.util.versionstring import get_version_string
- logger = logging.getLogger("synapse_port_db")
- BOOLEAN_COLUMNS = {
- "events": ["processed", "outlier", "contains_url"],
- "rooms": ["is_public"],
- "event_edges": ["is_state"],
- "presence_list": ["accepted"],
- "presence_stream": ["currently_active"],
- "public_room_list_stream": ["visibility"],
- "devices": ["hidden"],
- "device_lists_outbound_pokes": ["sent"],
- "users_who_share_rooms": ["share_private"],
- "groups": ["is_public"],
- "group_rooms": ["is_public"],
- "group_users": ["is_public", "is_admin"],
- "group_summary_rooms": ["is_public"],
- "group_room_categories": ["is_public"],
- "group_summary_users": ["is_public"],
- "group_roles": ["is_public"],
- "local_group_membership": ["is_publicised", "is_admin"],
- "e2e_room_keys": ["is_verified"],
- "account_validity": ["email_sent"],
- "redactions": ["have_censored"],
- "room_stats_state": ["is_federatable"],
- }
- APPEND_ONLY_TABLES = [
- "event_reference_hashes",
- "events",
- "event_json",
- "state_events",
- "room_memberships",
- "topics",
- "room_names",
- "rooms",
- "local_media_repository",
- "local_media_repository_thumbnails",
- "remote_media_cache",
- "remote_media_cache_thumbnails",
- "redactions",
- "event_edges",
- "event_auth",
- "received_transactions",
- "sent_transactions",
- "transaction_id_to_pdu",
- "users",
- "state_groups",
- "state_groups_state",
- "event_to_state_groups",
- "rejections",
- "event_search",
- "presence_stream",
- "push_rules_stream",
- "ex_outlier_stream",
- "cache_invalidation_stream",
- "public_room_list_stream",
- "state_group_edges",
- "stream_ordering_to_exterm",
- ]
- # Error returned by the run function. Used at the top-level part of the script to
- # handle errors and return codes.
- end_error = None
- # The exec_info for the error, if any. If error is defined but not exec_info the script
- # will show only the error message without the stacktrace, if exec_info is defined but
- # not the error then the script will show nothing outside of what's printed in the run
- # function. If both are defined, the script will print both the error and the stacktrace.
- end_error_exec_info = None
- class Store(
- ClientIpBackgroundUpdateStore,
- DeviceInboxBackgroundUpdateStore,
- DeviceBackgroundUpdateStore,
- EventsBackgroundUpdatesStore,
- MediaRepositoryBackgroundUpdateStore,
- RegistrationBackgroundUpdateStore,
- RoomBackgroundUpdateStore,
- RoomMemberBackgroundUpdateStore,
- SearchBackgroundUpdateStore,
- StateBackgroundUpdateStore,
- MainStateBackgroundUpdateStore,
- UserDirectoryBackgroundUpdateStore,
- StatsStore,
- ):
- def execute(self, f, *args, **kwargs):
- return self.db.runInteraction(f.__name__, f, *args, **kwargs)
- def execute_sql(self, sql, *args):
- def r(txn):
- txn.execute(sql, args)
- return txn.fetchall()
- return self.db.runInteraction("execute_sql", r)
- def insert_many_txn(self, txn, table, headers, rows):
- sql = "INSERT INTO %s (%s) VALUES (%s)" % (
- table,
- ", ".join(k for k in headers),
- ", ".join("%s" for _ in headers),
- )
- try:
- txn.executemany(sql, rows)
- except Exception:
- logger.exception("Failed to insert: %s", table)
- raise
- def set_room_is_public(self, room_id, is_public):
- raise Exception(
- "Attempt to set room_is_public during port_db: database not empty?"
- )
- class MockHomeserver:
- def __init__(self, config):
- self.clock = Clock(reactor)
- self.config = config
- self.hostname = config.server_name
- self.version_string = "Synapse/"+get_version_string(synapse)
- def get_clock(self):
- return self.clock
- def get_reactor(self):
- return reactor
- class Porter(object):
- def __init__(self, **kwargs):
- self.__dict__.update(kwargs)
- async def setup_table(self, table):
- if table in APPEND_ONLY_TABLES:
- # It's safe to just carry on inserting.
- row = await self.postgres_store.db.simple_select_one(
- table="port_from_sqlite3",
- keyvalues={"table_name": table},
- retcols=("forward_rowid", "backward_rowid"),
- allow_none=True,
- )
- total_to_port = None
- if row is None:
- if table == "sent_transactions":
- (
- forward_chunk,
- already_ported,
- total_to_port,
- ) = await self._setup_sent_transactions()
- backward_chunk = 0
- else:
- await self.postgres_store.db.simple_insert(
- table="port_from_sqlite3",
- values={
- "table_name": table,
- "forward_rowid": 1,
- "backward_rowid": 0,
- },
- )
- forward_chunk = 1
- backward_chunk = 0
- already_ported = 0
- else:
- forward_chunk = row["forward_rowid"]
- backward_chunk = row["backward_rowid"]
- if total_to_port is None:
- already_ported, total_to_port = await self._get_total_count_to_port(
- table, forward_chunk, backward_chunk
- )
- else:
- def delete_all(txn):
- txn.execute(
- "DELETE FROM port_from_sqlite3 WHERE table_name = %s", (table,)
- )
- txn.execute("TRUNCATE %s CASCADE" % (table,))
- await self.postgres_store.execute(delete_all)
- await self.postgres_store.db.simple_insert(
- table="port_from_sqlite3",
- values={"table_name": table, "forward_rowid": 1, "backward_rowid": 0},
- )
- forward_chunk = 1
- backward_chunk = 0
- already_ported, total_to_port = await self._get_total_count_to_port(
- table, forward_chunk, backward_chunk
- )
- return table, already_ported, total_to_port, forward_chunk, backward_chunk
- async def handle_table(
- self, table, postgres_size, table_size, forward_chunk, backward_chunk
- ):
- logger.info(
- "Table %s: %i/%i (rows %i-%i) already ported",
- table,
- postgres_size,
- table_size,
- backward_chunk + 1,
- forward_chunk - 1,
- )
- if not table_size:
- return
- self.progress.add_table(table, postgres_size, table_size)
- if table == "event_search":
- await self.handle_search_table(
- postgres_size, table_size, forward_chunk, backward_chunk
- )
- return
- if table in (
- "user_directory",
- "user_directory_search",
- "users_who_share_rooms",
- "users_in_pubic_room",
- ):
- # We don't port these tables, as they're a faff and we can regenreate
- # them anyway.
- self.progress.update(table, table_size) # Mark table as done
- return
- if table == "user_directory_stream_pos":
- # We need to make sure there is a single row, `(X, null), as that is
- # what synapse expects to be there.
- await self.postgres_store.db.simple_insert(
- table=table, values={"stream_id": None}
- )
- self.progress.update(table, table_size) # Mark table as done
- return
- forward_select = (
- "SELECT rowid, * FROM %s WHERE rowid >= ? ORDER BY rowid LIMIT ?" % (table,)
- )
- backward_select = (
- "SELECT rowid, * FROM %s WHERE rowid <= ? ORDER BY rowid LIMIT ?" % (table,)
- )
- do_forward = [True]
- do_backward = [True]
- while True:
- def r(txn):
- forward_rows = []
- backward_rows = []
- if do_forward[0]:
- txn.execute(forward_select, (forward_chunk, self.batch_size))
- forward_rows = txn.fetchall()
- if not forward_rows:
- do_forward[0] = False
- if do_backward[0]:
- txn.execute(backward_select, (backward_chunk, self.batch_size))
- backward_rows = txn.fetchall()
- if not backward_rows:
- do_backward[0] = False
- if forward_rows or backward_rows:
- headers = [column[0] for column in txn.description]
- else:
- headers = None
- return headers, forward_rows, backward_rows
- headers, frows, brows = await self.sqlite_store.db.runInteraction(
- "select", r
- )
- if frows or brows:
- if frows:
- forward_chunk = max(row[0] for row in frows) + 1
- if brows:
- backward_chunk = min(row[0] for row in brows) - 1
- rows = frows + brows
- rows = self._convert_rows(table, headers, rows)
- def insert(txn):
- self.postgres_store.insert_many_txn(txn, table, headers[1:], rows)
- self.postgres_store.db.simple_update_one_txn(
- txn,
- table="port_from_sqlite3",
- keyvalues={"table_name": table},
- updatevalues={
- "forward_rowid": forward_chunk,
- "backward_rowid": backward_chunk,
- },
- )
- await self.postgres_store.execute(insert)
- postgres_size += len(rows)
- self.progress.update(table, postgres_size)
- else:
- return
- async def handle_search_table(
- self, postgres_size, table_size, forward_chunk, backward_chunk
- ):
- select = (
- "SELECT es.rowid, es.*, e.origin_server_ts, e.stream_ordering"
- " FROM event_search as es"
- " INNER JOIN events AS e USING (event_id, room_id)"
- " WHERE es.rowid >= ?"
- " ORDER BY es.rowid LIMIT ?"
- )
- while True:
- def r(txn):
- txn.execute(select, (forward_chunk, self.batch_size))
- rows = txn.fetchall()
- headers = [column[0] for column in txn.description]
- return headers, rows
- headers, rows = await self.sqlite_store.db.runInteraction("select", r)
- if rows:
- forward_chunk = rows[-1][0] + 1
- # We have to treat event_search differently since it has a
- # different structure in the two different databases.
- def insert(txn):
- sql = (
- "INSERT INTO event_search (event_id, room_id, key,"
- " sender, vector, origin_server_ts, stream_ordering)"
- " VALUES (?,?,?,?,to_tsvector('english', ?),?,?)"
- )
- rows_dict = []
- for row in rows:
- d = dict(zip(headers, row))
- if "\0" in d["value"]:
- logger.warning("dropping search row %s", d)
- else:
- rows_dict.append(d)
- txn.executemany(
- sql,
- [
- (
- row["event_id"],
- row["room_id"],
- row["key"],
- row["sender"],
- row["value"],
- row["origin_server_ts"],
- row["stream_ordering"],
- )
- for row in rows_dict
- ],
- )
- self.postgres_store.db.simple_update_one_txn(
- txn,
- table="port_from_sqlite3",
- keyvalues={"table_name": "event_search"},
- updatevalues={
- "forward_rowid": forward_chunk,
- "backward_rowid": backward_chunk,
- },
- )
- await self.postgres_store.execute(insert)
- postgres_size += len(rows)
- self.progress.update("event_search", postgres_size)
- else:
- return
- def build_db_store(
- self, db_config: DatabaseConnectionConfig, allow_outdated_version: bool = False,
- ):
- """Builds and returns a database store using the provided configuration.
- Args:
- db_config: The database configuration
- allow_outdated_version: True to suppress errors about the database server
- version being too old to run a complete synapse
- Returns:
- The built Store object.
- """
- self.progress.set_state("Preparing %s" % db_config.config["name"])
- engine = create_engine(db_config.config)
- hs = MockHomeserver(self.hs_config)
- with make_conn(db_config, engine) as db_conn:
- engine.check_database(
- db_conn, allow_outdated_version=allow_outdated_version
- )
- prepare_database(db_conn, engine, config=self.hs_config)
- store = Store(Database(hs, db_config, engine), db_conn, hs)
- db_conn.commit()
- return store
- async def run_background_updates_on_postgres(self):
- # Manually apply all background updates on the PostgreSQL database.
- postgres_ready = (
- await self.postgres_store.db.updates.has_completed_background_updates()
- )
- if not postgres_ready:
- # Only say that we're running background updates when there are background
- # updates to run.
- self.progress.set_state("Running background updates on PostgreSQL")
- while not postgres_ready:
- await self.postgres_store.db.updates.do_next_background_update(100)
- postgres_ready = await (
- self.postgres_store.db.updates.has_completed_background_updates()
- )
- async def run(self):
- """Ports the SQLite database to a PostgreSQL database.
- When a fatal error is met, its message is assigned to the global "end_error"
- variable. When this error comes with a stacktrace, its exec_info is assigned to
- the global "end_error_exec_info" variable.
- """
- global end_error
- try:
- # we allow people to port away from outdated versions of sqlite.
- self.sqlite_store = self.build_db_store(
- DatabaseConnectionConfig("master-sqlite", self.sqlite_config),
- allow_outdated_version=True,
- )
- # Check if all background updates are done, abort if not.
- updates_complete = (
- await self.sqlite_store.db.updates.has_completed_background_updates()
- )
- if not updates_complete:
- end_error = (
- "Pending background updates exist in the SQLite3 database."
- " Please start Synapse again and wait until every update has finished"
- " before running this script.\n"
- )
- return
- self.postgres_store = self.build_db_store(
- self.hs_config.get_single_database()
- )
- await self.run_background_updates_on_postgres()
- self.progress.set_state("Creating port tables")
- def create_port_table(txn):
- txn.execute(
- "CREATE TABLE IF NOT EXISTS port_from_sqlite3 ("
- " table_name varchar(100) NOT NULL UNIQUE,"
- " forward_rowid bigint NOT NULL,"
- " backward_rowid bigint NOT NULL"
- ")"
- )
- # The old port script created a table with just a "rowid" column.
- # We want people to be able to rerun this script from an old port
- # so that they can pick up any missing events that were not
- # ported across.
- def alter_table(txn):
- txn.execute(
- "ALTER TABLE IF EXISTS port_from_sqlite3"
- " RENAME rowid TO forward_rowid"
- )
- txn.execute(
- "ALTER TABLE IF EXISTS port_from_sqlite3"
- " ADD backward_rowid bigint NOT NULL DEFAULT 0"
- )
- try:
- await self.postgres_store.db.runInteraction("alter_table", alter_table)
- except Exception:
- # On Error Resume Next
- pass
- await self.postgres_store.db.runInteraction(
- "create_port_table", create_port_table
- )
- # Step 2. Get tables.
- self.progress.set_state("Fetching tables")
- sqlite_tables = await self.sqlite_store.db.simple_select_onecol(
- table="sqlite_master", keyvalues={"type": "table"}, retcol="name"
- )
- postgres_tables = await self.postgres_store.db.simple_select_onecol(
- table="information_schema.tables",
- keyvalues={},
- retcol="distinct table_name",
- )
- tables = set(sqlite_tables) & set(postgres_tables)
- logger.info("Found %d tables", len(tables))
- # Step 3. Figure out what still needs copying
- self.progress.set_state("Checking on port progress")
- setup_res = await make_deferred_yieldable(
- defer.gatherResults(
- [
- run_in_background(self.setup_table, table)
- for table in tables
- if table not in ["schema_version", "applied_schema_deltas"]
- and not table.startswith("sqlite_")
- ],
- consumeErrors=True,
- )
- )
- # Step 4. Do the copying.
- self.progress.set_state("Copying to postgres")
- await make_deferred_yieldable(
- defer.gatherResults(
- [run_in_background(self.handle_table, *res) for res in setup_res],
- consumeErrors=True,
- )
- )
- # Step 5. Do final post-processing
- await self._setup_state_group_id_seq()
- self.progress.done()
- except Exception as e:
- global end_error_exec_info
- end_error = e
- end_error_exec_info = sys.exc_info()
- logger.exception("")
- finally:
- reactor.stop()
- def _convert_rows(self, table, headers, rows):
- bool_col_names = BOOLEAN_COLUMNS.get(table, [])
- bool_cols = [i for i, h in enumerate(headers) if h in bool_col_names]
- class BadValueException(Exception):
- pass
- def conv(j, col):
- if j in bool_cols:
- return bool(col)
- if isinstance(col, bytes):
- return bytearray(col)
- elif isinstance(col, string_types) and "\0" in col:
- logger.warning(
- "DROPPING ROW: NUL value in table %s col %s: %r",
- table,
- headers[j],
- col,
- )
- raise BadValueException()
- return col
- outrows = []
- for i, row in enumerate(rows):
- try:
- outrows.append(
- tuple(conv(j, col) for j, col in enumerate(row) if j > 0)
- )
- except BadValueException:
- pass
- return outrows
- async def _setup_sent_transactions(self):
- # Only save things from the last day
- yesterday = int(time.time() * 1000) - 86400000
- # And save the max transaction id from each destination
- select = (
- "SELECT rowid, * FROM sent_transactions WHERE rowid IN ("
- "SELECT max(rowid) FROM sent_transactions"
- " GROUP BY destination"
- ")"
- )
- def r(txn):
- txn.execute(select)
- rows = txn.fetchall()
- headers = [column[0] for column in txn.description]
- ts_ind = headers.index("ts")
- return headers, [r for r in rows if r[ts_ind] < yesterday]
- headers, rows = await self.sqlite_store.db.runInteraction("select", r)
- rows = self._convert_rows("sent_transactions", headers, rows)
- inserted_rows = len(rows)
- if inserted_rows:
- max_inserted_rowid = max(r[0] for r in rows)
- def insert(txn):
- self.postgres_store.insert_many_txn(
- txn, "sent_transactions", headers[1:], rows
- )
- await self.postgres_store.execute(insert)
- else:
- max_inserted_rowid = 0
- def get_start_id(txn):
- txn.execute(
- "SELECT rowid FROM sent_transactions WHERE ts >= ?"
- " ORDER BY rowid ASC LIMIT 1",
- (yesterday,),
- )
- rows = txn.fetchall()
- if rows:
- return rows[0][0]
- else:
- return 1
- next_chunk = await self.sqlite_store.execute(get_start_id)
- next_chunk = max(max_inserted_rowid + 1, next_chunk)
- await self.postgres_store.db.simple_insert(
- table="port_from_sqlite3",
- values={
- "table_name": "sent_transactions",
- "forward_rowid": next_chunk,
- "backward_rowid": 0,
- },
- )
- def get_sent_table_size(txn):
- txn.execute(
- "SELECT count(*) FROM sent_transactions" " WHERE ts >= ?", (yesterday,)
- )
- (size,) = txn.fetchone()
- return int(size)
- remaining_count = await self.sqlite_store.execute(get_sent_table_size)
- total_count = remaining_count + inserted_rows
- return next_chunk, inserted_rows, total_count
- async def _get_remaining_count_to_port(self, table, forward_chunk, backward_chunk):
- frows = await self.sqlite_store.execute_sql(
- "SELECT count(*) FROM %s WHERE rowid >= ?" % (table,), forward_chunk
- )
- brows = await self.sqlite_store.execute_sql(
- "SELECT count(*) FROM %s WHERE rowid <= ?" % (table,), backward_chunk
- )
- return frows[0][0] + brows[0][0]
- async def _get_already_ported_count(self, table):
- rows = await self.postgres_store.execute_sql(
- "SELECT count(*) FROM %s" % (table,)
- )
- return rows[0][0]
- async def _get_total_count_to_port(self, table, forward_chunk, backward_chunk):
- remaining, done = await make_deferred_yieldable(
- defer.gatherResults(
- [
- run_in_background(
- self._get_remaining_count_to_port,
- table,
- forward_chunk,
- backward_chunk,
- ),
- run_in_background(self._get_already_ported_count, table),
- ],
- )
- )
- remaining = int(remaining) if remaining else 0
- done = int(done) if done else 0
- return done, remaining + done
- def _setup_state_group_id_seq(self):
- def r(txn):
- txn.execute("SELECT MAX(id) FROM state_groups")
- curr_id = txn.fetchone()[0]
- if not curr_id:
- return
- next_id = curr_id + 1
- txn.execute("ALTER SEQUENCE state_group_id_seq RESTART WITH %s", (next_id,))
- return self.postgres_store.db.runInteraction("setup_state_group_id_seq", r)
- ##############################################
- # The following is simply UI stuff
- ##############################################
- class Progress(object):
- """Used to report progress of the port
- """
- def __init__(self):
- self.tables = {}
- self.start_time = int(time.time())
- def add_table(self, table, cur, size):
- self.tables[table] = {
- "start": cur,
- "num_done": cur,
- "total": size,
- "perc": int(cur * 100 / size),
- }
- def update(self, table, num_done):
- data = self.tables[table]
- data["num_done"] = num_done
- data["perc"] = int(num_done * 100 / data["total"])
- def done(self):
- pass
- class CursesProgress(Progress):
- """Reports progress to a curses window
- """
- def __init__(self, stdscr):
- self.stdscr = stdscr
- curses.use_default_colors()
- curses.curs_set(0)
- curses.init_pair(1, curses.COLOR_RED, -1)
- curses.init_pair(2, curses.COLOR_GREEN, -1)
- self.last_update = 0
- self.finished = False
- self.total_processed = 0
- self.total_remaining = 0
- super(CursesProgress, self).__init__()
- def update(self, table, num_done):
- super(CursesProgress, self).update(table, num_done)
- self.total_processed = 0
- self.total_remaining = 0
- for table, data in self.tables.items():
- self.total_processed += data["num_done"] - data["start"]
- self.total_remaining += data["total"] - data["num_done"]
- self.render()
- def render(self, force=False):
- now = time.time()
- if not force and now - self.last_update < 0.2:
- # reactor.callLater(1, self.render)
- return
- self.stdscr.clear()
- rows, cols = self.stdscr.getmaxyx()
- duration = int(now) - int(self.start_time)
- minutes, seconds = divmod(duration, 60)
- duration_str = "%02dm %02ds" % (minutes, seconds)
- if self.finished:
- status = "Time spent: %s (Done!)" % (duration_str,)
- else:
- if self.total_processed > 0:
- left = float(self.total_remaining) / self.total_processed
- est_remaining = (int(now) - self.start_time) * left
- est_remaining_str = "%02dm %02ds remaining" % divmod(est_remaining, 60)
- else:
- est_remaining_str = "Unknown"
- status = "Time spent: %s (est. remaining: %s)" % (
- duration_str,
- est_remaining_str,
- )
- self.stdscr.addstr(0, 0, status, curses.A_BOLD)
- max_len = max([len(t) for t in self.tables.keys()])
- left_margin = 5
- middle_space = 1
- items = self.tables.items()
- items = sorted(items, key=lambda i: (i[1]["perc"], i[0]))
- for i, (table, data) in enumerate(items):
- if i + 2 >= rows:
- break
- perc = data["perc"]
- color = curses.color_pair(2) if perc == 100 else curses.color_pair(1)
- self.stdscr.addstr(
- i + 2, left_margin + max_len - len(table), table, curses.A_BOLD | color
- )
- size = 20
- progress = "[%s%s]" % (
- "#" * int(perc * size / 100),
- " " * (size - int(perc * size / 100)),
- )
- self.stdscr.addstr(
- i + 2,
- left_margin + max_len + middle_space,
- "%s %3d%% (%d/%d)" % (progress, perc, data["num_done"], data["total"]),
- )
- if self.finished:
- self.stdscr.addstr(rows - 1, 0, "Press any key to exit...")
- self.stdscr.refresh()
- self.last_update = time.time()
- def done(self):
- self.finished = True
- self.render(True)
- self.stdscr.getch()
- def set_state(self, state):
- self.stdscr.clear()
- self.stdscr.addstr(0, 0, state + "...", curses.A_BOLD)
- self.stdscr.refresh()
- class TerminalProgress(Progress):
- """Just prints progress to the terminal
- """
- def update(self, table, num_done):
- super(TerminalProgress, self).update(table, num_done)
- data = self.tables[table]
- print(
- "%s: %d%% (%d/%d)" % (table, data["perc"], data["num_done"], data["total"])
- )
- def set_state(self, state):
- print(state + "...")
- ##############################################
- ##############################################
- if __name__ == "__main__":
- parser = argparse.ArgumentParser(
- description="A script to port an existing synapse SQLite database to"
- " a new PostgreSQL database."
- )
- parser.add_argument("-v", action="store_true")
- parser.add_argument(
- "--sqlite-database",
- required=True,
- help="The snapshot of the SQLite database file. This must not be"
- " currently used by a running synapse server",
- )
- parser.add_argument(
- "--postgres-config",
- type=argparse.FileType("r"),
- required=True,
- help="The database config file for the PostgreSQL database",
- )
- parser.add_argument(
- "--curses", action="store_true", help="display a curses based progress UI"
- )
- parser.add_argument(
- "--batch-size",
- type=int,
- default=1000,
- help="The number of rows to select from the SQLite table each"
- " iteration [default=1000]",
- )
- 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",
- }
- if args.curses:
- logging_config["filename"] = "port-synapse.log"
- logging.basicConfig(**logging_config)
- sqlite_config = {
- "name": "sqlite3",
- "args": {
- "database": args.sqlite_database,
- "cp_min": 1,
- "cp_max": 1,
- "check_same_thread": False,
- },
- }
- hs_config = yaml.safe_load(args.postgres_config)
- if "database" not in hs_config:
- sys.stderr.write("The configuration file must have a 'database' section.\n")
- sys.exit(4)
- postgres_config = hs_config["database"]
- if "name" not in postgres_config:
- sys.stderr.write("Malformed database config: no 'name'\n")
- sys.exit(2)
- if postgres_config["name"] != "psycopg2":
- sys.stderr.write("Database must use the 'psycopg2' connector.\n")
- sys.exit(3)
- config = HomeServerConfig()
- config.parse_config_dict(hs_config, "", "")
- def start(stdscr=None):
- if stdscr:
- progress = CursesProgress(stdscr)
- else:
- progress = TerminalProgress()
- porter = Porter(
- sqlite_config=sqlite_config,
- progress=progress,
- batch_size=args.batch_size,
- hs_config=config,
- )
- @defer.inlineCallbacks
- def run():
- with LoggingContext("synapse_port_db_run"):
- yield defer.ensureDeferred(porter.run())
- reactor.callWhenRunning(run)
- reactor.run()
- if args.curses:
- curses.wrapper(start)
- else:
- start()
- if end_error:
- if end_error_exec_info:
- exc_type, exc_value, exc_traceback = end_error_exec_info
- traceback.print_exception(exc_type, exc_value, exc_traceback)
- sys.stderr.write(end_error)
- sys.exit(5)
|