leftyer/MangoTestingPlatform
1
0
Fork 0
mirror of https://gitee.com/mao-peng/MangoTestingPlatform.git synced 2025-12-06 11:59:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化链接

Browse Source
This commit is contained in:
毛鹏
2025-12-05 15:10:10 +08:00
parent 22bce7e27f
commit 3f80288052
2 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
MangoServer/src/middleware/user_logs.py
View File

@@ -12,6 +12,7 @@ from django.core.handlers.asgi import ASGIRequest
from rest_framework.response import Response from rest_framework.response import Response
from src.auto_test.auto_user.models import User from src.auto_test.auto_user.models import User
from src.auto_test.auto_user.views.user_logs import UserLogsCRUD from src.auto_test.auto_user.views.user_logs import UserLogsCRUD
from django.db import close_old_connections
class UserLogsMiddleWare(MiddlewareMixin): class UserLogsMiddleWare(MiddlewareMixin):
@@ -43,8 +44,12 @@ class UserLogsMiddleWare(MiddlewareMixin):
user_name = request_data.get('username') user_name = request_data.get('username')
if isinstance(user_name, list): if isinstance(user_name, list):
user_name = user_name[0] user_name = user_name[0]
# 确保在数据库操作前后关闭连接
close_old_connections()
user_id = User._default_manager.get(username=user_name).id user_id = User._default_manager.get(username=user_name).id
close_old_connections()
except User.DoesNotExist: except User.DoesNotExist:
close_old_connections()
pass pass
try: try:
request_data = json.dumps(request_data, ensure_ascii=False) request_data = json.dumps(request_data, ensure_ascii=False)
@@ -63,6 +68,9 @@ class UserLogsMiddleWare(MiddlewareMixin):
self._save_user_logs_async(log_entry) self._save_user_logs_async(log_entry)
except Exception: except Exception:
pass pass
finally:
# 确保在异步处理完成后关闭所有数据库连接
close_old_connections()
def _capture_request_data(self, request: ASGIRequest, response: Response) -> dict: def _capture_request_data(self, request: ASGIRequest, response: Response) -> dict:
if request.method == 'POST' or request.method == 'PUT': if request.method == 'POST' or request.method == 'PUT':
@@ -115,6 +123,10 @@ class UserLogsMiddleWare(MiddlewareMixin):
def _save_user_logs_async(self, log_entry: dict) -> None: def _save_user_logs_async(self, log_entry: dict) -> None:
"""异步保存用户日志到数据库""" """异步保存用户日志到数据库"""
try: try:
# 确保在数据库操作前后关闭连接
close_old_connections()
UserLogsCRUD.inside_post(log_entry) UserLogsCRUD.inside_post(log_entry)
close_old_connections()
except Exception as e: except Exception as e:
close_old_connections()
print(e) print(e)

4
MangoServer/src/settings/__init__.py
View File

@@ -126,10 +126,12 @@ if not IS_SQLITE:
'connect_timeout': 20, # 增加到20秒 'connect_timeout': 20, # 增加到20秒
'read_timeout': 60, # 增加读取超时到60秒 'read_timeout': 60, # 增加读取超时到60秒
'write_timeout': 60, # 增加写入超时到60秒 'write_timeout': 60, # 增加写入超时到60秒
'init_command': "SET sql_mode='STRICT_TRANS_TABLES', wait_timeout=28800", 'init_command': "SET sql_mode='STRICT_TRANS_TABLES', wait_timeout=300",
'isolation_level': 'READ COMMITTED', 'isolation_level': 'READ COMMITTED',
'autocommit': True, 'autocommit': True,
}, },
'CONN_MAX_AGE': 300, # 5分钟连接最大存活时间
'CONN_HEALTH_CHECKS': True, # Django 4.1+ 健康检查
} }
} }
else: else: