fbuild 1.2.8__py3-none-any.whl → 1.2.15__py3-none-any.whl

fbuild/daemon/__init__.py

@@ -14,6 +14,12 @@ from fbuild.daemon.client import (
     request_monitor,
     stop_daemon,
 )
+from fbuild.daemon.connection import DaemonConnection, connect_daemon
+from fbuild.daemon.connection_registry import (
+    ConnectionRegistry,
+    ConnectionState,
+    PlatformSlot,
+)
 from fbuild.daemon.messages import (
     BuildRequest,
     DaemonState,
@@ -26,12 +32,17 @@ from fbuild.daemon.messages import (
 
 __all__ = [
     "BuildRequest",
+    "ConnectionRegistry",
+    "ConnectionState",
+    "DaemonConnection",
     "DaemonState",
     "DaemonStatus",
     "DeployRequest",
     "InstallDependenciesRequest",
     "MonitorRequest",
     "OperationType",
+    "PlatformSlot",
+    "connect_daemon",
     "ensure_daemon_running",
     "get_daemon_status",
     "request_build",

fbuild/daemon/async_client.py

@@ -20,7 +20,8 @@ from dataclasses import asdict, dataclass, field
 from typing import Any, Callable
 
 # Default heartbeat timeout: clients not sending heartbeat in this time are considered dead
-DEFAULT_HEARTBEAT_TIMEOUT = 30.0
+# Per TASK.md: "If daemon misses heartbeats for ~3–4s, daemon closes the connection"
+DEFAULT_HEARTBEAT_TIMEOUT = 4.0
 
 # Cleanup interval for checking dead clients
 CLEANUP_INTERVAL = 10.0
@@ -275,7 +276,7 @@ class ClientConnectionManager:
 
             # Remove client from registry
             del self._clients[client_id]
-            logging.info(f"Client unregistered: {client_id} " f"(had {resource_count} attached resources)")
+            logging.info(f"Client unregistered: {client_id} (had {resource_count} attached resources)")
             return True
 
     def heartbeat(self, client_id: str) -> bool:
@@ -359,7 +360,7 @@ class ClientConnectionManager:
             for client_id, client_info in list(self._clients.items()):
                 if not client_info.is_alive(timeout_seconds):
                     dead_clients.append(client_id)
-                    logging.warning(f"Client {client_id} is dead " f"(no heartbeat for {client_info.time_since_heartbeat():.1f}s)")
+                    logging.warning(f"Client {client_id} is dead (no heartbeat for {client_info.time_since_heartbeat():.1f}s)")
 
             # Clean up dead clients
             for client_id in dead_clients:
@@ -448,7 +449,7 @@ class ClientConnectionManager:
         """
         with self._lock:
             self._cleanup_callbacks.append(callback)
-            logging.debug(f"Cleanup callback registered " f"(total callbacks: {len(self._cleanup_callbacks)})")
+            logging.debug(f"Cleanup callback registered (total callbacks: {len(self._cleanup_callbacks)})")
 
     def _call_cleanup_callbacks_unlocked(self, client_id: str) -> None:
         """Call all cleanup callbacks for a client (must hold lock).