From ad281f591e2d3dc0a6fed99803008b85f4492d23 Mon Sep 17 00:00:00 2001 From: Corentin Flr <10946971+cogk@users.noreply.github.com> Date: Tue, 2 Jul 2024 10:00:56 +0200 Subject: [PATCH] fix: Use mariadb commands instead of mysql --- agent/bench.py | 4 ++-- agent/database.py | 2 +- agent/proxysql.py | 2 +- agent/site.py | 24 ++++++++++++------------ 4 files changed, 16 insertions(+), 16 deletions(-) diff --git a/agent/bench.py b/agent/bench.py index cd6cd8bc..e7ccc58c 100644 --- a/agent/bench.py +++ b/agent/bench.py @@ -267,7 +267,7 @@ def create_mariadb_user(self, site, mariadb_root_password, database=None): ] for query in queries: command = ( - f"mysql -h {self.host} -uroot -p{mariadb_root_password}" + f"mariadb -h {self.host} -uroot -p{mariadb_root_password}" f' -e "{query}"' ) self.execute(command) @@ -283,7 +283,7 @@ def drop_mariadb_user(self, site, mariadb_root_password, database=None): ] for query in queries: command = ( - f"mysql -h {self.host} -uroot -p{mariadb_root_password}" + f"mariadb -h {self.host} -uroot -p{mariadb_root_password}" f' -e "{query}"' ) self.execute(command) diff --git a/agent/database.py b/agent/database.py index eb2aef6a..7814e592 100644 --- a/agent/database.py +++ b/agent/database.py @@ -32,7 +32,7 @@ def search_binary_log( log = os.path.join(self.mariadb_directory, log) LINES_TO_SKIP = r"^(USE|COMMIT|START TRANSACTION|DELIMITER|ROLLBACK|#)" command = ( - f"mysqlbinlog --short-form --database {database} " + f"mariadb-binlog --short-form --database {database} " f"--start-datetime '{start_datetime}' " f"--stop-datetime '{stop_datetime}' " f" {log} | grep -Piv '{LINES_TO_SKIP}'" diff --git a/agent/proxysql.py b/agent/proxysql.py index 50dabff2..6b06721d 100644 --- a/agent/proxysql.py +++ b/agent/proxysql.py @@ -18,7 +18,7 @@ def __init__(self, directory=None): def proxysql_execute(self, command): command = ( - "mysql -h 127.0.0.1 -P 6032 " + "mariadb -h 127.0.0.1 -P 6032 " f"-u frappe -p{self.proxysql_admin_password} " f"--disable-column-names -e '{command}'" ) diff --git a/agent/site.py b/agent/site.py index 946e1172..4603be24 100644 --- a/agent/site.py +++ b/agent/site.py @@ -272,7 +272,7 @@ def create_database_access_credentials(self, mode, mariadb_root_password): ] for query in queries: command = ( - f"mysql -h {self.host} -uroot -p{mariadb_root_password}" + f"mariadb -h {self.host} -uroot -p{mariadb_root_password}" f' -e "{query}"' ) self.execute(command) @@ -288,7 +288,7 @@ def revoke_database_access_credentials(self, user, mariadb_root_password): ] for query in queries: command = ( - f"mysql -h {self.host} -uroot -p{mariadb_root_password}" + f"mariadb -h {self.host} -uroot -p{mariadb_root_password}" f' -e "{query}"' ) self.execute(command) @@ -318,7 +318,7 @@ def restore_site_tables(self): output = self.execute( "set -o pipefail && " f"gunzip -c '{backup_file_path}' | " - f"mysql -h {self.host} -u {self.user} -p{self.password} " + f"mariadb -h {self.host} -u {self.user} -p{self.password} " f"{self.database}", executable="/bin/bash", ) @@ -474,7 +474,7 @@ def tablewise_backup(self): ) output = self.execute( "set -o pipefail && " - "mysqldump --single-transaction --quick --lock-tables=false " + "mariadb-dump --single-transaction --quick --lock-tables=false " f"-h {self.host} -u {self.user} -p{self.password} " f"{self.database} '{table}' " f" | gzip > '{backup_file}'", @@ -544,7 +544,7 @@ def restore_touched_tables(self): output = self.execute( "set -o pipefail && " f"gunzip -c '{backup_file}' | " - f"mysql -h {self.host} -u {self.user} -p{self.password} " + f"mariadb -h {self.host} -u {self.user} -p{self.password} " f"{self.database}", executable="/bin/bash", ) @@ -559,7 +559,7 @@ def drop_new_tables(self): data = {"dropped": {}} for table in new_tables: output = self.execute( - f"mysql -h {self.host} -u {self.user} -p{self.password} " + f"mariadb -h {self.host} -u {self.user} -p{self.password} " f"{self.database} -e 'DROP TABLE `{table}`'" ) data["dropped"][table] = output @@ -641,7 +641,7 @@ def timezone(self): ) try: timezone = self.execute( - f"mysql -h {self.host} -u{self.database} -p{self.password} " + f"mariadb -h {self.host} -u{self.database} -p{self.password} " f'--connect-timeout 3 -sN -e "{query}"' )["output"].strip() except Exception: @@ -651,7 +651,7 @@ def timezone(self): @property def tables(self): return self.execute( - "mysql --disable-column-names -B -e 'SHOW TABLES' " + "mariadb --disable-column-names -B -e 'SHOW TABLES' " f"-h {self.host} -u {self.user} -p{self.password} {self.database}" )["output"].split("\n") @@ -685,7 +685,7 @@ def optimize_tables(self): for table in tables: query = f"OPTIMIZE TABLE `{table}`" self.execute( - f"mysql -sN -h {self.host} -u{self.user} -p{self.password}" + f"mariadb -sN -h {self.host} -u{self.user} -p{self.password}" f" {self.database} -e '{query}'" ) @@ -761,7 +761,7 @@ def get_database_size(self): " GROUP BY `table_schema`" ) command = ( - f"mysql -sN -h {self.host} -u{self.user} -p{self.password}" + f"mariadb -sN -h {self.host} -u{self.user} -p{self.password}" f" -e '{query}'" ) database_size = self.execute(command).get("output") @@ -810,7 +810,7 @@ def get_database_free_size(self): " GROUP BY `table_schema`" ) command = ( - f"mysql -sN -h {self.host} -u{self.user} -p{self.password}" + f"mariadb -sN -h {self.host} -u{self.user} -p{self.password}" f" -e '{query}'" ) database_size = self.execute(command).get("output") @@ -831,7 +831,7 @@ def get_database_free_tables(self): " OR `data_free` > 100 * 1024 * 1024)" ) command = ( - f"mysql -sN -h {self.host} -u{self.user} -p{self.password}" + f"mariadb -sN -h {self.host} -u{self.user} -p{self.password}" f" -e '{query}'" ) output = self.execute(command).get("output")