hostel-protocol-python 0.4.1__tar.gz → 0.4.2__tar.gz

{hostel_protocol_python-0.4.1 → hostel_protocol_python-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hostel-protocol-python
-Version: 0.4.1
+Version: 0.4.2
 Summary: Pydantic models, transport and client for the Hostel protocol
 Requires-Python: >=3.12
 Requires-Dist: hostel-protocol>=0.4.0

{hostel_protocol_python-0.4.1 → hostel_protocol_python-0.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hostel-protocol-python"
-version = "0.4.1"
+version = "0.4.2"
 description = "Pydantic models, transport and client for the Hostel protocol"
 requires-python = ">=3.12"
 dependencies = [

{hostel_protocol_python-0.4.1 → hostel_protocol_python-0.4.2}/src/hostel/transport/__init__.py

@@ -1,8 +1,9 @@
 """hostel.transport — Transport abstractions and implementations."""
 
-from hostel.transport.transport import ClientId, ServerTransport, Transport
+from hostel.transport.transport import ClientDisconnectedError, ClientId, ServerTransport, Transport
 
 __all__ = [
+    "ClientDisconnectedError",
     "ClientId",
     "Transport",
     "ServerTransport",

{hostel_protocol_python-0.4.1 → hostel_protocol_python-0.4.2}/src/hostel/transport/transport.py

@@ -7,6 +7,14 @@ from hostel.protocol.models import HostelMessage
 ClientId = bytes
 
 
+class ClientDisconnectedError(Exception):
+    """Raised when a send targets a peer that is no longer connected."""
+
+    def __init__(self, client_id: ClientId) -> None:
+        self.client_id = client_id
+        super().__init__(f"Client {client_id!r} is not connected")
+
+
 class Transport(ABC):
 
     @abstractmethod

{hostel_protocol_python-0.4.1 → hostel_protocol_python-0.4.2}/src/hostel/transport/zeromq.py

@@ -1,5 +1,6 @@
 """ZeroMQ transport implementations for Hostel."""
 
+import asyncio
 import logging
 import uuid
 
@@ -9,7 +10,7 @@ from hostel.protocol.v1 import message_pb2
 
 from hostel.protocol.converter import proto_to_pydantic, pydantic_to_proto
 from hostel.protocol.models import HostelMessage
-from hostel.transport.transport import ClientId, ServerTransport, Transport
+from hostel.transport.transport import ClientDisconnectedError, ClientId, ServerTransport, Transport
 
 logger = logging.getLogger(__name__)
 
@@ -76,9 +77,11 @@ class ZmqRouterTransport(ServerTransport):
         endpoint: str,
         *,
         linger_ms: int = 0,
+        send_timeout: float = 5.0,
     ):
         self.endpoint = endpoint
         self.linger_ms = linger_ms
+        self.send_timeout = send_timeout
 
         self._ctx: zmq.asyncio.Context | None = None
         self._socket: zmq.asyncio.Socket | None = None
@@ -90,6 +93,7 @@ class ZmqRouterTransport(ServerTransport):
         self._ctx = zmq.asyncio.Context.instance()
         self._socket = self._ctx.socket(zmq.ROUTER)
         self._socket.setsockopt(zmq.LINGER, self.linger_ms)
+        self._socket.setsockopt(zmq.ROUTER_MANDATORY, 1)
         self._socket.bind(self.endpoint)
 
     async def receive(self) -> tuple[ClientId, HostelMessage]:
@@ -111,7 +115,18 @@ class ZmqRouterTransport(ServerTransport):
 
         proto = pydantic_to_proto(message)
         raw = proto.SerializeToString()
-        await self._socket.send_multipart([client_id, b"", raw])
+        try:
+            await asyncio.wait_for(
+                self._socket.send_multipart([client_id, b"", raw]),
+                timeout=self.send_timeout,
+            )
+        except asyncio.TimeoutError:
+            logger.warning("send to client %r timed out after %ss", client_id, self.send_timeout)
+            raise
+        except zmq.ZMQError as exc:
+            if exc.errno == zmq.EHOSTUNREACH:
+                raise ClientDisconnectedError(client_id) from exc
+            raise
 
     async def close(self) -> None:
         if self._socket is not None: