Add debug endpoint for haproxy

Author: olethanh

src/aleph/vm/orchestrator/supervisor.py

@@ -32,6 +32,7 @@
     about_execution_records,
     about_executions,
     about_login,
+    debug_haproxy,
     list_executions,
     list_executions_v2,
     notify_allocation,
@@ -146,6 +147,7 @@ def setup_webapp(pool: VmPool | None):
         web.post("/control/machine/{ref}/confidential/initialize", operate_confidential_initialize),
         web.get("/control/machine/{ref}/confidential/measurement", operate_confidential_measurement),
         web.post("/control/machine/{ref}/confidential/inject_secret", operate_confidential_inject_secret),
+        web.get("/debug/haproxy", debug_haproxy),
         # /status APIs are used to check that the VM Orchestrator is running properly
         web.get("/status/check/fastapi", status_check_fastapi),
         web.get("/status/check/fastapi/legacy", status_check_fastapi_legacy),

src/aleph/vm/orchestrator/views/__init__.py

@@ -18,6 +18,7 @@
 from aleph_message.models import InstanceContent, ItemHash, MessageType, PaymentType
 from pydantic import ValidationError
 
+from aleph.vm import haproxy
 from aleph.vm.conf import settings
 from aleph.vm.controllers.firecracker.executable import (
     ResourceDownloadError,
@@ -148,6 +149,26 @@ async def about_login(request: web.Request) -> web.Response:
         return web.json_response({"success": False}, status=401)
 
 
+async def debug_haproxy(request: web.Request) -> web.Response:
+    """ "Debug endpoint to check the status of HAProxy and the domains mapped to it.
+
+    This is a debug endpoint and should not be used in production. The interface is subject to change.
+    """
+    socket = settings.HAPROXY_SOCKET
+    import pathlib
+
+    if not pathlib.Path(socket).exists():
+        logger.info("HAProxy not running? socket not found, skip domain mapping update")
+        return web.json_response({"status": "no socket"}, status=http.HTTPStatus)
+    r: dict = {"status": "ok", "backends": {}}
+    for backend in haproxy.HAPROXY_BACKENDS:
+        r["backends"][str(backend["name"])] = haproxy.get_current_backends(socket, backend["name"])
+    return web.json_response(
+        r,
+        dumps=dumps_for_json,
+    )
+
+
 @cors_allow_all
 async def about_executions(request: web.Request) -> web.Response:
     "/about/executions/details Debugging endpoint with full execution details."