introduce normalised sessionId and CommandId for (near) complete backend abstraction

status is still aligned with Thrift

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/backend/databricks_client.py

@@ -2,6 +2,7 @@
 from typing import Dict, Tuple, List, Optional, Any, Union
 
 from databricks.sql.thrift_api.TCLIService import ttypes
+from databricks.sql.ids import SessionId, CommandId
 from databricks.sql.utils import ExecuteResponse
 from databricks.sql.types import SSLOptions
 
@@ -20,19 +21,19 @@ def open_session(
         session_configuration: Optional[Dict[str, Any]],
         catalog: Optional[str],
         schema: Optional[str],
-    ) -> ttypes.TOpenSessionResp:
+    ) -> SessionId:
         pass
 
     @abstractmethod
-    def close_session(self, session_handle: ttypes.TSessionHandle) -> None:
+    def close_session(self, session_id: SessionId) -> None:
         pass
 
     # == Query Execution, Command Management ==
     @abstractmethod
     def execute_command(
         self,
         operation: str,
-        session_handle: ttypes.TSessionHandle,
+        session_id: SessionId,
         max_rows: int,
         max_bytes: int,
         lz4_compression: bool,
@@ -45,25 +46,21 @@ def execute_command(
         pass
 
     @abstractmethod
-    def cancel_command(self, operation_handle: ttypes.TOperationHandle) -> None:
+    def cancel_command(self, command_id: CommandId) -> None:
         pass
 
     @abstractmethod
-    def close_command(
-        self, operation_handle: ttypes.TOperationHandle
-    ) -> ttypes.TStatus:
+    def close_command(self, command_id: CommandId) -> ttypes.TStatus:
         pass
 
     @abstractmethod
-    def get_query_state(
-        self, operation_handle: ttypes.TOperationHandle
-    ) -> ttypes.TOperationState:
+    def get_query_state(self, command_id: CommandId) -> ttypes.TOperationState:
         pass
 
     @abstractmethod
     def get_execution_result(
         self,
-        operation_handle: ttypes.TOperationHandle,
+        command_id: CommandId,
         cursor: Any,
     ) -> ExecuteResponse:
         pass
@@ -72,7 +69,7 @@ def get_execution_result(
     @abstractmethod
     def get_catalogs(
         self,
-        session_handle: ttypes.TSessionHandle,
+        session_id: SessionId,
         max_rows: int,
         max_bytes: int,
         cursor: Any,
@@ -82,7 +79,7 @@ def get_catalogs(
     @abstractmethod
     def get_schemas(
         self,
-        session_handle: ttypes.TSessionHandle,
+        session_id: SessionId,
         max_rows: int,
         max_bytes: int,
         cursor: Any,
@@ -94,7 +91,7 @@ def get_schemas(
     @abstractmethod
     def get_tables(
         self,
-        session_handle: ttypes.TSessionHandle,
+        session_id: SessionId,
         max_rows: int,
         max_bytes: int,
         cursor: Any,
@@ -108,7 +105,7 @@ def get_tables(
     @abstractmethod
     def get_columns(
         self,
-        session_handle: ttypes.TSessionHandle,
+        session_id: SessionId,
         max_rows: int,
         max_bytes: int,
         cursor: Any,
@@ -121,11 +118,11 @@ def get_columns(
 
     # == Utility Methods ==
     @abstractmethod
-    def handle_to_id(self, handle: ttypes.TSessionHandle) -> bytes:
+    def handle_to_id(self, session_id: SessionId) -> Any:
         pass
 
     @abstractmethod
-    def handle_to_hex_id(self, handle: ttypes.TSessionHandle) -> str:
+    def handle_to_hex_id(self, session_id: SessionId) -> str:
         pass
 
     # Properties related to specific backend features