singlestore-sql-validator 0.0.3__tar.gz → 0.1.1__tar.gz

{singlestore_sql_validator-0.0.3 → singlestore_sql_validator-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: singlestore-sql-validator
-Version: 0.0.3
+Version: 0.1.1
 Summary: SQL grammar validator for SingleStore queries
 License: Apache License (2.0)
 License-File: LICENSE
@@ -21,6 +21,7 @@ Description-Content-Type: text/markdown
 
 # singlestore-sql-validator
 
+*Attention*: The code in this repository is intended for experimental use only and is not fully tested, documented, or supported by SingleStore. Visit the SingleStore Forums to ask questions about this repository.
 
 `singlestore-sql-validator` is a Python library for validating SQL queries using the SingleStore Language Server (version 0.1.3 and later). It provides a convenient interface to check SQL syntax and semantics by communicating with a SingleStore language server instance over TCP or WebSocket (with optional TLS support).
 
@@ -118,7 +119,7 @@ else:
 
 ## Notes
 
-- This library is intended to be used with a running SingleStore Language Server instance (version 0.1.3 or later).
+- This library is intended to be used with a running SingleStore Language Server instance (version 0.1.13 or later).
 - It does not start or manage the server process itself.
 - Documentation for the SingleStore Language Server can be found at: [https://github.com/singlestore-labs/language-server](https://github.com/singlestore-labs/language-server)
 

{singlestore_sql_validator-0.0.3 → singlestore_sql_validator-0.1.1}/README.md

@@ -1,5 +1,6 @@
 # singlestore-sql-validator
 
+*Attention*: The code in this repository is intended for experimental use only and is not fully tested, documented, or supported by SingleStore. Visit the SingleStore Forums to ask questions about this repository.
 
 `singlestore-sql-validator` is a Python library for validating SQL queries using the SingleStore Language Server (version 0.1.3 and later). It provides a convenient interface to check SQL syntax and semantics by communicating with a SingleStore language server instance over TCP or WebSocket (with optional TLS support).
 
@@ -97,7 +98,7 @@ else:
 
 ## Notes
 
-- This library is intended to be used with a running SingleStore Language Server instance (version 0.1.3 or later).
+- This library is intended to be used with a running SingleStore Language Server instance (version 0.1.13 or later).
 - It does not start or manage the server process itself.
 - Documentation for the SingleStore Language Server can be found at: [https://github.com/singlestore-labs/language-server](https://github.com/singlestore-labs/language-server)
 

{singlestore_sql_validator-0.0.3 → singlestore_sql_validator-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "singlestore-sql-validator"
-version = "0.0.3"
+version = "0.1.1"
 description = "SQL grammar validator for SingleStore queries"
 authors = [
     {name = "Volodymyr Tkachuk",email = "vtkachuk-ua@singlestore.com"}

{singlestore_sql_validator-0.0.3 → singlestore_sql_validator-0.1.1}/sql_validator/result.py

@@ -45,12 +45,14 @@ class ValidationResult:
     """
     Represents the result of validating an SQL query.
      - is_valid: Indicates if the SQL query is valid.
+     - is_complete: Indicates if the SQL query is complete and does not require additional tokens.
      - error_token: The token that caused the validation error, if any.
      - error_token_position: The position of the error token, if any.
      - alternative_tokens: A list of alternative tokens that could be used to fix the error, if any.
     """
 
     is_valid: bool
+    is_complete: bool
     error_token: Optional[str] = None
     error_token_position: Optional[TokenPosition] = None
     alternative_tokens: Optional[CompletionList] = None
@@ -64,6 +66,7 @@ class ValidationResult:
         """
         result = resp.get("result", {})
         is_valid = result.get("isValid", False)
+        is_complete = result.get("isComplete", False)
         error_token = result.get("errorToken")
 
         error_token_position = None
@@ -94,6 +97,7 @@ class ValidationResult:
 
         return cls(
             is_valid=is_valid,
+            is_complete=is_complete,
             error_token=error_token,
             error_token_position=error_token_position,
             alternative_tokens=alternative_tokens,

{singlestore_sql_validator-0.0.3 → singlestore_sql_validator-0.1.1}/sql_validator/validator.py

@@ -113,6 +113,7 @@ class Validator(ABC):
         # Initialize connection states
         self._ws_connection = None
         self._ws_context = None
+        self._ws_loop = None
 
         if self.connection_mode == ConnectionMode.TCP:
             self._connection = self._create_tcp_connection()
@@ -240,9 +241,11 @@ class Validator(ABC):
         data_str = json.dumps(request)
         data_with_content_length = f"Content-Length: {len(data_str)}\r\n\r\n{data_str}"
         if self.connection_mode == ConnectionMode.TCP:
-            # FIX: Double-check your server specification!
-            # If it expects \r\n or a null-byte (\x00), swap '\n' out here.
-            data_bytes = (data_with_content_length + "\n").encode("utf-8")
+            # LSP framing: the body must NOT be followed by any trailing bytes.
+            # An extra '\n' after the JSON payload causes the SingleStore
+            # language server to treat the connection as malformed and close
+            # it, breaking any subsequent validate() calls on the same socket.
+            data_bytes = data_with_content_length.encode("utf-8")
             # Send the data
             self._connection.sendall(data_bytes)
             # FIX: If using TLS, the underlying stream writer may need an explicit flush
@@ -255,10 +258,16 @@ class Validator(ABC):
             response = self._recv_all(self._connection)
         elif self.connection_mode in (ConnectionMode.WS, ConnectionMode.WSS):
             # WebSockets pass frames natively (newlines are not required as delimiters)
+            # Reuse a single event loop across calls so that the persistent
+            # websocket connection (which is bound to the loop that created it)
+            # stays valid for subsequent validate() invocations.
             try:
                 loop = asyncio.get_running_loop()
             except RuntimeError:
-                loop = asyncio.new_event_loop()
+                loop = self._ws_loop
+                if loop is None or loop.is_closed():
+                    loop = asyncio.new_event_loop()
+                    self._ws_loop = loop
                 asyncio.set_event_loop(loop)
 
             response = loop.run_until_complete(