Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 11 additions & 6 deletions airflow-core/src/airflow/models/asset.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,12 +112,17 @@ def remove_references_to_deleted_dags(session: Session):
DagScheduleAssetAliasReference,
TaskOutletAssetReference,
]
for model in models_to_check:
session.execute(
delete(model)
.where(model.dag_id.in_(select(DagModel.dag_id).where(DagModel.is_stale)))
.execution_options(synchronize_session="fetch")
)

# The queries need to be done in separate steps, because in the case of multiple
# dag processors on MySQL, there could be a deadlock caused by acquiring both an
# exclusive lock for deletion and shared lock for query in reverse sequence
if stale_dag_ids := session.scalars(select(DagModel.dag_id).where(DagModel.is_stale)).all():
for model in models_to_check:
session.execute(
delete(model)
.where(model.dag_id.in_(stale_dag_ids))
.execution_options(synchronize_session="fetch")
)


alias_association_table = Table(
Expand Down