Browse Source

perf: 日志装饰器异常处理增加logger打印日志

master
insistence 5 months ago
parent
commit
82c84b2afe
  1. 15
      ruoyi-fastapi-backend/module_admin/annotation/log_annotation.py

15
ruoyi-fastapi-backend/module_admin/annotation/log_annotation.py

@ -16,6 +16,7 @@ from module_admin.service.login_service import LoginService
from config.enums import BusinessType from config.enums import BusinessType
from config.env import AppConfig from config.env import AppConfig
from exceptions.exception import LoginException, ServiceException, ServiceWarning from exceptions.exception import LoginException, ServiceException, ServiceWarning
from utils.log_util import logger
from utils.response_util import ResponseUtil from utils.response_util import ResponseUtil
@ -116,13 +117,14 @@ class Log:
try: try:
# 调用原始函数 # 调用原始函数
result = await func(*args, **kwargs) result = await func(*args, **kwargs)
except LoginException as e: except (LoginException, ServiceWarning) as e:
logger.warning(e.message)
result = ResponseUtil.failure(data=e.data, msg=e.message) result = ResponseUtil.failure(data=e.data, msg=e.message)
except ServiceException as e: except ServiceException as e:
logger.error(e.message)
result = ResponseUtil.error(data=e.data, msg=e.message) result = ResponseUtil.error(data=e.data, msg=e.message)
except ServiceWarning as e:
result = ResponseUtil.failure(data=e.data, msg=e.message)
except Exception as e: except Exception as e:
logger.exception(e)
result = ResponseUtil.error(msg=str(e)) result = ResponseUtil.error(msg=str(e))
# 获取请求耗时 # 获取请求耗时
cost_time = float(time.time() - start_time) * 100 cost_time = float(time.time() - start_time) * 100
@ -295,13 +297,14 @@ def log_decorator(
try: try:
# 调用原始函数 # 调用原始函数
result = await func(*args, **kwargs) result = await func(*args, **kwargs)
except LoginException as e: except (LoginException, ServiceWarning) as e:
logger.warning(e.message)
result = ResponseUtil.failure(data=e.data, msg=e.message) result = ResponseUtil.failure(data=e.data, msg=e.message)
except ServiceException as e: except ServiceException as e:
logger.error(e.message)
result = ResponseUtil.error(data=e.data, msg=e.message) result = ResponseUtil.error(data=e.data, msg=e.message)
except ServiceWarning as e:
result = ResponseUtil.failure(data=e.data, msg=e.message)
except Exception as e: except Exception as e:
logger.exception(e)
result = ResponseUtil.error(msg=str(e)) result = ResponseUtil.error(msg=str(e))
# 获取请求耗时 # 获取请求耗时
cost_time = float(time.time() - start_time) * 100 cost_time = float(time.time() - start_time) * 100

Loading…
Cancel
Save