eschutho commented on code in PR #16326:
URL: https://github.com/apache/superset/pull/16326#discussion_r906591207
##########
superset/db_engine_specs/redshift.py:
##########
@@ -101,3 +105,39 @@ def _mutate_label(label: str) -> str:
:return: Conditionally mutated label
"""
return label.lower()
+
+ @classmethod
+ def get_cancel_query_id(cls, cursor: Any, query: Query) -> Optional[str]:
+ """
+ Get Redshift PID that will be used to cancel all other running
+ queries in the same session.
+
+ :param cursor: Cursor instance in which the query will be executed
+ :param query: Query instance
+ :return: Redshift PID
+ """
+ cursor.execute("SELECT pg_backend_pid()")
+ row = cursor.fetchone()
+ return row[0]
+
+ @classmethod
+ def cancel_query(cls, cursor: Any, query: Query, cancel_query_id: str) ->
bool:
+ """
+ Cancel query in the underlying database.
+
+ :param cursor: New cursor instance to the db of the query
+ :param query: Query instance
+ :param cancel_query_id: Redshift PID
+ :return: True if query cancelled successfully, False otherwise
+ """
+ try:
+ logger.info("Killing Redshift PID:%s",str(cancel_query_id))
Review Comment:
```suggestion
logger.info("Killing Redshift PID:%s", str(cancel_query_id))
```
--
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]
