Skip to content

Commit

Permalink
fix: Use mariadb commands instead of mysql
Browse files Browse the repository at this point in the history
  • Loading branch information
cogk committed Jul 2, 2024
1 parent fe8dfef commit f615f9a
Show file tree
Hide file tree
Showing 4 changed files with 16 additions and 16 deletions.
4 changes: 2 additions & 2 deletions agent/bench.py
Original file line number Diff line number Diff line change
Expand Up @@ -257,7 +257,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)
Expand All @@ -273,7 +273,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)
Expand Down
2 changes: 1 addition & 1 deletion agent/database.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,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}'"
Expand Down
2 changes: 1 addition & 1 deletion agent/proxysql.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}'"
)
Expand Down
24 changes: 12 additions & 12 deletions agent/site.py
Original file line number Diff line number Diff line change
Expand Up @@ -271,7 +271,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)
Expand All @@ -287,7 +287,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)
Expand Down Expand Up @@ -317,7 +317,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",
)
Expand Down Expand Up @@ -460,7 +460,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}'",
Expand Down Expand Up @@ -530,7 +530,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",
)
Expand All @@ -545,7 +545,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
Expand Down Expand Up @@ -627,7 +627,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'-sN -e "{query}"'
)["output"].strip()
except Exception:
Expand All @@ -637,7 +637,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")

Expand Down Expand Up @@ -671,7 +671,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}'"
)

Expand Down Expand Up @@ -747,7 +747,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")
Expand Down Expand Up @@ -788,7 +788,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")
Expand All @@ -809,7 +809,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")
Expand Down

0 comments on commit f615f9a

Please sign in to comment.