feat: 添加 WebSocket 日志推送功能，支持实时日志监控和历史日志加载

src/webui/logs_ws.py

@@ -0,0 +1,138 @@
+"""WebSocket 日志推送模块"""
+from fastapi import APIRouter, WebSocket, WebSocketDisconnect
+from typing import Set
+import json
+from pathlib import Path
+from src.common.logger import get_logger
+
+logger = get_logger("webui.logs_ws")
+router = APIRouter()
+
+# 全局 WebSocket 连接池
+active_connections: Set[WebSocket] = set()
+
+
+def load_recent_logs(limit: int = 100) -> list[dict]:
+    """从日志文件中加载最近的日志
+    
+    Args:
+        limit: 返回的最大日志条数
+        
+    Returns:
+        日志列表
+    """
+    logs = []
+    log_dir = Path("logs")
+    
+    if not log_dir.exists():
+        return logs
+    
+    # 获取所有日志文件,按修改时间排序
+    log_files = sorted(log_dir.glob("app_*.log.jsonl"), key=lambda f: f.stat().st_mtime, reverse=True)
+    
+    # 用于生成唯一 ID 的计数器
+    log_counter = 0
+    
+    # 从最新的文件开始读取
+    for log_file in log_files:
+        if len(logs) >= limit:
+            break
+            
+        try:
+            with open(log_file, "r", encoding="utf-8") as f:
+                lines = f.readlines()
+                # 从文件末尾开始读取
+                for line in reversed(lines):
+                    if len(logs) >= limit:
+                        break
+                    try:
+                        log_entry = json.loads(line.strip())
+                        # 转换为前端期望的格式
+                        # 使用时间戳 + 计数器生成唯一 ID
+                        timestamp_id = log_entry.get("timestamp", "0").replace("-", "").replace(" ", "").replace(":", "")
+                        formatted_log = {
+                            "id": f"{timestamp_id}_{log_counter}",
+                            "timestamp": log_entry.get("timestamp", ""),
+                            "level": log_entry.get("level", "INFO").upper(),
+                            "module": log_entry.get("logger_name", ""),
+                            "message": log_entry.get("event", ""),
+                        }
+                        logs.append(formatted_log)
+                        log_counter += 1
+                    except (json.JSONDecodeError, KeyError):
+                        continue
+        except Exception as e:
+            logger.error(f"读取日志文件失败 {log_file}: {e}")
+            continue
+    
+    # 反转列表，使其按时间顺序排列（旧到新）
+    return list(reversed(logs))
+
+
+@router.websocket("/ws/logs")
+async def websocket_logs(websocket: WebSocket):
+    """WebSocket 日志推送端点
+    
+    客户端连接后会持续接收服务器端的日志消息
+    """
+    await websocket.accept()
+    active_connections.add(websocket)
+    logger.info(f"📡 WebSocket 客户端已连接，当前连接数: {len(active_connections)}")
+    
+    # 连接建立后，立即发送历史日志
+    try:
+        recent_logs = load_recent_logs(limit=100)
+        logger.info(f"发送 {len(recent_logs)} 条历史日志到客户端")
+        
+        for log_entry in recent_logs:
+            await websocket.send_text(json.dumps(log_entry, ensure_ascii=False))
+    except Exception as e:
+        logger.error(f"发送历史日志失败: {e}")
+    
+    try:
+        # 保持连接，等待客户端消息或断开
+        while True:
+            # 接收客户端消息（用于心跳或控制指令）
+            data = await websocket.receive_text()
+            
+            # 可以处理客户端的控制消息，例如：
+            # - "ping" -> 心跳检测
+            # - {"filter": "ERROR"} -> 设置日志级别过滤
+            if data == "ping":
+                await websocket.send_text("pong")
+                
+    except WebSocketDisconnect:
+        active_connections.discard(websocket)
+        logger.info(f"📡 WebSocket 客户端已断开，当前连接数: {len(active_connections)}")
+    except Exception as e:
+        logger.error(f"❌ WebSocket 错误: {e}")
+        active_connections.discard(websocket)
+
+
+async def broadcast_log(log_data: dict):
+    """广播日志到所有连接的 WebSocket 客户端
+    
+    Args:
+        log_data: 日志数据字典
+    """
+    if not active_connections:
+        return
+    
+    # 格式化为 JSON
+    message = json.dumps(log_data, ensure_ascii=False)
+    
+    # 记录需要断开的连接
+    disconnected = set()
+    
+    # 广播到所有客户端
+    for connection in active_connections:
+        try:
+            await connection.send_text(message)
+        except Exception:
+            # 发送失败，标记为断开
+            disconnected.add(connection)
+    
+    # 清理断开的连接
+    if disconnected:
+        active_connections.difference_update(disconnected)
+        logger.debug(f"清理了 {len(disconnected)} 个断开的 WebSocket 连接")

src/webui/manager.py

@@ -31,6 +31,14 @@ def setup_webui(mode: str = "production") -> bool:
 
 def setup_dev_mode() -> bool:
     """设置开发模式 - 仅启用 CORS，前端自行启动"""
+    from src.common.server import get_global_server
+    from .logs_ws import router as logs_router
+    
+    # 注册 WebSocket 日志路由（开发模式也需要）
+    server = get_global_server()
+    server.register_router(logs_router)
+    logger.info("✅ WebSocket 日志推送路由已注册")
+    
     logger.info("📝 WebUI 开发模式已启用")
     logger.info("🌐 请手动启动前端开发服务器: cd webui && npm run dev")
     logger.info("💡 前端将运行在 http://localhost:7999")
@@ -42,6 +50,7 @@ def setup_production_mode() -> bool:
     try:
         from src.common.server import get_global_server
         from starlette.responses import FileResponse
+        from .logs_ws import router as logs_router
         import mimetypes
         
         # 确保正确的 MIME 类型映射
@@ -52,6 +61,11 @@ def setup_production_mode() -> bool:
         mimetypes.add_type('application/json', '.json')
         
         server = get_global_server()
+        
+        # 注册 WebSocket 日志路由
+        server.register_router(logs_router)
+        logger.info("✅ WebSocket 日志推送路由已注册")
+        
         base_dir = Path(__file__).parent.parent.parent
         static_path = base_dir / "webui" / "dist"