isolate 0.22.0__py3-none-any.whl

isolate/server/health_server.py

@@ -0,0 +1,40 @@
+import asyncio
+from dataclasses import dataclass
+from typing import AsyncIterator
+
+from grpc.aio import ServicerContext
+
+from isolate.server import health
+
+
+@dataclass
+class HealthServicer(health.HealthServicer):
+    def __post_init__(self):
+        self._state = {
+            # Empty refers to the whole server
+            "": health.HealthCheckResponse.ServingStatus.SERVING,
+            "isolate": health.HealthCheckResponse.ServingStatus.SERVING,
+        }
+
+    def _get_status(
+        self, service: str
+    ) -> health.HealthCheckResponse.ServingStatus.ValueType:
+        status = self._state.get(
+            service,
+            health.HealthCheckResponse.ServingStatus.SERVICE_UNKNOWN,
+        )
+        return status
+
+    def Check(
+        self, request: health.HealthCheckRequest, context: ServicerContext
+    ) -> health.HealthCheckResponse:
+        return health.HealthCheckResponse(status=self._get_status(request.service))
+
+    async def Watch(
+        self,
+        request: health.HealthCheckRequest,
+        context: ServicerContext,
+    ) -> AsyncIterator[health.HealthCheckResponse]:
+        while True:
+            yield health.HealthCheckResponse(status=self._get_status(request.service))
+            await asyncio.sleep(2)

isolate/server/interface.py

@@ -0,0 +1,27 @@
+from typing import Any, Dict
+
+from isolate.backends import BaseEnvironment
+from isolate.connections.grpc.interface import (
+    from_grpc,
+    to_grpc,
+    to_serialized_object,
+)
+from isolate.server import definitions
+
+__all__ = ["from_grpc", "to_grpc", "to_serialized_object", "to_struct"]
+
+
+@from_grpc.register
+def _(message: definitions.EnvironmentDefinition) -> BaseEnvironment:
+    from isolate import prepare_environment
+
+    return prepare_environment(
+        message.kind,
+        **definitions.struct_to_dict(message.configuration),
+    )
+
+
+def to_struct(data: Dict[str, Any]) -> definitions.Struct:
+    struct = definitions.Struct()
+    struct.update(data)
+    return struct