zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: base execute_raw_sql 增加timeout参数

This commit is contained in:
caoqianming 2025-10-24 16:10:37 +08:00
parent d4744d2602
commit b1fe8831cc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/utils/sql.py
View File

@ -2,7 +2,7 @@ from django.db import connection
from django.utils import timezone from django.utils import timezone
from datetime import datetime from datetime import datetime
def execute_raw_sql(sql: str, params=None): def execute_raw_sql(sql: str, params=None, timeout=30):
"""执行原始sql并返回rows, columns数据 """执行原始sql并返回rows, columns数据
Args: Args:
@ -10,7 +10,7 @@ def execute_raw_sql(sql: str, params=None):
params (_type_, optional): 参数列表. Defaults to None. params (_type_, optional): 参数列表. Defaults to None.
""" """
with connection.cursor() as cursor: with connection.cursor() as cursor:
cursor.execute("SET statement_timeout TO %s;", [30000]) cursor.execute(f"SET statement_timeout TO '{int(timeout*1000)}ms';")
if params: if params:
cursor.execute(sql, params=params) cursor.execute(sql, params=params)
else: else: