Exclude db_restore from waiting on proportional index dag

WordPress · Mar 14, 2024 · cffaa7a · cffaa7a
1 parent 34f59de
commit cffaa7a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/catalog/dags/database/staging_database_restore/staging_database_restore_dag.py b/catalog/dags/database/staging_database_restore/staging_database_restore_dag.py
@@ -54,6 +54,9 @@
     restore_staging_from_snapshot,
     skip_restore,
 )
+from elasticsearch_cluster.create_proportional_by_source_staging_index.create_proportional_by_source_staging_index_dag import (
+    DAG_ID as CREATE_PROPORTIONAL_BY_SOURCE_DAG_ID,
+)
 
 
 log = logging.getLogger(__name__)
@@ -80,6 +83,9 @@ def restore_staging_database():
     # Wait for any DAGs that operate on the staging elasticsearch cluster
     wait_for_recreate_full_staging_index = wait_for_external_dags_with_tag(
         tag=STAGING_ES_CONCURRENCY_TAG,
+        # No need to wait on the proportional index DAG, which does not use
+        # the underlying database
+        excluded_dag_ids=[CREATE_PROPORTIONAL_BY_SOURCE_DAG_ID],
     )
     should_skip = skip_restore()
     latest_snapshot = get_latest_prod_snapshot()