Expose an API which can return dagRun status in most optimized time

[sonalprsd]

Description

Airflow/MWAA does not seem to have any scalable API for returning the status of a dagRun, the APIs states-for-dag-run or list-runs are not scaling well. To fetch the dagRun status, every team seems to have some custom solution using sns_notification or updating the status to some external data store via Airflow callbacks.

The ask is to expose an API which can return dagRun status in most optimized time/by an internal query operation and not a scan.

Discussion #27765

Use case/motivation

My use case is to fetch the Dag status of all the Active runs and update the status tables in the system. There is a poller (with a timeout of 150s configured based on our SLA). The states-for-dag-run API seems to be doing scan operation internally. As the number of DAG runs in system increases, the time to get the status of dagRun increases further. Initially, fetching the status of 100 runs took 2.5 minutes. With increase of dagRuns in the system by 50, the fetch operation to get status for 100 dagRuns is taking more than 5 minutes.

Related issues

NA

Are you willing to submit a PR?

• Yes I am willing to submit a PR!

Code of Conduct

• I agree to follow this project's Code of Conduct

[boring-cyborg]

Thanks for opening your first issue here! Be sure to follow the issue template!

[potiuk]

Marked as good first issue.

[potiuk]

As discussed in #27765 just exposing updated_at should be enough

[Adityamalik123]

@potiuk Can i take this task up?

[potiuk]

assigned.

[vincbeck]

Hey @Adityamalik123. Did you get a chance to start working on this task? There is absolutely no rush, I was just asking in case you need help

[Adityamalik123]

Hey @Adityamalik123. Did you get a chance to start working on this task? There is absolutely no rush, I was just asking in case you need help

Thanks for checking in @vincbeck. I am planning to get started (and probably wrap this task up) this week, I will definitely reach you out in case i get stuck.

[sonalprsd]

Hi @Adityamalik123, can you share the documentation for the API you wrote as part of this task?