Enable DatabricksJobRunLink for Databricks plugin; skip provide_session usage and direct db calls in Airflow3 #52228
+377
−90
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR introduces support for the "See Databricks Job Run" extra link in the Databricks workflow provider plugin for Airflow 3. The implementation stores the job run URL in XCom during task execution and retrieves it when the extra link is accessed.
Additionally, when using Airflow 3, the PR refactors the plugin code to eliminate the use of
@provide_session
and direct database access for compatibility with Airflow 3. These changes address the concerns raised in issue #49187 regarding the Databricks provider plugin.Support for the Databricks workflow repair functionality in Airflow 3 is still pending. A follow-up issue #52280 has been filed to explore a new approach for implementing repair in Airflow 3.
Related: #49187