Skip Aurora table deletion to speed up restarts

mitdbg · May 4, 2024 · 7fb971f · 7fb971f
1 parent 68954e7
commit 7fb971f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/brad/config/file.py b/src/brad/config/file.py
@@ -206,6 +206,14 @@ def skip_athena_table_deletion(self) -> bool:
             # Skip by default.
             return True
 
+    @property
+    def skip_aurora_table_deletion(self) -> bool:
+        try:
+            return self._raw["skip_aurora_table_deletion"]
+        except KeyError:
+            # Skip by default.
+            return True
+
     @property
     def use_preset_redshift_clusters(self) -> bool:
         try:

diff --git a/src/brad/daemon/transition_orchestrator.py b/src/brad/daemon/transition_orchestrator.py
@@ -484,6 +484,7 @@ async def _run_aurora_post_transition(
             table_diffs is not None
             and len(table_diffs) > 0
             and self._config.disable_table_movement is False
+            and self._config.skip_aurora_table_deletion is False
         ):
             if self._system_event_logger is not None:
                 self._system_event_logger.log(