PowLu/RuoYi-Vue3-FastAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf: 优化在线用户模块service层及异常处理

Browse Source
This commit is contained in:
insistence
2024-07-10 20:50:39 +08:00
parent 55babcee4c
commit 4d31fe7a92
2 changed files with 13 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
ruoyi-fastapi-backend/module_admin/controller/online_controller.py
View File

@@ -16,28 +16,18 @@ onlineController = APIRouter(prefix='/monitor/online', dependencies=[Depends(Log
@onlineController.get("/list", response_model=PageResponseModel, dependencies=[Depends(CheckUserInterfaceAuth('monitor:online:list'))])
async def get_monitor_online_list(request: Request, online_page_query: OnlineQueryModel = Depends(OnlineQueryModel.as_query)):
try:
# 获取全量数据
online_query_result = await OnlineService.get_online_list_services(request, online_page_query)
logger.info('获取成功')
return ResponseUtil.success(model_content=PageResponseModel(rows=online_query_result, total=len(online_query_result)))
except Exception as e:
logger.exception(e)
return ResponseUtil.error(msg=str(e))
# 获取全量数据
online_query_result = await OnlineService.get_online_list_services(request, online_page_query)
logger.info('获取成功')
return ResponseUtil.success(model_content=PageResponseModel(rows=online_query_result, total=len(online_query_result)))
@onlineController.delete("/{token_ids}", dependencies=[Depends(CheckUserInterfaceAuth('monitor:online:forceLogout'))])
@log_decorator(title='在线用户', business_type=BusinessType.FORCE)
async def delete_monitor_online(request: Request, token_ids: str, query_db: AsyncSession = Depends(get_db)):
try:
delete_online = DeleteOnlineModel(tokenIds=token_ids)
delete_online_result = await OnlineService.delete_online_services(request, delete_online)
if delete_online_result.is_success:
logger.info(delete_online_result.message)
return ResponseUtil.success(msg=delete_online_result.message)
else:
logger.warning(delete_online_result.message)
return ResponseUtil.failure(msg=delete_online_result.message)
except Exception as e:
logger.exception(e)
return ResponseUtil.error(msg=str(e))
delete_online = DeleteOnlineModel(tokenIds=token_ids)
delete_online_result = await OnlineService.delete_online_services(request, delete_online)
logger.info(delete_online_result.message)
return ResponseUtil.success(msg=delete_online_result.message)

6
ruoyi-fastapi-backend/module_admin/service/online_service.py
View File

@@ -3,6 +3,7 @@ from jose import jwt
from config.env import JwtConfig, RedisInitKeyConfig
from module_admin.entity.vo.online_vo import *
from module_admin.entity.vo.common_vo import CrudResponseModel
from exceptions.exception import ServiceException
from utils.common_util import CamelCaseUtil
@@ -65,7 +66,6 @@ class OnlineService:
token_id_list = page_object.token_ids.split(',')
for token_id in token_id_list:
await request.app.state.redis.delete(f"{RedisInitKeyConfig.ACCESS_TOKEN.get('key')}:{token_id}")
result = dict(is_success=True, message='强退成功')
return CrudResponseModel(is_success=True, message='强退成功')
else:
result = dict(is_success=False, message='传入session_id为空')
return CrudResponseModel(**result)
raise ServiceException(message='传入session_id为空')