Skip to content

Conversation

@hussein-awala
Copy link
Member

related: #7370

This PR improves the performance of the method _manage_executor_state by fetching the state of all the running TIs in a single query instead of refreshing them one by one from the DB.

@boring-cyborg boring-cyborg bot added the area:Scheduler including HA (high availability) scheduler label Dec 25, 2023
@hussein-awala hussein-awala added the use public runners Makes sure that Public runners are used even if commiters creates the PR (useful for testing) label Dec 25, 2023
@hussein-awala hussein-awala reopened this Dec 25, 2023
Copy link
Contributor

@dirrao dirrao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice work. LGTM.

@potiuk
Copy link
Member

potiuk commented Dec 29, 2023

Nice one! @hussein-awala -> I just wonder if that could be the reason for #36454

@potiuk potiuk added this to the Airflow 2.8.1 milestone Dec 29, 2023
@potiuk potiuk merged commit 9d45db9 into apache:main Dec 30, 2023
@potiuk
Copy link
Member

potiuk commented Dec 30, 2023

It fails in main for MSSQL. We are planning to cherry-pick that one for 2.8.1 so I hope it's the last that requires MSSQL special treatment (I guess you need to open fixup PR and set full-tests-needed

potiuk added a commit to potiuk/airflow that referenced this pull request Dec 30, 2023
potiuk added a commit that referenced this pull request Dec 30, 2023
hussein-awala added a commit to hussein-awala/airflow that referenced this pull request Dec 30, 2023
potiuk pushed a commit that referenced this pull request Dec 30, 2023
Refactor _manage_executor_state by refreshing TIs in batch (#36418)" (#36500)"

Handle Microsoft SQL Server
@ephraimbuddy ephraimbuddy added type:misc/internal Changelog: Misc changes that should appear in change log changelog:skip Changes that should be skipped from the changelog (CI, tests, etc..) and removed type:misc/internal Changelog: Misc changes that should appear in change log labels Jan 10, 2024
ephraimbuddy pushed a commit that referenced this pull request Jan 11, 2024
* Refactor _manage_executor_state by refreshing TIs in batch

* Use a short key without retry number

(cherry picked from commit 9d45db9)
ephraimbuddy pushed a commit that referenced this pull request Jan 11, 2024
ephraimbuddy pushed a commit that referenced this pull request Jan 11, 2024
Refactor _manage_executor_state by refreshing TIs in batch (#36418)" (#36500)"

Handle Microsoft SQL Server

(cherry picked from commit 9cf5f6f)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:Scheduler including HA (high availability) scheduler changelog:skip Changes that should be skipped from the changelog (CI, tests, etc..) use public runners Makes sure that Public runners are used even if commiters creates the PR (useful for testing)

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants