init

transport/backend/app/services/websocket_manager.py

@@ -0,0 +1,63 @@
+import json
+from datetime import datetime
+from typing import Set
+from fastapi import WebSocket
+
+from app.models import Position, Event
+
+
+class ConnectionManager:
+    def __init__(self):
+        self.active_connections: Set[WebSocket] = set()
+
+    async def connect(self, websocket: WebSocket):
+        await websocket.accept()
+        self.active_connections.add(websocket)
+
+    def disconnect(self, websocket: WebSocket):
+        self.active_connections.discard(websocket)
+
+    async def broadcast(self, message: dict):
+        """Отправить сообщение всем подключенным клиентам"""
+        disconnected = set()
+        for connection in self.active_connections:
+            try:
+                await connection.send_json(message)
+            except Exception:
+                disconnected.add(connection)
+
+        # Remove disconnected clients
+        self.active_connections -= disconnected
+
+    async def broadcast_position(self, position: Position):
+        """Отправить обновление позиции"""
+        message = {
+            "type": "position_update",
+            "data": {
+                "vehicle_id": position.vehicle_id,
+                "lat": position.lat,
+                "lon": position.lon,
+                "speed": position.speed,
+                "heading": position.heading,
+                "timestamp": position.timestamp.isoformat()
+            }
+        }
+        await self.broadcast(message)
+
+    async def broadcast_event(self, event: Event):
+        """Отправить событие"""
+        message = {
+            "type": "event",
+            "data": {
+                "id": event.id,
+                "vehicle_id": event.vehicle_id,
+                "type": event.type,
+                "payload": event.payload,
+                "timestamp": event.timestamp.isoformat()
+            }
+        }
+        await self.broadcast(message)
+
+
+# Global instance
+manager = ConnectionManager()