Skip to content

Enable DatabricksJobRunLink for Databricks plugin; skip provide_session usage and direct db calls in Airflow3 #52228

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jun 27, 2025

Conversation

pankajkoti
Copy link
Member

@pankajkoti pankajkoti commented Jun 25, 2025

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

@pankajkoti pankajkoti force-pushed the databricks-plugin-airflow3 branch 2 times, most recently from 7910e95 to a93701c Compare June 25, 2025 08:51
@pankajkoti
Copy link
Member Author

Databricks Job Run Link - extra link in action
Jun-26-2025 13-11-58

@pankajkoti pankajkoti changed the title Enable DatabricksJobRunLink for Databricks plugin, skip provide_session usage in Airflow3 Enable DatabricksJobRunLink for Databricks plugin, skip provide_session usage and direct db calls in Airflow3 Jun 26, 2025
@pankajkoti pankajkoti changed the title Enable DatabricksJobRunLink for Databricks plugin, skip provide_session usage and direct db calls in Airflow3 Enable DatabricksJobRunLink for Databricks plugin; skip provide_session usage and direct db calls in Airflow3 Jun 26, 2025
@pankajkoti pankajkoti marked this pull request as ready for review June 26, 2025 08:11
Copy link
Member

@potiuk potiuk left a comment

Choose a reason for hiding this comment

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

Nice!

@pankajkoti pankajkoti force-pushed the databricks-plugin-airflow3 branch from a93701c to ae34034 Compare June 27, 2025 07:02
@pankajkoti pankajkoti merged commit a3a6fe9 into apache:main Jun 27, 2025
9 checks passed
@pankajkoti pankajkoti deleted the databricks-plugin-airflow3 branch June 27, 2025 07:03
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.

2 participants