Bläddra i källkod

HUE-9270 [sqlalchemy] Reuse close_statement in cancel API

Romain 5 år sedan
förälder
incheckning
c46b8ad8c9
1 ändrade filer med 1 tillägg och 10 borttagningar
  1. 1 10
      desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

+ 1 - 10
desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

@@ -269,16 +269,7 @@ class SqlAlchemyApi(Api):
 
   @query_error_handler
   def cancel(self, notebook, snippet):
-    result = {'status': -1}
-    try:
-      guid = snippet['result']['handle']['guid']
-      connection = CONNECTION_CACHE.get(guid)
-      if connection:
-        connection['connection'].close()
-        del CONNECTION_CACHE[guid]
-      result['status'] = 0
-    finally:
-      return result
+    return self.close_statement(notebook, snippet)
 
 
   @query_error_handler