From 7fb971f2519b9a58115875e51647eb8c17bf93bb Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Sat, 4 May 2024 10:33:51 -0400 Subject: [PATCH] Skip Aurora table deletion to speed up restarts --- src/brad/config/file.py | 8 ++++++++ src/brad/daemon/transition_orchestrator.py | 1 + 2 files changed, 9 insertions(+) diff --git a/src/brad/config/file.py b/src/brad/config/file.py index b8ef4054..c14facc2 100644 --- 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 index 8a94165e..a08f5e26 100644 --- 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(