Add example websocket endpoint

Author: febus982

src/http_app/routes/__init__.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 
 from http_app.routes import api, events, graphql, hello, ping, user_registered_hook, asyncapi_docs
+from . import websocket
 
 
 def init_routes(app: FastAPI) -> None:
@@ -11,3 +12,4 @@ def init_routes(app: FastAPI) -> None:
     app.include_router(events.router)
     app.include_router(user_registered_hook.router)
     app.include_router(graphql.router, prefix="/graphql")
+    app.include_router(websocket.router)

src/http_app/routes/websocket/__init__.py

@@ -0,0 +1,6 @@
+from fastapi import APIRouter
+
+from . import chat
+
+router = APIRouter(prefix="/ws")
+router.include_router(chat.router)

src/http_app/routes/websocket/chat.py

@@ -0,0 +1,39 @@
+from fastapi import APIRouter
+from starlette.websockets import WebSocket, WebSocketDisconnect
+
+router = APIRouter(prefix="/chat")
+
+
+class ConnectionManager:
+    def __init__(self):
+        self.active_connections: list[WebSocket] = []
+
+    async def connect(self, websocket: WebSocket):
+        await websocket.accept()
+        self.active_connections.append(websocket)
+
+    def disconnect(self, websocket: WebSocket):
+        self.active_connections.remove(websocket)
+
+    async def send_personal_message(self, message: str, websocket: WebSocket):
+        await websocket.send_text(message)
+
+    async def broadcast(self, message: str):
+        for connection in self.active_connections:
+            await connection.send_text(message)
+
+
+manager = ConnectionManager()
+
+
+@router.websocket("/{client_id}")
+async def websocket_endpoint(websocket: WebSocket, client_id: int):
+    await manager.connect(websocket)
+    try:
+        while True:
+            data = await websocket.receive_text()
+            await manager.send_personal_message(f"You wrote: {data}", websocket)
+            await manager.broadcast(f"Client #{client_id} says: {data}")
+    except WebSocketDisconnect:
+        manager.disconnect(websocket)
+        await manager.broadcast(f"Client #{client_id} left the chat")