mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-22 21:06:32 +00:00
zhangtao
1d54dca76c
refactor: 优化代码结构和可读性 feat: 添加http_limit模块实现请求限制功能 fix: 修复异步任务中使用time.sleep的问题 chore: 更新依赖项并添加pytest测试框架 docs: 更新项目描述信息 perf: 优化Redis序列化方式使用JSON替代pickle test: 添加测试相关配置和依赖
105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Body, Depends
|
|
from fastapi.responses import JSONResponse
|
|
from redis.asyncio.client import Redis
|
|
|
|
from app.common.request import PaginationService
|
|
from app.common.response import ErrorResponse, SuccessResponse
|
|
from app.core.base_params import PaginationQueryParam
|
|
from app.core.dependencies import AuthPermission, redis_getter
|
|
from app.core.logger import log
|
|
from app.core.router_class import OperationLogRoute
|
|
|
|
from .schema import OnlineQueryParam
|
|
from .service import OnlineService
|
|
|
|
OnlineRouter = APIRouter(route_class=OperationLogRoute, prefix="/online", tags=["在线用户"])
|
|
|
|
|
|
@OnlineRouter.get(
|
|
"/list",
|
|
dependencies=[Depends(AuthPermission(["module_monitor:online:query"]))],
|
|
summary="获取在线用户列表",
|
|
description="获取在线用户列表",
|
|
)
|
|
async def get_online_list_controller(
|
|
redis: Annotated[Redis, Depends(redis_getter)],
|
|
paging_query: Annotated[PaginationQueryParam, Depends()],
|
|
search: Annotated[OnlineQueryParam, Depends()],
|
|
) -> JSONResponse:
|
|
"""
|
|
获取在线用户列表
|
|
|
|
参数:
|
|
- redis (Redis): Redis异步客户端实例。
|
|
- paging_query (PaginationQueryParam): 分页查询参数模型。
|
|
- search (OnlineQueryParam): 查询参数模型。
|
|
|
|
返回:
|
|
- JSONResponse: 包含在线用户列表的JSON响应。
|
|
"""
|
|
result_dict_list = await OnlineService.get_online_list_service(redis=redis, search=search)
|
|
result_dict = await PaginationService.paginate(
|
|
data_list=result_dict_list,
|
|
page_no=paging_query.page_no,
|
|
page_size=paging_query.page_size,
|
|
)
|
|
log.info("获取成功")
|
|
|
|
return SuccessResponse(data=result_dict, msg="获取成功")
|
|
|
|
|
|
@OnlineRouter.delete(
|
|
"/delete",
|
|
dependencies=[Depends(AuthPermission(["module_monitor:online:delete"]))],
|
|
summary="强制下线",
|
|
description="强制下线",
|
|
)
|
|
async def delete_online_controller(
|
|
session_id: Annotated[str, Body(description="会话编号")],
|
|
redis: Annotated[Redis, Depends(redis_getter)],
|
|
) -> JSONResponse:
|
|
"""
|
|
强制下线指定在线用户
|
|
|
|
参数:
|
|
- session_id (str): 在线用户会话ID。
|
|
- redis (Redis): Redis异步客户端实例。
|
|
|
|
返回:
|
|
- JSONResponse: 包含操作结果的JSON响应。
|
|
"""
|
|
is_ok = await OnlineService.delete_online_service(redis=redis, session_id=session_id)
|
|
if is_ok:
|
|
log.info("强制下线成功")
|
|
return SuccessResponse(msg="强制下线成功")
|
|
log.info("强制下线失败")
|
|
return ErrorResponse(msg="强制下线失败")
|
|
|
|
|
|
@OnlineRouter.delete(
|
|
"/clear",
|
|
dependencies=[Depends(AuthPermission(["module_monitor:online:delete"]))],
|
|
summary="清除所有在线用户",
|
|
description="清除所有在线用户",
|
|
)
|
|
async def clear_online_controller(
|
|
redis: Annotated[Redis, Depends(redis_getter)],
|
|
) -> JSONResponse:
|
|
"""
|
|
清除所有在线用户
|
|
|
|
参数:
|
|
- redis (Redis): Redis异步客户端实例。
|
|
|
|
返回:
|
|
- JSONResponse: 包含操作结果的JSON响应。
|
|
"""
|
|
is_ok = await OnlineService.clear_online_service(redis=redis)
|
|
if is_ok:
|
|
log.info("清除所有在线用户成功")
|
|
return SuccessResponse(msg="清除所有在线用户成功")
|
|
log.info("清除所有在线用户失败")
|
|
return ErrorResponse(msg="清除所有在线用户失败")
|