sadpandajoe commented on code in PR #42900:
URL: https://github.com/apache/superset/pull/42900#discussion_r4020926452


##########
superset/db_engine_specs/databricks.py:
##########
@@ -775,6 +776,11 @@ class 
DatabricksPythonConnectorEngineSpec(DatabricksDynamicBaseEngineSpec):
 
     parameters_schema = DatabricksPythonConnectorSchema()
 
+    # The driver only populates `cursor.active_command_id` once a statement has
+    # actually been executed, so the cancel id can't be captured up front like
+    # it can for engines where it's tied to the session rather than the query.
+    has_query_id_before_execute = False

Review Comment:
   Agreed—the post-execute hook cannot make this cancellable while the 
statement is running because BaseEngineSpec.execute() blocks in 
cursor.execute(), so the SEA ID is persisted only after completion and a later 
block can expose the previous block's stale ID. Could this path publish the 
current ID before polling and cover Stop during an in-flight statement?



-- 
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]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to