Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms"); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links"); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + '
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode"); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })();
Skip to content

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests - #62369

Merged
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries
Mar 18, 2026
Merged

Stream grid task-instance summaries as NDJSON to eliminate N+1 requests#62369
pierrejeambrun merged 8 commits into
apache:mainfrom
xvega:feat/batch-ti-summaries

Conversation

@xvega

@xvegaxvega commented Feb 23, 2026

Copy link
Copy Markdown
Contributor

This pull request introduces a significant improvement to the Airflow Grid view by adding a new streaming API endpoint for fetching task-instance summaries for multiple DAG runs in a single request. This change reduces the number of HTTP requests from one-per-run to a single NDJSON (newline-delimited JSON) stream, allowing the frontend to progressively render grid columns as data arrives, improving both performance and user experience. The implementation includes both backend (Python/FastAPI) and frontend (TypeScript/React) changes to support the new streaming behavior, as well as updates to API documentation and client code generation.

Backend API and Streaming Implementation:

  • Added a new endpoint GET /ui/grid/ti_summaries/{dag_id} that streams task-instance summaries for multiple DAG runs as NDJSON, emitting one JSON object per run, and reuses the serialized DAG structure for runs with the same version to avoid redundant deserialization. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py, airflow-core/src/airflow/api_fastapi/core_api/openapi/_private_ui.yaml) [1][2]

  • Refactored backend logic to build and yield summaries efficiently, using a generator and caching the serialized DAG model per version. (airflow-core/src/airflow/api_fastapi/core_api/routes/ui/grid.py)

Frontend and Client Code Generation:

  • Generated new TypeScript client hooks and utilities to consume the NDJSON stream, including useGridServiceGetGridTiSummariesStream, and updated related query, prefetch, and suspense hooks. (airflow-core/src/airflow/ui/openapi-gen/queries/common.ts, airflow-core/src/airflow/ui/openapi-gen/queries/queries.ts, airflow-core/src/airflow/ui/openapi-gen/queries/prefetch.ts, airflow-core/src/airflow/ui/openapi-gen/queries/suspense.ts, airflow-core/src/airflow/ui/openapi-gen/queries/ensureQueryData.ts) [1][2][3][4][5]

Documentation and Release Notes:

  • Added a detailed release note describing the new streaming behavior, endpoint, and frontend integration, highlighting the performance and UX improvements. (airflow-core/newsfragments/62369.significant.rst)

Overall, this change streamlines the Grid view's data fetching, reduces server and network load, and enables a smoother, more responsive UI for users monitoring multiple DAG runs.

Fixes#63720

@boring-cyborg

Copy link
Copy Markdown

Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contributors' Guide (https://github.com/apache/airflow/blob/main/contributing-docs/README.rst)
Here are some useful points:

  • Pay attention to the quality of your code (ruff, mypy and type annotations). Our prek-hooks will help you with that.
  • In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
  • Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
  • Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
  • Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
  • Be sure to read the Airflow Coding style.
  • Always keep your Pull Requests rebased, otherwise your build might fail due to changes not related to your commits.
    Apache Airflow is a community-driven project and together we are making it better 🚀.
    In case of doubts contact the developers at:
    Mailing List: dev@airflow.apache.org
    Slack: https://s.apache.org/airflow-slack

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 696c19e to 84d786bCompareFebruary 24, 2026 07:36
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 4 times, most recently from 5ebf3b3 to b8d6df7CompareFebruary 25, 2026 12:57
@jedcunningham

Copy link
Copy Markdown
Member

Please make sure to test with many very large dag runs - we intentionally moved from a single call to 1 per in #51805. There is a dag in there you can use.

@xvegaxvega changed the title Add batch endpoint for grid TI summaries to reduce N+1 requestsStream grid task-instance summaries as NDJSON to eliminate N+1 requestsFeb 28, 2026
@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from fd62a40 to 7f65f2fCompareFebruary 28, 2026 13:59

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

@bbovenzi

Copy link
Copy Markdown
Contributor

Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 17f6302 to 900882dCompareMarch 16, 2026 19:04
@xvega

Copy link
Copy Markdown
ContributorAuthor

Looks good to me. Tested and working as expected.

Need to fix the CI. (rebase + regenerate api + ui code gen, formatting, and eslint small issue)

Fixed the above-mentioned issues 8e2c186

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch 2 times, most recently from 3087603 to a615be4CompareMarch 17, 2026 12:11
@xvega

Copy link
Copy Markdown
ContributorAuthor

@bbovenzi >Looks like we need to rerun our type generation and formatting. And then double check our e2e tests

I noticed the tests weren’t passing, but they didn’t seem to be related to the changes I made.

@xvega
xvegaforce-pushed the feat/batch-ti-summaries branch from 2ab365a to 40432e1CompareMarch 18, 2026 08:01

@pierrejeambrunpierrejeambrun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

looks like e2e tests are not happy.

@pierrejeambrun
pierrejeambrun merged commit 7c77a6d into apache:mainMar 18, 2026
375 of 380 checks passed
@boring-cyborg

Copy link
Copy Markdown

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
imrichardwu pushed a commit to imrichardwu/airflow that referenced this pull request Mar 18, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
fat-catTW pushed a commit to fat-catTW/airflow that referenced this pull request Mar 22, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
techcodie pushed a commit to techcodie/airflow that referenced this pull request Mar 23, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
abhijeets25012-tech pushed a commit to abhijeets25012-tech/airflow that referenced this pull request Apr 9, 2026
…ts (apache#62369)
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection to eliminate N+1 requests
* Stream task instance summaries for multiple DAG runs over a single NDJSON connection, replacing individual requests to improve performance and eliminate N+1 query issues.
* Fix capitalization of "Dag" in documentation and code comments for consistency
* Refactor GridTISummaries schema and update streaming endpoint to improve clarity and performance
* Fix formatting and linter issues
* Fix static check
* Fix static check
* Fix static check
@pierrejeambrunpierrejeambrun mentioned this pull request Jul 15, 2026
2 tasks
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:APIAirflow's REST/HTTP APIarea:UIRelated to UI/UX. For Frontend Developers.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Task and dagrun state is not in sync when user clears the dagrun

5 participants

@xvega@jedcunningham@bbovenzi@pierrejeambrun@choo121600