diff --git a/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py b/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py index ef30793..5ec87ce 100644 --- a/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py +++ b/ruoyi-fastapi-backend/module_admin/controller/sys_statistics_controller.py @@ -70,6 +70,40 @@ async def add_model_call_count( else: logger.error('大模型调用次数加1失败') return ResponseUtil.error(msg=data.message) + +# 门禁通行量次数加1 +@sys_statisticsController.get( + '/add_door_pass_count' +) +async def add_door_pass_count( + request: Request, + query_db: AsyncSession = Depends(get_db), + user: CurrentUserModel = Depends(LoginService.get_current_user), +): + data = await Sys_statisticsService.add_door_pass_count_services(query_db, user.user.user_name) + if data.is_success: + logger.info('门禁通行量次数加1成功') + return ResponseUtil.success(msg=data.message) + else: + logger.error('门禁通行量次数加1失败') + return ResponseUtil.error(msg=data.message) +# 访客引导次数加1 +@sys_statisticsController.get( + '/add_visitor_guide_count' +) +async def add_visitor_guide_count( + request: Request, + query_db: AsyncSession = Depends(get_db), + user: CurrentUserModel = Depends(LoginService.get_current_user), +): + data = await Sys_statisticsService.add_visitor_guide_count_services(query_db, user.user.user_name) + if data.is_success: + logger.info('访客引导次数加1成功') + return ResponseUtil.success(msg=data.message) + else: + logger.error('访客引导次数加1失败') + return ResponseUtil.error(msg=data.message) + # @sys_statisticsController.get( # '/list', response_model=PageResponseModel # # , dependencies=[Depends(CheckUserInterfaceAuth('system:sys_statistics:list'))] diff --git a/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py b/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py index 104c0f1..b41e2b8 100644 --- a/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py +++ b/ruoyi-fastapi-backend/module_admin/service/sys_statistics_service.py @@ -14,6 +14,29 @@ class Sys_statisticsService: """ 系统统计数据模块服务层 """ + # 访客引导量加1 + @classmethod + async def add_visitor_guide_count_services(cls, query_db: AsyncSession, user_name: str): + """ + 新增访客引导量 + :param query_db: orm对象 + :return: + """ + await Sys_statisticsDao.add_count(query_db, "visitor_guide", user_name, 1) + await query_db.commit() + return CrudResponseModel(is_success=True, message='更新成功') + # 门禁通行量加1 + @classmethod + async def add_door_pass_count_services(cls, query_db: AsyncSession, user_name: str): + """ + 新增门禁通行量 + :param query_db: orm对象 + :return: + """ + await Sys_statisticsDao.add_count(query_db, "access_control_count", user_name, 1) + await query_db.commit() + return CrudResponseModel(is_success=True, message='更新成功') + # 大模型调用量加1 @classmethod async def add_model_call_count_services(cls, query_db: AsyncSession, user_name: str): """