Skip to content

Add dag_bundle_name option to subprocess coordinators - #70805

Open
jason810496 wants to merge 4 commits into
apache:mainfrom
jason810496:feature/coordinator/dag-bundle-name
Open

Add dag_bundle_name option to subprocess coordinators#70805
jason810496 wants to merge 4 commits into
apache:mainfrom
jason810496:feature/coordinator/dag-bundle-name

Conversation

@jason810496

@jason810496 jason810496 commented Jul 31, 2026

Copy link
Copy Markdown
Member

Why

Language-SDK subprocess coordinators (Java, Executable, Node) required a hand-managed filesystem root (jars_root / executables_root / bundles_root) for compiled artifacts. Users should be able to deploy those artifacts by existing DagBundle mechanism as well.

How

  • Add an optional dag_bundle_name kwarg on the SubprocessCoordinator base, giving three ways to locate artifacts:
    • explicit root set: scan that path (unchanged legacy behavior).
    • dag_bundle_name set: scan that configured bundle's path at its latest version.
    • neither set: scan the task's own bundle, pinned to the run's version (artifacts co-located with the Python stub Dag).
  • Keep the subclass _build_execute_task_command signature unchanged: execute_task stashes the task's bundle_info for the duration of the call, and _scan_roots resolves the scan root on demand.
  • Reuse the Dag-bundle lifecycle through task_runner.initialize_ti_bundle (extracted from parse) for resolution.
  • Validate at construction that an explicit root and dag_bundle_name are not both set, and that a named bundle is configured via a new lightweight DagBundlesManager.is_bundle_configured that checks configuration without constructing the bundle instance.

Notes

  • Version semantics are intentional: a co-located bundle (no dag_bundle_name) is pinned to the run's bundle_info.version because it is the same artifact the run was created from, while a separately named artifact bundle tracks its own latest release independently of any run.

Was generative AI tooling used to co-author this PR?

@boring-cyborg boring-cyborg Bot added area:coordinator Coordinator: The interface to spawn Lang-SDK subprocesses area:DAG-processing area:task-sdk labels Jul 31, 2026
@jason810496 jason810496 self-assigned this Jul 31, 2026
Comment thread task-sdk/src/airflow/sdk/coordinators/_subprocess.py Outdated
Comment thread task-sdk/src/airflow/sdk/coordinators/_subprocess.py
Comment thread task-sdk/src/airflow/sdk/coordinators/_subprocess.py
Comment thread task-sdk/src/airflow/sdk/coordinators/_subprocess.py
Comment thread airflow-core/src/airflow/dag_processing/bundles/manager.py Outdated
Comment thread task-sdk/src/airflow/sdk/coordinators/java/coordinator.py
Comment thread task-sdk/src/airflow/sdk/coordinators/_subprocess.py Outdated
Comment thread task-sdk/tests/task_sdk/coordinators/executable/test_coordinator.py Outdated
@jason810496
jason810496 marked this pull request as draft July 31, 2026 15:24
@jason810496
jason810496 force-pushed the feature/coordinator/dag-bundle-name branch from 1b76bfe to 182919a Compare August 1, 2026 15:29
@jason810496
jason810496 marked this pull request as ready for review August 3, 2026 08:54
@jason810496 jason810496 added this to the Airflow 3.4.0 milestone Aug 5, 2026
@jason810496
jason810496 force-pushed the feature/coordinator/dag-bundle-name branch 2 times, most recently from bc97b03 to 12de628 Compare August 5, 2026 06:34
Language-SDK coordinators previously required a hand-managed filesystem root (jars_root/executables_root/bundles_root) for compiled artifacts, kept outside Airflow's Dag-bundle machinery and prone to drifting from the Dags they serve. Allowing artifacts to be located through a configured Dag bundle -- or the task's own bundle when none is named -- lets them be delivered and versioned the same way as everything else.
Language-SDK subprocess coordinators materialized artifact bundles without
the version-lock protection the Python task path holds, so bundle cleanup
could delete a version a running JVM was still reading from. Resolution now
holds the version lock for the subprocess lifetime, mirroring the Python task
path, and materialization failures surface in the task log rather than the
worker log.

Classification moves to the base coordinator so a subclass that adds no
explicit-root kwarg falls back to the task-bundle default instead of failing
every task, the Java coordinator now requires an explicit entrypoint when it
has to scan a whole bundle, and configured-bundle lookup checks names only so
an unrelated unimportable bundle no longer breaks coordinator construction.
When a subprocess coordinator locates artifacts through a Dag bundle rather than an explicit root, JavaCoordinator previously rejected a missing main_class outright. That made the co-located default unusable without extra configuration and left the entrypoint no more determinate than an auto-detected scan would be. Auto-detect over the resolved bundle now handles that case, with the remaining cross-bundle ambiguity deferred to an IMPORT_ERROR-stage check under AIP-85.

A named Dag bundle resolves to its latest version, so its version was unset and the version lock held around the subprocess silently did nothing, letting bundle cleanup delete artifacts a running task was still reading. The concrete current version is now pinned before locking.
@jason810496
jason810496 force-pushed the feature/coordinator/dag-bundle-name branch from 12de628 to c7ab92a Compare August 6, 2026 03:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:coordinator Coordinator: The interface to spawn Lang-SDK subprocesses area:DAG-processing area:task-sdk

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Java-SDK artifact deployment strategies

2 participants