|
16 | 16 | from config.enums import BusinessType |
17 | 17 | from config.env import AppConfig |
18 | 18 | from exceptions.exception import LoginException, ServiceException, ServiceWarning |
| 19 | +from utils.log_util import logger |
19 | 20 | from utils.response_util import ResponseUtil |
20 | 21 |
|
21 | 22 |
|
@@ -116,13 +117,14 @@ async def wrapper(*args, **kwargs): |
116 | 117 | try: |
117 | 118 | # 调用原始函数 |
118 | 119 | result = await func(*args, **kwargs) |
119 | | - except LoginException as e: |
| 120 | + except (LoginException, ServiceWarning) as e: |
| 121 | + logger.warning(e.message) |
120 | 122 | result = ResponseUtil.failure(data=e.data, msg=e.message) |
121 | 123 | except ServiceException as e: |
| 124 | + logger.error(e.message) |
122 | 125 | result = ResponseUtil.error(data=e.data, msg=e.message) |
123 | | - except ServiceWarning as e: |
124 | | - result = ResponseUtil.failure(data=e.data, msg=e.message) |
125 | 126 | except Exception as e: |
| 127 | + logger.exception(e) |
126 | 128 | result = ResponseUtil.error(msg=str(e)) |
127 | 129 | # 获取请求耗时 |
128 | 130 | cost_time = float(time.time() - start_time) * 100 |
@@ -295,13 +297,14 @@ async def wrapper(*args, **kwargs): |
295 | 297 | try: |
296 | 298 | # 调用原始函数 |
297 | 299 | result = await func(*args, **kwargs) |
298 | | - except LoginException as e: |
| 300 | + except (LoginException, ServiceWarning) as e: |
| 301 | + logger.warning(e.message) |
299 | 302 | result = ResponseUtil.failure(data=e.data, msg=e.message) |
300 | 303 | except ServiceException as e: |
| 304 | + logger.error(e.message) |
301 | 305 | result = ResponseUtil.error(data=e.data, msg=e.message) |
302 | | - except ServiceWarning as e: |
303 | | - result = ResponseUtil.failure(data=e.data, msg=e.message) |
304 | 306 | except Exception as e: |
| 307 | + logger.exception(e) |
305 | 308 | result = ResponseUtil.error(msg=str(e)) |
306 | 309 | # 获取请求耗时 |
307 | 310 | cost_time = float(time.time() - start_time) * 100 |
|
0 commit comments