First Commit

Author: TutorialsAndroid

Procfile

@@ -0,0 +1 @@
+web: uvicorn server:app --host 0.0.0.0 --port $PORT

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+websockets

server.py

@@ -0,0 +1,51 @@
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+from typing import List
+import uvicorn
+import json
+
+app = FastAPI()
+
+
+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 broadcast(self, message: str):
+
+        for connection in self.active_connections:
+            await connection.send_text(message)
+
+
+manager = ConnectionManager()
+
+
+@app.websocket("/chat")
+async def websocket_endpoint(websocket: WebSocket):
+
+    await manager.connect(websocket)
+
+    try:
+
+        while True:
+
+            data = await websocket.receive_text()
+
+            message = json.loads(data)
+
+            await manager.broadcast(json.dumps(message))
+
+    except WebSocketDisconnect:
+
+        manager.disconnect(websocket)
+
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)