appmesh 1.6.0__py3-none-any.whl → 1.6.1__py3-none-any.whl

appmesh/tcp_transport.py

@@ -108,7 +108,7 @@ class TCPTransport:
             self._socket = ssl_socket
         except (socket.error, ssl.SSLError) as e:
             sock.close()
-            raise RuntimeError(f"Failed to connect to {self.tcp_address}: {e}")
+            raise RuntimeError(f"Failed to connect to {self.tcp_address}: {e}") from e
 
     def close(self) -> None:
         """Close socket connection"""
@@ -137,7 +137,7 @@ class TCPTransport:
                 self._socket.sendall(data)
         except (socket.error, ssl.SSLError) as e:
             self.close()
-            raise RuntimeError(f"Error sending message: {e}")
+            raise RuntimeError(f"Error sending message: {e}") from e
 
     def receive_message(self) -> Optional[bytearray]:
         """Receive a message with a prefixed header indicating its length and validate it"""
@@ -156,39 +156,46 @@ class TCPTransport:
             return None
         except (socket.error, ssl.SSLError) as e:
             self.close()
-            raise RuntimeError(f"Error receiving message: {e}")
+            raise RuntimeError(f"Error receiving message: {e}") from e
 
-    def _recvall(self, length: int) -> bytearray:
-        """socket recv data with fixed length
+    def _recvall(self, length: int) -> bytes:
+        """Receive exactly `length` bytes from the socket.
            https://stackoverflow.com/questions/64466530/using-a-custom-socket-recvall-function-works-only-if-thread-is-put-to-sleep
+
         Args:
-            length (int): data length to be received
+            length (int): Number of bytes to receive.
 
         Returns:
-            bytearray: Received data
+            bytes: Received data.
 
         Raises:
-            EOFError: If connection closes prematurely
-            ValueError: If length is invalid
+            EOFError: If connection closes before receiving all data.
+            ValueError: If length is not positive.
+            socket.timeout: If socket operation times out.
         """
         if length <= 0:
             raise ValueError(f"Invalid length: {length}")
 
-        # Pre-allocate buffer of exact size needed
+        # Pre-allocate buffer
         buffer = bytearray(length)
-        view = memoryview(buffer)
+        mv = memoryview(buffer)
         bytes_received = 0
 
         while bytes_received < length:
-            # Use recv_into to read directly into our buffer
             try:
-                chunk_size = self._socket.recv_into(view[bytes_received:], length - bytes_received)
+                # Receive directly into buffer
+                remaining = length - bytes_received
+                chunk_size = self._socket.recv_into(mv, remaining)
 
                 if chunk_size == 0:
                     raise EOFError("Connection closed by peer")
 
+                mv = mv[chunk_size:]  # advance memoryview
                 bytes_received += chunk_size
-            except socket.timeout:
-                raise socket.timeout(f"Socket operation timed out after receiving {bytes_received}/{length} bytes")
 
-        return buffer
+            except InterruptedError:
+                continue
+            except socket.timeout as e:
+                raise socket.timeout(f"Socket operation timed out after receiving {bytes_received}/{length} bytes") from e
+
+        return bytes(buffer)  # safer than bytearray

{appmesh-1.6.0.dist-info → appmesh-1.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appmesh
-Version: 1.6.0
+Version: 1.6.1
 Summary: Client SDK for App Mesh
 Home-page: https://github.com/laoshanxi/app-mesh
 Author: laoshanxi

{appmesh-1.6.0.dist-info → appmesh-1.6.1.dist-info}/RECORD

@@ -8,8 +8,8 @@ appmesh/client_tcp.py,sha256=hE0T_2Z0OQZdF5zi--iuvu2_-ka0DfSSQ4BP3oHlg44,11243
 appmesh/server_http.py,sha256=zr9sfS8g_mtP2kuAfmz-qU7rLSFTVt3srbFXaCbl8Y0,4253
 appmesh/server_tcp.py,sha256=biBFF5IGWFOw2ru831cfmzn1DVXcBm9e-W6CP2VkfzE,1444
 appmesh/tcp_messages.py,sha256=4XRv5lSm_8ElMg_37SuyIRrfxI7XFNNP_7SdO7us3PA,1023
-appmesh/tcp_transport.py,sha256=3B0w8AHXnb-E49Odxm5bPNBve1P-YiMHNDxFO2YYYVE,8690
-appmesh-1.6.0.dist-info/METADATA,sha256=VKKRa3mUsX0oahVUosSDJCR4mlsqTX-ua51-JskhSwk,11847
-appmesh-1.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-appmesh-1.6.0.dist-info/top_level.txt,sha256=-y0MNQOGJxUzLdHZ6E_Rfv5_LNCkV-GTmOBME_b6pg8,8
-appmesh-1.6.0.dist-info/RECORD,,
+appmesh/tcp_transport.py,sha256=f28zfZNH46tUHfT8F1PrCM1wUXiSBIW7R3ipMsXJqIU,8946
+appmesh-1.6.1.dist-info/METADATA,sha256=DvJTf5922gT6fc-iOWizqodwljYIkIlCCI1SeTd0juM,11847
+appmesh-1.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+appmesh-1.6.1.dist-info/top_level.txt,sha256=-y0MNQOGJxUzLdHZ6E_Rfv5_LNCkV-GTmOBME_b6pg8,8
+appmesh-1.6.1.dist-info/RECORD,,