hd-crane-gs/backend/app/api/v1/module_monitor/online/controller.py

# -*- coding: utf-8 -*-

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 SuccessResponse,ErrorResponse
from app.core.dependencies import AuthPermission, redis_getter
from app.core.base_params import PaginationQueryParam
from app.core.router_class import OperationLogRoute
from app.core.logger import log

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: Redis = Depends(redis_getter), 
    paging_query: PaginationQueryParam = Depends(),
    search: 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: str = Body(..., description="会话编号"),
    redis: 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="强制下线成功")
    else:
        log.info("强制下线失败")
        return ErrorResponse(msg="强制下线失败")

@OnlineRouter.delete(
    '/clear', 
    dependencies=[Depends(AuthPermission(['module_monitor:online:delete']))],
    summary="清除所有在线用户",
    description="清除所有在线用户"
)
async def clear_online_controller(
    redis: 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="清除所有在线用户成功")
    else:
        log.info("清除所有在线用户失败")
        return ErrorResponse(msg="清除所有在线用户失败")