Programmer-RD-AI opened a new pull request, #56164: URL: https://github.com/apache/airflow/pull/56164
This PR introduces cancellation support for Snowflake queries executed through the SQL API: * Extend `get_request_url_header_params` to accept an optional `url_suffix` for appending endpoint paths (e.g., `/cancel` or `/result`). * Add `_cancel_sql_api_query_execution` and `cancel_queries` methods to `SnowflakeSqlApiHook` for invoking the `/cancel` endpoint. * Implement `on_kill` in `SnowflakeSqlApiOperator` to automatically cancel active queries when a task is killed. * Add unit test verifying that cancellation requests are issued for all tracked query IDs. Closes: #54451 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
