Skip to content

Add DatabricksSQLStatementsOperator#48507

Merged
pankajkoti merged 8 commits into
apache:mainfrom
astronomer:add-databricks-sql-statements-operator-with-async
Apr 3, 2025
Merged

Add DatabricksSQLStatementsOperator#48507
pankajkoti merged 8 commits into
apache:mainfrom
astronomer:add-databricks-sql-statements-operator-with-async

Conversation

@pankajkoti

@pankajkoti pankajkoti commented Mar 28, 2025

Copy link
Copy Markdown
Member

This PR introduces the DatabricksSQLStatementsOperator to the Databricks provider, enabling the submission of dynamic SQL statements and polling their execution status. Additionally, it includes support for deferrable mode.

The operator utilises the Databricks Statement Execution API, allowing SQL submission via the POST endpoint and subsequent polling using the statement ID (returned by the POST endpoint) through the GET endpoint.

closes: #46549

@pankajkoti
pankajkoti force-pushed the add-databricks-sql-statements-operator-with-async branch 2 times, most recently from 1a49b2c to 2b079d3 Compare April 1, 2025 14:31
@pankajkoti
pankajkoti force-pushed the add-databricks-sql-statements-operator-with-async branch from 2b079d3 to d5b07d8 Compare April 1, 2025 15:10
@pankajkoti
pankajkoti marked this pull request as ready for review April 1, 2025 15:34
@pankajkoti

Copy link
Copy Markdown
Member Author

cc: @pankajastro @phanikumv @tatiana tagging for seeking review on the PR

@pankajkoti
pankajkoti requested a review from rawwar April 1, 2025 17:00
Comment thread providers/databricks/src/airflow/providers/databricks/hooks/databricks.py Outdated
Comment thread providers/databricks/src/airflow/providers/databricks/hooks/databricks.py Outdated
Comment thread providers/databricks/docs/operators/sql_statements.rst Outdated
@pankajkoti
pankajkoti requested review from phanikumv and rawwar April 2, 2025 07:52
@pankajkoti
pankajkoti force-pushed the add-databricks-sql-statements-operator-with-async branch 2 times, most recently from 260a0be to 42ab985 Compare April 2, 2025 18:56
@pankajkoti
pankajkoti force-pushed the add-databricks-sql-statements-operator-with-async branch from 6e7447f to 46d01f3 Compare April 3, 2025 07:50
@pankajkoti
pankajkoti merged commit 7fb5aba into apache:main Apr 3, 2025
@pankajkoti
pankajkoti deleted the add-databricks-sql-statements-operator-with-async branch April 3, 2025 10:47
@pankajkoti pankajkoti mentioned this pull request Apr 4, 2025
2 tasks
aaron-y-chen pushed a commit to aaron-y-chen/airflow that referenced this pull request Apr 4, 2025
This PR introduces the DatabricksSQLStatementsOperator to the Databricks provider, enabling the submission of dynamic SQL statements and polling their execution status. Additionally, it includes support for deferrable mode.

The operator utilises the [Databricks Statement Execution API](https://docs.databricks.com/api/workspace/statementexecution), allowing SQL submission via the [POST](https://docs.databricks.com/api/workspace/statementexecution/executestatement) endpoint and subsequent polling using the `statement ID` (returned by the POST endpoint) through the [GET](https://docs.databricks.com/api/workspace/statementexecution/getstatement) endpoint.

closes: apache#46549
diogotrodrigues pushed a commit to diogotrodrigues/airflow that referenced this pull request Apr 6, 2025
This PR introduces the DatabricksSQLStatementsOperator to the Databricks provider, enabling the submission of dynamic SQL statements and polling their execution status. Additionally, it includes support for deferrable mode.

The operator utilises the [Databricks Statement Execution API](https://docs.databricks.com/api/workspace/statementexecution), allowing SQL submission via the [POST](https://docs.databricks.com/api/workspace/statementexecution/executestatement) endpoint and subsequent polling using the `statement ID` (returned by the POST endpoint) through the [GET](https://docs.databricks.com/api/workspace/statementexecution/getstatement) endpoint.

closes: apache#46549
simonprydden pushed a commit to simonprydden/airflow that referenced this pull request Apr 8, 2025
This PR introduces the DatabricksSQLStatementsOperator to the Databricks provider, enabling the submission of dynamic SQL statements and polling their execution status. Additionally, it includes support for deferrable mode.

The operator utilises the [Databricks Statement Execution API](https://docs.databricks.com/api/workspace/statementexecution), allowing SQL submission via the [POST](https://docs.databricks.com/api/workspace/statementexecution/executestatement) endpoint and subsequent polling using the `statement ID` (returned by the POST endpoint) through the [GET](https://docs.databricks.com/api/workspace/statementexecution/getstatement) endpoint.

closes: apache#46549
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.

[Databricks provider] Add deferrable support to DatabricksSqlOperator

4 participants