[MDB-29707] Metatdata cleaner with system drop replica (#241)

* Metatdata cleaner with system drop replica * Add replicated databases in conf * Fix 22.8 * Review fixes * More fixes
yandex · Oct 8, 2024 · 6502ec3 · 6502ec3
1 parent 8e11eb0
commit 6502ec3
Show file tree

Hide file tree

Showing 6 changed files with 356 additions and 142 deletions.
diff --git a/ch_tools/chadmin/internal/database_replica.py b/ch_tools/chadmin/internal/database_replica.py
@@ -0,0 +1,10 @@
+from ch_tools.chadmin.internal.utils import execute_query
+
+
+def system_database_drop_replica(ctx, database_zk_path, replica, dry_run=False):
+    """
+    Perform "SYSTEM DROP DATABASE REPLICA" query.
+    """
+    timeout = ctx.obj["config"]["clickhouse"]["drop_replica_timeout"]
+    query = f"SYSTEM DROP DATABASE REPLICA '{replica}' FROM ZKPATH '{database_zk_path}'"
+    execute_query(ctx, query, timeout=timeout, echo=True, dry_run=dry_run, format_=None)
diff --git a/ch_tools/chadmin/internal/table_replica.py b/ch_tools/chadmin/internal/table_replica.py
@@ -144,3 +144,12 @@ def restore_table_replica(
     if cluster:
         query += f" ON CLUSTER '{cluster}'"
     execute_query(ctx, query, timeout=timeout, echo=True, dry_run=dry_run, format_=None)
+
+
+def system_table_drop_replica(ctx, table_zk_path, replica, dry_run=False):
+    """
+    Perform "SYSTEM DROP REPLICA" query.
+    """
+    timeout = ctx.obj["config"]["clickhouse"]["drop_replica_timeout"]
+    query = f"SYSTEM DROP REPLICA '{replica}' FROM ZKPATH '{table_zk_path}'"
+    execute_query(ctx, query, timeout=timeout, echo=True, dry_run=dry_run, format_=None)