task/socketio add base websocket service

pyproject.toml

@@ -37,6 +37,7 @@ redis = ">=4.2.0rc1"
 pydantic = {version = "^2.5.1", extras = ["email"]}
 py-fast-grpc = "^0.3.4"
 cryptography = "^42.0.4"
+python-socketio = "^5.11.1"
 
 [tool.poetry.extras]
 sqlite = ["aiosqlite"]

sapphire/common/api/service.py

@@ -6,7 +6,7 @@
 from fastapi.middleware.cors import CORSMiddleware
 from loguru import logger
 
-from .uvicorn_server import UvicornServer
+from sapphire.common.uvicorn_server import UvicornServer
 
 
 class BaseAPIService(ServiceMixin):

sapphire/common/socketio/service.py

@@ -0,0 +1,44 @@
+from typing import Iterable
+
+import socketio
+import uvicorn
+from facet import ServiceMixin
+from loguru import logger
+
+from sapphire.common.uvicorn_server import UvicornServer
+
+
+class BaseSocketIOService(ServiceMixin):
+    def __init__(
+        self,
+        root_path: str = "",
+        allowed_origins: Iterable[str] = (),
+        port: int = 8000,
+        namespaces: dict[str, socketio.AsyncNamespace] | None = None,
+    ):
+        self._root_path = root_path
+        self._port = port
+        self._server = socketio.AsyncServer(
+            async_mode="asgi",
+            cors_allowed_origins=allowed_origins,
+        )
+        if namespaces:
+            for name, namespace in namespaces.items():
+                self._server.register_namespace(namespace(name))
+
+    def get_app(self) -> socketio.ASGIApp:
+        sio_app = socketio.ASGIApp(
+            socketio_server=self._server,
+            socketio_path=self._root_path,
+        )
+        return sio_app
+
+    async def start(self):
+        config = uvicorn.Config(app=self.get_app(), host="0.0.0.0", port=self._port)
+        server = UvicornServer(config)
+
+        logger.info("Start Socket IO service")
+        self.add_task(server.serve())
+
+    async def stop(self):
+        logger.info("Stop Socket IO service")

sapphire/common/socketio/settings.py

@@ -0,0 +1,8 @@
+from pydantic import conint
+from pydantic_settings import BaseSettings
+
+
+class BaseSocketIOSettings(BaseSettings):
+    port: conint(ge=1, le=65535) = 8080
+    root_path: str = ""
+    allowed_origins: list[str] = []