Skip to content

Conversation

kaxil
Copy link
Member

@kaxil kaxil commented May 28, 2025

closes #50500

Adds a new safeguard for cases where the task subprocess closes before all pipe sockets send EOF.

While the supervisor "shouldn't" get stuck, the reality is that under extreme load the selector can miss final EOF events - which might have happened previously under high CPU load + the bug in #51023. This adds a safety net to prevent resource leaks and hanging processes (Pods accumulate for CE & KE). The timeout is generous (60s) to avoid premature cleanup of legitimate slow socket operations. I tried reproducing it reliably after my first attempt in #51020 but haven't been able to reproduce it.

Changes

  • Add workers.socket_cleanup_timeout config (default 60s) to force-close stuck sockets after task process exits
  • Improve child process socket cleanup to reduce EOF misses
  • Add some logs for better FD identification if this happens. Previously when I saw it, it was mainly requests & logs fd.

^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in airflow-core/newsfragments.

closes apache#50500

Adds a new safeguard for cases where the task subprocess closes
before all pipe sockets send EOF.

The supervisor now records the
process exit time and forcibly closes any sockets still open after
`workers.socket_cleanup_timeout`. This stops the
supervisor loop from hanging indefinitely and allows the process
to exit cleanly.
Copy link
Contributor

@amoghrajesh amoghrajesh left a comment

Choose a reason for hiding this comment

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

Seems like something that could work, lets ship it

Co-authored-by: Amogh Desai <[email protected]>
@kaxil kaxil merged commit a2651f1 into apache:main May 29, 2025
71 checks passed
@kaxil kaxil deleted the socket-timeout branch May 29, 2025 09:34
sanederchik pushed a commit to sanederchik/airflow that referenced this pull request Jun 7, 2025
closes apache#50500

Adds a new safeguard for cases where the task subprocess closes
before all pipe sockets send EOF.

The supervisor now records the
process exit time and forcibly closes any sockets still open after
`workers.socket_cleanup_timeout`. This stops the
supervisor loop from hanging indefinitely and allows the process
to exit cleanly.
ashb pushed a commit that referenced this pull request Jun 20, 2025
closes #50500

Adds a new safeguard for cases where the task subprocess closes
before all pipe sockets send EOF.

The supervisor now records the
process exit time and forcibly closes any sockets still open after
`workers.socket_cleanup_timeout`. This stops the
supervisor loop from hanging indefinitely and allows the process
to exit cleanly.

(cherry picked from commit a2651f1)
github-actions bot pushed a commit that referenced this pull request Jun 20, 2025
…1180)

closes #50500

Adds a new safeguard for cases where the task subprocess closes
before all pipe sockets send EOF.

The supervisor now records the
process exit time and forcibly closes any sockets still open after
`workers.socket_cleanup_timeout`. This stops the
supervisor loop from hanging indefinitely and allows the process
to exit cleanly.
(cherry picked from commit a2651f1)

Co-authored-by: Kaxil Naik <[email protected]>
Copy link

Backport successfully created: v3-0-test

Status Branch Result
v3-0-test PR Link

jedcunningham pushed a commit that referenced this pull request Jun 20, 2025
…1180) (#51970)

closes #50500

Adds a new safeguard for cases where the task subprocess closes
before all pipe sockets send EOF.

The supervisor now records the
process exit time and forcibly closes any sockets still open after
`workers.socket_cleanup_timeout`. This stops the
supervisor loop from hanging indefinitely and allows the process
to exit cleanly.
(cherry picked from commit a2651f1)

Co-authored-by: Kaxil Naik <[email protected]>
jose-lehmkuhl pushed a commit to jose-lehmkuhl/airflow that referenced this pull request Jul 11, 2025
closes apache#50500

Adds a new safeguard for cases where the task subprocess closes
before all pipe sockets send EOF.

The supervisor now records the
process exit time and forcibly closes any sockets still open after
`workers.socket_cleanup_timeout`. This stops the
supervisor loop from hanging indefinitely and allows the process
to exit cleanly.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

The task supervisor continues running indefinitely, even after the associated task process has completed
2 participants