142 lines
4.6 KiB
Python
142 lines
4.6 KiB
Python
"""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 连接")
|