add websockets support22312

backend/app/routers/requests.py

@@ -2,6 +2,7 @@
 from fastapi import APIRouter, Depends, HTTPException, Query, WebSocket, WebSocketDisconnect
 from sqlalchemy.orm import Session
 from typing import List, Optional
+import logging
 from ..database import get_db
 from ..crud import requests
 from ..schemas.request import Request, RequestCreate, RequestUpdate
@@ -11,16 +12,19 @@ from ..utils.telegram import notify_new_request
 from ..websockets.notifications import notification_manager
 
 router = APIRouter()
+logger = logging.getLogger(__name__)
 
 @router.websocket("/ws/admin")
 async def websocket_admin_endpoint(websocket: WebSocket):
     """WebSocket endpoint для админов"""
+    logger.info("Admin WebSocket connection attempt")
     await notification_manager.connect(websocket, "admin")
     try:
         while True:
             data = await websocket.receive_text()
-            # Здесь можно добавить обработку сообщений от админа
+            logger.info(f"Received message from admin: {data}")
     except WebSocketDisconnect:
+        logger.info("Admin WebSocket disconnected")
         notification_manager.disconnect(websocket, "admin")
 
 @router.websocket("/ws/employee/{employee_id}")
@@ -41,12 +45,15 @@ async def create_request(
     current_employee: dict = Depends(get_current_employee)
 ):
     """Create new request"""
+    logger.info(f"Creating new request from employee {current_employee['id']}")
     db_request = requests.create_request(db, request, current_employee["id"])
+    
     # Отправляем уведомление в Telegram
     await notify_new_request(db_request.id)
     
     # Получаем актуальную статистику
     stats = requests.get_statistics(db)
+    logger.info(f"Current statistics: {stats}")
     
     # Получаем полные данные о заявке для отправки через WebSocket
     request_data = {
@@ -61,15 +68,17 @@ async def create_request(
         "created_at": db_request.created_at.isoformat()
     }
     
-    # Отправляем уведомление через WebSocket всем админам
-    await notification_manager.broadcast_to_admins({
+    # Формируем сообщение для WebSocket
+    ws_message = {
         "type": "new_request",
         "data": request_data,
-        "statistics": {
-            "total": stats["total"],
-            "by_status": stats["by_status"]
-        }
-    })
+        "statistics": stats
+    }
+    
+    logger.info(f"Broadcasting WebSocket message: {ws_message}")
+    # Отправляем уведомление через WebSocket всем админам
+    await notification_manager.broadcast_to_admins(ws_message)
+    
     return db_request
 
 @router.get("/my", response_model=List[Request])
@@ -99,25 +108,29 @@ async def update_request_status(
     _: dict = Depends(get_current_admin)
 ):
     """Update request status (admin only)"""
+    logger.info(f"Updating request {request_id} status to {request_update.status}")
     db_request = requests.update_request_status(db, request_id, request_update.status)
     if db_request is None:
         raise HTTPException(status_code=404, detail="Request not found")
     
     # Получаем актуальную статистику
     stats = requests.get_statistics(db)
+    logger.info(f"Current statistics after status update: {stats}")
     
-    # Отправляем уведомление через WebSocket
-    await notification_manager.broadcast_to_admins({
+    # Формируем сообщение для WebSocket
+    ws_message = {
         "type": "status_update",
         "data": {
             "id": request_id,
             "status": db_request.status
         },
-        "statistics": {
-            "total": stats["total"],
-            "by_status": stats["by_status"]
-        }
-    })
+        "statistics": stats
+    }
+    
+    logger.info(f"Broadcasting WebSocket message for status update: {ws_message}")
+    # Отправляем уведомление через WebSocket
+    await notification_manager.broadcast_to_admins(ws_message)
+    
     return db_request
 
 @router.get("/statistics")

frontend/src/plugins/websocket.ts

@@ -22,65 +22,86 @@ class WebSocketClient {
       `${baseUrl}/api/requests/ws/admin` : 
       `${baseUrl}/api/requests/ws/employee/${id}`
 
-    console.log('Connecting to WebSocket:', url)
+    console.log('WebSocket: Connecting to', url)
     
     if (this.socket) {
-      console.log('Closing existing connection')
+      console.log('WebSocket: Closing existing connection')
       this.socket.close()
+      this.socket = null
     }
 
-    this.socket = new WebSocket(url)
+    try {
+      this.socket = new WebSocket(url)
 
-    this.socket.onopen = () => {
-      console.log('WebSocket connected')
-      this.isConnected.value = true
-      this.reconnectAttempts = 0
-    }
-
-    this.socket.onmessage = (event) => {
-      console.log('WebSocket message received:', event.data)
-      try {
-        const data = JSON.parse(event.data)
-        this.messageHandlers.forEach(handler => handler(data))
-      } catch (error) {
-        console.error('Error parsing WebSocket message:', error)
+      this.socket.onopen = () => {
+        console.log('WebSocket: Connected successfully')
+        this.isConnected.value = true
+        this.reconnectAttempts = 0
       }
-    }
 
-    this.socket.onclose = () => {
-      console.log('WebSocket disconnected')
+      this.socket.onmessage = (event) => {
+        console.log('WebSocket: Message received', event.data)
+        try {
+          const data = JSON.parse(event.data)
+          this.messageHandlers.forEach(handler => {
+            try {
+              handler(data)
+            } catch (error) {
+              console.error('WebSocket: Error in message handler:', error)
+            }
+          })
+        } catch (error) {
+          console.error('WebSocket: Error parsing message:', error)
+        }
+      }
+
+      this.socket.onclose = (event) => {
+        console.log('WebSocket: Connection closed', event.code, event.reason)
+        this.isConnected.value = false
+        this.tryReconnect()
+      }
+
+      this.socket.onerror = (error) => {
+        console.error('WebSocket: Error occurred:', error)
+        this.isConnected.value = false
+      }
+    } catch (error) {
+      console.error('WebSocket: Error creating connection:', error)
       this.isConnected.value = false
       this.tryReconnect()
     }
-
-    this.socket.onerror = (error) => {
-      console.error('WebSocket error:', error)
-      this.isConnected.value = false
-    }
   }
 
   private tryReconnect() {
     if (this.reconnectAttempts < this.maxReconnectAttempts && this.currentType) {
       this.reconnectAttempts++
-      console.log(`Attempting to reconnect (${this.reconnectAttempts}/${this.maxReconnectAttempts})`)
+      const delay = this.reconnectTimeout * Math.min(this.reconnectAttempts, 3)
+      console.log(`WebSocket: Attempting to reconnect (${this.reconnectAttempts}/${this.maxReconnectAttempts}) in ${delay}ms`)
+      
       setTimeout(() => {
+        console.log('WebSocket: Reconnecting...')
         this.connect(this.currentType!, this.currentId)
-      }, this.reconnectTimeout * this.reconnectAttempts)
+      }, delay)
+    } else {
+      console.log('WebSocket: Max reconnection attempts reached')
     }
   }
 
   addMessageHandler(handler: (data: any) => void) {
     this.messageHandlers.push(handler)
+    console.log('WebSocket: Message handler added')
   }
 
   removeMessageHandler(handler: (data: any) => void) {
     const index = this.messageHandlers.indexOf(handler)
     if (index > -1) {
       this.messageHandlers.splice(index, 1)
+      console.log('WebSocket: Message handler removed')
     }
   }
 
   disconnect() {
+    console.log('WebSocket: Disconnecting...')
     if (this.socket) {
       this.socket.close()
       this.socket = null

frontend/src/views/admin/AdminDashboard.vue

@@ -98,20 +98,24 @@ const fetchData = async () => {
 }
 
 // Обработчик WebSocket сообщений
-const handleWebSocketMessage = async (data: any) => {
+const handleWebSocketMessage = (data: any) => {
   console.log('Received WebSocket message:', data)
   
   // Обновляем статистику, если она пришла в сообщении
   if (data.statistics) {
-    statistics.value = data.statistics
+    console.log('Updating statistics:', data.statistics)
+    statistics.value = {
+      total: data.statistics.total,
+      by_status: { ...data.statistics.by_status }
+    }
   }
   
-  if (data.type === 'new_request') {
+  if (data.type === 'new_request' && data.data) {
+    console.log('Adding new request:', data.data)
     // Добавляем новую заявку в начало списка
-    if (data.data) {
-      requests.value = [data.data, ...requests.value]
-    }
-  } else if (data.type === 'status_update') {
+    requests.value = [data.data, ...requests.value.slice(0, 99)]
+  } else if (data.type === 'status_update' && data.data) {
+    console.log('Updating request status:', data.data)
     // Обновляем статус заявки в списке
     const request = requests.value.find(r => r.id === data.data.id)
     if (request) {
@@ -122,9 +126,12 @@ const handleWebSocketMessage = async (data: any) => {
 
 // Подключение к WebSocket при монтировании компонента
 onMounted(() => {
+  console.log('Component mounted, fetching initial data')
   fetchData()
+  
   // Добавляем небольшую задержку перед подключением WebSocket
   setTimeout(() => {
+    console.log('Connecting to WebSocket')
     wsClient.connect('admin')
     wsClient.addMessageHandler(handleWebSocketMessage)
   }, 1000)
@@ -132,7 +139,9 @@ onMounted(() => {
 
 // Переподключение WebSocket при потере соединения
 watch(() => wsClient.isConnected, (isConnected) => {
+  console.log('WebSocket connection status changed:', isConnected)
   if (!isConnected) {
+    console.log('Attempting to reconnect WebSocket')
     setTimeout(() => {
       wsClient.connect('admin')
     }, 3000)
@@ -141,6 +150,7 @@ watch(() => wsClient.isConnected, (isConnected) => {
 
 // Отключение от WebSocket при размонтировании компонента
 onUnmounted(() => {
+  console.log('Component unmounting, disconnecting WebSocket')
   wsClient.removeMessageHandler(handleWebSocketMessage)
   wsClient.disconnect()
 })