94 lines
2.6 KiB
Python
94 lines
2.6 KiB
Python
|
from fastapi import APIRouter, Depends, Query
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
from pydantic import BaseModel
|
||
|
|
from typing import Optional
|
||
|
|
from app.database import get_db
|
||
|
|
from app.models.user import User
|
||
|
|
from app.routers.auth import get_current_user
|
||
|
|
from app.services.log_service import LogService
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/api/logs", tags=["Log"])
|
||
|
|
|
||
|
|
|
||
|
|
class LogOut(BaseModel):
|
||
|
|
id: str
|
||
|
|
level: str
|
||
|
|
type: str
|
||
|
|
user_id: Optional[str]
|
||
|
|
message: str
|
||
|
|
source: Optional[str]
|
||
|
|
details: Optional[str]
|
||
|
|
duration_ms: Optional[str]
|
||
|
|
created_at: str
|
||
|
|
updated_at: str
|
||
|
|
|
||
|
|
model_config = {"from_attributes": True}
|
||
|
|
|
||
|
|
|
||
|
|
class LogStatsOut(BaseModel):
|
||
|
|
total: int
|
||
|
|
by_type: dict
|
||
|
|
by_level: dict
|
||
|
|
|
||
|
|
|
||
|
|
class LogQueryOut(BaseModel):
|
||
|
|
logs: list[LogOut]
|
||
|
|
total: int
|
||
|
|
page: int
|
||
|
|
page_size: int
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("", response_model=LogQueryOut)
|
||
|
|
async def list_logs(
|
||
|
|
log_type: Optional[str] = Query(None, description="日志类型: agent/system/chat"),
|
||
|
|
level: Optional[str] = Query(None, description="日志级别: debug/info/warning/error"),
|
||
|
|
source: Optional[str] = Query(None, description="来源模块"),
|
||
|
|
page: int = Query(1, ge=1),
|
||
|
|
page_size: int = Query(50, ge=1, le=200),
|
||
|
|
current_user: User = Depends(get_current_user),
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
"""查询日志列表"""
|
||
|
|
svc = LogService(db)
|
||
|
|
offset = (page - 1) * page_size
|
||
|
|
logs, total = await svc.list_logs(
|
||
|
|
log_type=log_type,
|
||
|
|
level=level,
|
||
|
|
user_id=current_user.id,
|
||
|
|
source=source,
|
||
|
|
limit=page_size,
|
||
|
|
offset=offset,
|
||
|
|
)
|
||
|
|
return LogQueryOut(
|
||
|
|
logs=[LogOut.model_validate(log) for log in logs],
|
||
|
|
total=total,
|
||
|
|
page=page,
|
||
|
|
page_size=page_size,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/stats", response_model=LogStatsOut)
|
||
|
|
async def get_log_stats(
|
||
|
|
hours: int = Query(24, ge=1, le=168),
|
||
|
|
current_user: User = Depends(get_current_user),
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
"""获取日志统计"""
|
||
|
|
svc = LogService(db)
|
||
|
|
stats = await svc.get_log_stats(hours=hours)
|
||
|
|
return LogStatsOut(**stats)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/recent", response_model=list[LogOut])
|
||
|
|
async def get_recent_logs(
|
||
|
|
log_type: Optional[str] = Query(None),
|
||
|
|
hours: int = Query(24, ge=1, le=168),
|
||
|
|
limit: int = Query(50, ge=1, le=200),
|
||
|
|
current_user: User = Depends(get_current_user),
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
"""获取最近的日志"""
|
||
|
|
svc = LogService(db)
|
||
|
|
logs = await svc.get_recent_logs(log_type=log_type, hours=hours, limit=limit)
|
||
|
|
return [LogOut.model_validate(log) for log in logs]
|