mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-01 17:49:58 +08:00
5d92ccd3b0
* 性能优化 * 代码改进 * 优化浏览器代际切换逻辑 * 统一缓存与生命周期 * 添加aiomysql依赖 * 优化插件路径处理逻辑，简化条件判断；在虚拟环境包管理器中添加编码和错误处理参数以增强稳定性 * 🚨 auto fix by pre-commit hooks * 优化Windows下的关闭逻辑 * 代码优化 * bugfix:修复配置重载问题 * bugfix:修复插件加载启动竞态问题 * 收敛事件入口和权限上下文 * 优化 Windows launcher 关闭重启兜底 --------- Co-authored-by: HibiKier <775757368@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from fastapi import APIRouter
|
|
from loguru import logger
|
|
from nonebot.utils import escape_tag
|
|
from starlette.websockets import WebSocket, WebSocketDisconnect, WebSocketState
|
|
|
|
from .log_manager import LOG_STORAGE, ensure_log_sink_started, stop_log_sink_if_idle
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.websocket("/logs")
|
|
async def system_logs_realtime(websocket: WebSocket):
|
|
await websocket.accept()
|
|
await ensure_log_sink_started()
|
|
|
|
async def log_listener(log: str):
|
|
await websocket.send_text(log)
|
|
|
|
if not LOG_STORAGE.add_listener(log_listener):
|
|
await websocket.send_text("日志连接数已达上限，请稍后再试。")
|
|
await websocket.close()
|
|
stop_log_sink_if_idle()
|
|
return
|
|
try:
|
|
while websocket.client_state == WebSocketState.CONNECTED:
|
|
recv = await websocket.receive()
|
|
logger.trace(
|
|
f"{system_logs_realtime.__name__!r} received "
|
|
f"<e>{escape_tag(repr(recv))}</e>"
|
|
)
|
|
except WebSocketDisconnect:
|
|
pass
|
|
finally:
|
|
LOG_STORAGE.remove_listener(log_listener)
|
|
stop_log_sink_if_idle()
|